search

WongKinYiu / yolov9

Implementation of paper - YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information
GNU General Public License v3.0
9.04k stars 1.44k forks source link

how to use nms when i use export.py to export onnx #523

Open jifeng632 opened 5 months ago

jifeng632 commented 5 months ago

when i use export.py to export onnx model, i found there's no nms but after i tried export.py with --nms the exported model still need to write nms when use the onnx model

these models yolov9-s.pt gelan-s.pt

jifeng632 commented 5 months ago

when i use https://huggingface.co/Xenova/gelan-c_all , i found img_v3_02cd_fcecef1c-badb-4468-ab39-ce96000ec6bh there's already nms in the exported model , but when i use export.py
img_v3_02cd_416142c8-7983-49e5-99d4-e0dfdb2e01ch here we need to write nms , how to export onnx model with nms and the output like [xxx, 6]