Import yolov5s.onnx with a yolov5ObjectDetector

6 views (last 30 days)
I am trying to import a yolov5.onnx into the Gorund Truth Labeler App.
I have already imported and merged the layers of the network. However, there is no yolov5Detector. Will something like this be imported in the foreseeable future or is there another possibility to create a detector based on my own yolov5?

Answers (1)

David Willingham
David Willingham on 7 Oct 2021
Hi Niklas,
Are you able to point to where the ONNX yolov5 was obtained? It's hard to say why the detector is not there without looking at it directly.
David
  2 Comments
Niklas F
Niklas F on 17 Oct 2021
Edited: Niklas F on 17 Oct 2021
Hello David,
thanks for your answer.
The yolov5 onnx is a standard network that we trained on our own data at the university. It is an onnx because our network runs on python and we generate our training material with the Ground Truth Labeler App.
The problem I have now is that I can import the network, but cannot create a detector from it to create an algorithm and use it in the Ground Truth Labeler App. My goal is to implement my trained network in the Ground Truth Labeler App in order to then generate new training material with this Network.
The path I have followed so far is to implement the ONNX as a layergraph and then to put it together into a network.
For a yolov3 there is then the possibility to generate a yolov3ObjectDetector from a network. Will this option also be available for yolov5?
Regards
Niklas
David Willingham
David Willingham on 19 Oct 2021
Hi Niklas,
We don't have yolov5 support in MATLAB, however this yolov4 example could be used as a guide. In it a "detectYOLOv4.m" constructs the detector. Something similar to this could be called in the Ground Truth Labeler App.
It may also work with your yolov5, but without having the onnx file, it would be diffucult to confirm this. If you are able to share this, we could take a look.

Sign in to comment.

Products


Release

R2021b

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!