Update src/pixel3dmm/tracking/tracker.py

src/pixel3dmm/tracking/tracker.py

@@ -128,17 +128,12 @@ if COMPILE:
 
 
 class Tracker(object):
-    def __init__(self, config, flame_module,
+    def __init__(self, config, flame_module, renderer,
                  device='cuda:0',
                  ):
         self.config = config
         self.flame = flame_module
-
-        self.diff_renderer = NVDRenderer(self.config.size,
-                                                 obj_filename=mesh_file,
-                                                 no_sh=self.no_sh,
-                                                 white_bg= True,
-                                                 ).to(self.device)
+        self.diff_renderer = renderer
     
         self.config = config
         self.device = device
@@ -258,8 +253,9 @@ 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]
 
@@ -1728,4 +1724,3 @@ class Tracker(object):
 
 
 
-