Open mitesh741 opened 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?
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.
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?