search

WongKinYiu / yolov9

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

Do yolo-seg support exporting to "ONNX END2END"? #368

Open itachi1232gg opened 4 months ago

itachi1232gg commented 4 months ago

python export.py --weights ./gelan-c-seg.pt --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640 --include onnx_end2end

ONNX END2END: export failure 2s: 'tuple' object has no attribute 'permute'

levipereira commented 4 months ago

Only Detection models are supported by TRT End2End. I will review the code, to raise a warning.

levipereira commented 3 months ago

Check this implementation #446