search

marcoslucianops / DeepStream-Yolo

NVIDIA DeepStream SDK 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models
MIT License
1.38k stars 343 forks source link

Model Export #514

Open tommyshelby4 opened 4 months ago

tommyshelby4 commented 4 months ago

Hello everybody and congratulations on your amazing work. Could somebody confirm that gen_wts_yoloV5.py is replaced by export_yolo_V5.py?

Samjith888 commented 4 months ago

Hey @tommyshelby4 You don't need to replace any script. Just copy the export_yoloV5.py script into your yolov5 repo and generate the files by using this script. (Note: Don't generate the onnx file by using yolov5/export.py script)