search

AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
http://pjreddie.com/darknet/
Other
21.65k stars 7.96k forks source link

How to train Yolo 9000 for only image classification? #5475

Open mitesh741 opened 4 years ago

mitesh741 commented 4 years ago

I want to train Yolo 9000 for image classification.

Lets say for dogs and cats. I have two separate folders containing images of cats and dogs respectively.

How can I train it?

Siset commented 4 years ago

You could just use yolo on the coco dataset as usual (or with less classes if you want #5486 ) and then postprocess the image. For example, the simplest way is to classify a dog when a dog instance is found.