Update demo_dense_visualize.py

demo_dense_visualize.py

@@ -108,13 +108,13 @@ class Tracker:
             device: Torch device, defaults to 'cuda'.
         """
         self.model = model
-        self.mean = mean
-        self.std = std
         self.S = S
         self.stride = stride
         self.inference_iters = inference_iters
         self.device = device
         self.target_res = target_res
+        self.mean = mean.to(device)
+        self.std = std.to(device)
 
         self.padder = None
         self.cnt = 0