Open itachi1232gg opened 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'
Only Detection models are supported by TRT End2End. I will review the code, to raise a warning.
Check this implementation #446
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'