MATLAB Answers

Training YOLO V2 with multiple (more than one) classes

12 ビュー (過去 30 日間)
Zahra Moayed
Zahra Moayed 2019 年 8 月 5 日
コメント済み: Sanjeev Madhave 2020 年 11 月 18 日
Hi all,
When I train YOLOV2 with single class (person) using trainYOLOv2ObjectDetector, I can get precision/recall of 0.92 but when I add another class (car) with same images and just few car labels, the accuracy is 0, meaning even the person cannot be detected in any of the images even my training images!
I even use AnchorBox estimation and treid many times.
All the matlab examples are tarined only on single objects but how about if we have more than one class to be trained? Does anyone have any success to help me please?
  4 件のコメント
Sanjeev Madhave
Sanjeev Madhave 2020 年 11 月 18 日
Hi Zahra Moayed,
I was also trying to train the yolo with multi class. I have a doubt. How did you mapped the training dataset table?
In my case, in some training image one class may not be there. In such cases how to fill the table? just leave it as empty? in that case matlab is throwing error. any help is much appreciated


回答 (2 件)

Srivardhan Gadila
Srivardhan Gadila 2019 年 8 月 14 日
編集済み: Srivardhan Gadila 2019 年 8 月 14 日
The procedure is same for both single and multi-class. The zero accuracy may imply that the dataset is biased, so try having nearly equal number of labels for cars and persons.
  3 件のコメント
Zahra Moayed
Zahra Moayed 2019 年 8 月 26 日
Hi Srivardhan,
To update, I built another labelling session to contain both People and Car. Number of objects are 137 and 141 for people and car, respectively so the dataset is totally fine.
Still after multiple trial, I got 0 accuracy, nothing is detected at all.
One question: do you train any network with YOLOV2 with multiple classes before? I want to narrow down the problem to see if the issue is from my side or the YOLO V2 release has got issue.
Thanks a lot.


Zahra Moayed
Zahra Moayed 2019 年 10 月 7 日
Is there any updates in the new version of Matlab regarding this issue? Or does anyone have any experience in training multiple classes?
  3 件のコメント
Oscar Lema
Oscar Lema 2020 年 3 月 26 日


Community Treasure Hunt

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

Start Hunting!

Translated by