Update visualizer.py
Browse files- visualizer.py +3 -3
visualizer.py
CHANGED
@@ -101,9 +101,9 @@ class Visualizer:
|
|
101 |
):
|
102 |
if compensate_for_camera_motion:
|
103 |
assert segm_mask is not None
|
104 |
-
if segm_mask is not None:
|
105 |
-
|
106 |
-
|
107 |
|
108 |
video = F.pad(
|
109 |
video,
|
|
|
101 |
):
|
102 |
if compensate_for_camera_motion:
|
103 |
assert segm_mask is not None
|
104 |
+
# if segm_mask is not None:
|
105 |
+
# coords = tracks[0, query_frame].round().long()
|
106 |
+
# segm_mask = segm_mask[0, query_frame][coords[:, 1], coords[:, 0]].long()
|
107 |
|
108 |
video = F.pad(
|
109 |
video,
|