search

PUTvision / qgis-plugin-deepness

Deepness is a remote sensing plugin that enables deep learning inference in QGIS
https://www.buymeacoffee.com/deepness
Apache License 2.0
108 stars 25 forks source link

Loading onnx model gives error #108

Closed bspckishore closed 1 year ago

bspckishore commented 1 year ago

Hi, I have trained a model and exported to onnx in ultralytics hub. But when I am loading the same in deepness plugin in qgis gives error : TypeError: 'type' object cannot be interpreted as an integer image

przemyslaw-aszkowski commented 1 year ago

Hi,

Seems like some bug related to reading the model parameters. Do you mind sharing the model file (or some simplified version of it)? We would debug it then.

Cheers, Przemek

Amjath22 commented 1 year ago

Hi, I had the same issue when i trained a detection model in Yolov7. After some minor modifications i rectified it. if you like i can share the code for .pt to ONNX conversion.

bspckishore commented 1 year ago

Hi Przemek, Please find below the model file in onnx format. model_- 18 september 2023 14_50.zip

bspckishore commented 1 year ago

Hi Amjath, Can you please share the code to bspckishore@gmail.com.

Thanks

Amjath22 commented 1 year ago

Okay.

przemyslaw-aszkowski commented 1 year ago

Hi both, please let me later know if you managed to solve the issue or if should I investigate.

bspckishore commented 1 year ago

Hi Przemek, The code shared by Amjath didnt solved my problem, can you please look into it.

Thanks

przemyslaw-aszkowski commented 1 year ago

Hi @bspckishore, Please have a look here: https://github.com/PUTvision/qgis-plugin-deepness/issues/114 Is your model also YOLO segmentation?