How to reduce number of classes in YOLOv3 files?

后端 未结 1 862
予麋鹿
予麋鹿 2020-12-01 13:25

I am using YOLOv3 to detect cars in videos. I downloaded three files used in my code coco.names, yolov3.cfg and yolov3.weights which a

相关标签:
1条回答
  • 2020-12-01 13:59

    For easy and simple way using COCO dataset, follow these steps :

    • Modify (or copy for backup) the coco.names file in darknet\data\coco.names
    • Delete all other classes except car
    • Modify your cfg file (e.g. yolov3.cfg), change the 3 classes on line 610, 696, 783 from 80 to 1
    • Change the 3 filters in cfg file on line 603, 689, 776 from 255 to 18 (derived from (classes+5)x3)
    • Run the detector ./darknet detector test cfg/coco.data cfg/yolov3.cfg yolov3.weights data/your_image.jpg

    For more advance way using COCO dataset you can use this repo to create yolo datasets based on voc, coco or open images. https://github.com/holger-prause/yolo_utils .
    Also refer to this : How can I download a specific part of Coco Dataset?

    Would be great if you can train YOLO model using your own dataset. There are so many tutorial on the internet of how to build your own dataset. Like this, this, this or this.

    Note : reducing number of classes won't make your inference speed faster. By reducing classes, you will detect less object and somehow will probably make your program run faster if you do post-processing for each detection.

    0 讨论(0)
提交回复
热议问题