[YOLOv5] v7.0 TensorRT support

FeiYull / TensorRT-Alpha

🔥🔥🔥TensorRT for YOLOv8、YOLOv8-Pose、YOLOv8-Seg、YOLOv8-Cls、YOLOv7、YOLOv6、YOLOv5、YOLONAS......🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎

GNU General Public License v2.0

1.28k stars 198 forks source link

[YOLOv5] v7.0 TensorRT support #2

Closed sonderlau closed 1 year ago

sonderlau commented 1 year ago

The latest verson of YOLOv5 is v7.0

I noticed that git checkout 6.0 command is noted in YOLOv5 TensorRT README.md

It has been a few month since the v7.0 released, and due to some reasons, I have trained my model on the latest version YOLOv5 (v7.0).

I wonder whether there's a chance that this repo may add support for v7.0 of YOLOv5 ?

Any further information is helpful. Looking forward to your reply.

FeiYull commented 1 year ago

@sonderlau Hi，TensorRT-Alpha support yolov5.7.0 and yolov5.6.0 now， Thank you for your suggestion！ This is doc：https://github.com/FeiYull/tensorrt-alpha/blob/main/yolov5/README.md

sonderlau commented 1 year ago

Sorry for being late, I hadn't have time to test it yet. I'll keep you updated if anything goes wrong. Anyway, thank you for your excellent work and your quick response. Happy lunar new year !