Classified networks deploy

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.25k stars 195 forks source link

Open huangqi111 opened 4 months ago

huangqi111 commented 4 months ago

Can you provide the deployment of the yolov8 classification network

FeiYull commented 4 months ago

@huangqi111 do it later

Sencc commented 4 months ago

@huangqi111 The inference of yolov8-cls is very simple. You only need to remove its post-processing operation on yolov8-det.