microsoft / onnxruntime-extensions

onnxruntime-extensions: A specialized pre- and post- processing library for ONNX Runtime
MIT License
341 stars 91 forks source link

trained pt files convert pre_post_processing.onnx file Predict no result #851

Open FengYanNMG opened 3 days ago

FengYanNMG commented 3 days ago

The pre_post_processing.onnx files obtained by converting yolo11n.pt or yolov8n.pt with yoloe2e can be used correctly, but the pt files trained by me with data sets cannot Detection objects correctly after conversion, and the converted ONNX files can also be predicted. But the pre_post_processing.onnx file has no result

Predict with best.onnx Image Predict with pre_post_processing.onnx Image here is my .pt file https://github.com/FengYanNMG/NewRepo/blob/master/best.pt

FengYanNMG commented 2 days ago

@wenbingl

FengYanNMG commented 2 days ago

I'm sure there's something wrong with my model, but I don't know where the problem lies.