search

ppogg / YOLOv5-Lite

🍅🍅🍅YOLOv5-Lite: Evolved from yolov5 and the size of model is only 900+kb (int8) and 1.7M (fp16). Reach 15 FPS on the Raspberry Pi 4B~
GNU General Public License v3.0
2.24k stars 405 forks source link

onnx export with NMS does not work. #258

Open omair18 opened 6 months ago

omair18 commented 6 months ago

Hi @ppogg . Exporting model to ONNX with NMS support does not work. I can see all the export end2end lines have been commented out. https://github.com/ppogg/YOLOv5-Lite/blob/075697b49e845fb1ee34b1fb8004313f79e585b4/export.py#L103

also, non_max_suppression_export function in common.py is missing. https://github.com/ppogg/YOLOv5-Lite/blob/075697b49e845fb1ee34b1fb8004313f79e585b4/models/common.py#L397