From acc73e9bb04193c44042a6d117a89d0e5f5a0e7d Mon Sep 17 00:00:00 2001 From: JianHe0628 <jl02958@surrey.ac.uk> Date: Sat, 14 Jun 2025 16:25:28 +0100 Subject: [PATCH] trying self.mesh --- hamer/models/hamer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hamer/models/hamer.py b/hamer/models/hamer.py index 2615ebf..d42a828 100644 --- a/hamer/models/hamer.py +++ b/hamer/models/hamer.py @@ -55,9 +55,9 @@ class HAMER(pl.LightningModule): # Buffer that shows whetheer we need to initialize ActNorm layers self.register_buffer('initialized', torch.tensor(False)) # Setup renderer for visualization - if init_renderer: + if self.mesh: self.renderer = SkeletonRenderer(self.cfg) - # self.mesh_renderer = MeshRenderer(self.cfg, faces=self.mano.faces) + self.mesh_renderer = MeshRenderer(self.cfg, faces=self.mano.faces) else: self.renderer = None self.mesh_renderer = None -- GitLab