• DepthAI
  • Bad results with YoloV6N blob model

I'm using this colab notebook to train a YoloV6N model.

The pt model I get worked perfectly, like this

As shown in the notebook, I train my model and then download my pytorch weights. I use the online tool to convert it to blob format. I then use this file to check the inference of this blob model on the same image.
Unfortunately, I get extremely bad results, as shown here

This doesn't look like a problem to be solved with IoU threshold change (since I see no correct bounding boxes around objects at all)

I use "640" and "640 640" both in the online tool for converting pt to blob model. It didn't fix the issue (in the colab training notebook, I used "--img-size 640" for training).

Can someone please help with this? I already opened an issue on github here

  • erik replied to this.

    Hi msharad19 ,
    We apologize for the issue. I am still curious about the IoU, and perhaps it would be fastest if you could share the .pt as well, so our team can try to debug this as well. Thoughts?
    Thanks, Erik

    Upgraded depthai to 2.20.2.0 (as pointed out in the github issue). It now works.

    Thanks!