Update demo_dense_visualize.py

demo_dense_visualize.py

@@ -152,7 +152,7 @@ class Tracker:
         self.rgbs.append(rgb_tensor.cpu())
             
         # import pdb; pdb.set_trace()
-        rgb_tensor = (rgb_tensor - self.mean) / self.std
+        rgb_tensor = (rgb_tensor - self.mean.cuda()) / self.std.cuda()
         return rgb_tensor
 
     @torch.no_grad()