wow

main.py

@@ -44,6 +44,7 @@ best_model.eval()
 #### Initiatize tracker
 tracker_model = "./checkpoints/ckpt.t7"
 tracker = DeepSort(model_path=tracker_model,max_age=30,nn_budget=100, max_iou_distance=0.7, max_dist=0.2)
+tracker = tracker.to("cuda" if torch.cuda.is_available() else "cpu")  
 
 
 ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif', 'mp4', 'avi','webm'}