Update src/pixel3dmm/tracking/tracker.py

src/pixel3dmm/tracking/tracker.py

@@ -128,12 +128,17 @@ if COMPILE:
 
 
 class Tracker(object):
-    def __init__(self, config, flame_module, renderer,
+    def __init__(self, config, flame_module,
                  device='cuda:0',
                  ):
         self.config = config
         self.flame = flame_module
-        self.diff_renderer = renderer
+
+        self.diff_renderer = NVDRenderer(self.config.size,
+                                                 obj_filename=mesh_file,
+                                                 no_sh=self.no_sh,
+                                                 white_bg= True,
+                                                 ).to(self.device)
     
         self.config = config
         self.device = device
@@ -253,9 +258,8 @@ class Tracker(object):
             self.opt_post = torch.compile(self.opt_post)
             self.actual_smooth = torch.compile(self.actual_smooth)
 
-
         self.renderer = self.diff_renderer  # already global
-
+        
 
         self.faces = load_obj(mesh_file)[1]