search

GeekAlexis / FastMOT

High-performance multiple object tracking based on YOLO, Deep SORT, and KLT 🚀
MIT License
1.12k stars 256 forks source link

Support for YOLOV5 #53

Closed satyajitghana closed 3 years ago

satyajitghana commented 3 years ago

How can i replace the current yolov4 model in repo with yolov5 from https://github.com/ultralytics/yolov5 ?

GeekAlexis commented 3 years ago

@satyajitghana The current TensorRT plugin doesn't support YOLOv5. You can try using PyTorch for inference instead. Since I'm going to support Scaled YOLOv4, which is better than YOLOv5 in my opinion, I don't have plans to support YOLOv5 at the moment.