How to customize or reduce the number of classes in yolov4 model?

4 次查看（过去 30 天）
Ramya Sivanandham
Ramya Sivanandham 2022-7-18
回答： Sanjana 2023-9-1
There are 80 number of classes are there in the yolov4 - coco matlab model. But, i dont require the all those clasess. With the same model, is any possible to call the particular class while doing the testing? Like, I need only 3 classes such as Car, truck, Person - In that case, is any possible to extract those 3 class trained data alone? If yes, please let me know.
or
Is anyone has the yolov2 pretrained model for person detection along with vehicle? (Totally, 2 classes)

请先登录，再回答此问题。

回答（1 个）

Sanjana
Sanjana 2023-9-1
Hi Jesse,
I understand that you are looking for a way to customize the “yolov4ObjectDetector” by reducing the no. of classes, which are equal to 80 for a standard “yolov4ObjectDetector”.
Transfer learning can be applied on the pre-trained “yolov4ObjectDetector”, to detect any number of classes.
Please refer to the following documentation, for creating a custom “yolov4ObjectDetector”,
https://www.mathworks.com/help/vision/ref/yolov4objectdetector.html#mw_89976544-2f97-4677-9268-d2d6f8918fed
Hope this helps!
Regards,
Sanjana

类别

在 Help Center 和 File Exchange 中查找有关 ThingSpeak 的更多信息

标签

产品


版本

R2022a

Community Treasure Hunt

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

Start Hunting!

Translated by