diff --git a/hamer/models/hamer.py b/hamer/models/hamer.py index 2615ebf49a414cb868e20c83619c8430442c39c9..d42a8281bd4c0828f7776e6143ac8f3e600e3613 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