Qengineering / YoloV5-NPU

YoloV5 NPU for the RK3566/68/88

Home Page:https://qengineering.eu/deep-learning-examples-on-raspberry-32-64-os.html

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Custom Yolo Model

programmeddeath1 opened this issue · comments

Hi team,

Are you aware of how this can be implemented with custom models?

I tried converting my custom yolo models to onnx and then to rknn, but the inference logic fail because of mismatch in output layers.

Are you aware of how this can be setup for custom number of classes?

@programmeddeath1,

We are aware of the issue and addressing it.

Is there already a solution to this problem?

Not yet.
To be honest, I'm very busy with other projects. In the coming months I do no have any time to address this issue.