achen353 / Taiwanese-Traffic-Object-Detection

Training and fine-tuning YOLOv4 Tiny on custom object detection dataset for Taiwanese traffic
Other
10 stars 3 forks source link

Please provide yolov4-tiny-obj.cfg #3

Open sriks6711 opened 8 months ago

sriks6711 commented 8 months ago

@achen353 - Please provide the 'yolov4-tiny-obj.cfg' so as to run the command [1] successfully. I checked the cfg/ folder and also search the whole repository but could not locate it.

[1] $./darknet detector demo data/obj.data cfg/yolov4-tiny-obj.cfg yolov4-tiny-obj_best.weights sample_videos/...

xinthemayanksachan commented 2 months ago

Below are the changes that we have made for successful execution.

Few Modifications we have Made:

A. In File cfg/yolov4-tiny-custom.cfg:
    1. Replace filters=255 to filters=27 at line 212 and 263
    2. Replace classes=80 to classes=4 at line 218 and 267

B. In File cfg/coco.data:
    1. Replace classes= 80 to classes= 4 at line 1

C. In File data/coco.names:
    Replace the file contents as below:
        vehicle
        scooter
        pedestrian
        bicycle

After above modifications are made, ececute the below command: ./darknet detector demo cfg/coco.data cfg/yolov4-tiny-custom.cfg backup/yolov4-tiny-obj_best.weights sample_input video eg: ABC.mp4