custom model do not detect object

[masgh21]

@kkarahainko

Hello I am student and trained my custom model with colab
[https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data
]
(https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data)

it works with detect.py
But after I convert it to onnx , it doesn't detect anything
what Is Problem ?

I upload my model this link and
my test image
onnx Model
https://drive.google.com/file/d/1FpNythBNten9OKu3TlLamIcDtZQd980_/view?usp=sharing
test Image

thanks

[masgh21]

@keesschollaart81
Hello
Can you help me ?

[MasIgor]

how did you convert it to onnx? are you running it still with detect.py as onnx or in a different environemnt?

[masgh21]

Hello @tanzerlana
I test it with detect.py and with .pt format and it works.
after converted it to onnx ,
it didnt detect anything .

[keesschollaart81]

This has not much to do with this repo I believe. I think you're better of creating an issue of the Yolov5 repo, right? Then I also suggest to include this commands you run for the export to onnx and (if possible) the original pt file.

[masgh21]

@tanzerlana
@keesschollaart81
i upload my .ppt model here.
model

[keesschollaart81]

I'm not going to look at it, I'm sorry. If you have issues with Yolov5 conversion/quality, please report at their repo. This repo only is about doing inference with a working model using .NET.

[masgh21]

I want to use it with yolov5net
and i don't know how to use it.
@keesschollaart81

[keesschollaart81]

Before you can use it with yolov5-net, the onnx version first need to work, if you have an issue with that --> https://github.com/ultralytics/yolov5/issues

When the model works with native yolov5 (python) and you have issues with the .NET inference, discuss here.

[masgh21]

@keesschollaart81
my problem is
onnx version
and it does not work , and i need help to use onnx and yolov5net .

[masgh21]

@keesschollaart81
I converted my .pt model to onnx
after that i couldn't load it and I got this Error

public override int Dimensions { get; set; } = 6;
public override List Labels { get; set; } = new List()
{
new YoloLabel { Id = 1, Name = "helmet" }
};

[MasIgor]

verify that your onnx file is properly working. run detect.py and input the onnx file instead of the pt file. so you know that your onnx file is ok in the first place.

[masgh21]

@tanzerlana
I test it with detect.py now and it works with onnx format

in visual studio i got same error
IndexError