tracker

main.py

@@ -43,8 +43,8 @@ best_model.eval()
 
 #### Initiatize tracker
 tracker_model = "./checkpoints/ckpt.t7"
+tracker_model = tracker_model.to("cuda" if torch.cuda.is_available() else "cpu") 
 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'}