fix ValueError: tile cannot extend outside image

mikel-brostrom · May 23, 2023 · 22f1e76 · 22f1e76
1 parent b720bf6
commit 22f1e76
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/yolo_nas_track.py b/examples/yolo_nas_track.py
@@ -128,7 +128,7 @@ def run(args):
                     prediction.labels[:, np.newaxis]
                 ], axis=1
             )
-            preds = torch.from_numpy(preds)
+            preds = torch.from_numpy(preds).int()
         predictor.results = [None]
         # # Postprocess
         with predictor.profilers[2]: