Update demo_dense_visualize.py

demo_dense_visualize.py

@@ -146,6 +146,8 @@ class Tracker:
         # Convert to tensor, normalize and move to device.
         rgb_tensor = torch.from_numpy(rgb_resized).permute(2, 0, 1).float().unsqueeze(0).to(self.device)
         rgb_tensor = rgb_tensor / 255.0
+
+        print(self.device, rgb_tensor.device, self.mean.device, self.std.device)
         
         self.rgbs.append(rgb_tensor.cpu())