ayooshkathuria / YOLO_v3_tutorial_from_scratch

Accompanying code for Paperspace tutorial series "How to Implement YOLO v3 Object Detector from Scratch"
https://blog.paperspace.com/how-to-implement-a-yolo-object-detector-in-pytorch/
2.32k stars 725 forks source link

bad results when change data/coco.names to data/voc.names #13

Open zjcbatman opened 6 years ago

zjcbatman commented 6 years ago

I have changed 80 to 20 and nearly everything related to class numbers in the code, but I got bad results. Why?

ayooshkathuria commented 6 years ago

I guess the weights file is trained on COCO dataset, and not Pascal VOC? The dataset has an effect on the shape of feature map in the output code.