CODEWITHSUNDEEP
tensorflowdeep-learningobject-detection
Kamel BOUYACOUB
Kamel BOUYACOUB

Reputation: 647

Number of images in each classes

Currently, I am training a model for image detection, and I want to know how many images do I need per class, do i need to have the same numbers of each object.

Please i need some advice.

I use Tensorflow, and Yolo v2 model.

Thanks,

Upvotes: 1

Views: 1904

Answers (1)

Mihail Burduja
Mihail Burduja

Reputation: 3256

You need as many as you can get, but definitely in the order of tens of thousands at least if you're training the network from scratch (there are pre-trained weights for YOLOv2 trained on - http://host.robots.ox.ac.uk/pascal/VOC/).

It's best to have balanced classes, meaning the number of images for each class must be close, it's easier to train this way.

Why are you training the network yourself? Can't you use some pre-trained models, drop the FC layers and insert your own classes? This way it's much faster, and you don't need that many images.

Upvotes: 1

Related Questions