每个班级中的图像数量

时间:2017-03-30 11:37:50

标签: tensorflow deep-learning object-detection

目前,我正在训练图像检测模型,我想知道每个类需要多少图像,我是否需要每个对象的数量相同。

我需要一些建议。

我使用Tensorflow和Yolo v2模型。

谢谢,

1 个答案:

答案 0 :(得分:1)

你需要尽可能多的数量,但至少如果你从头开始训练网络,你肯定需要数万个数量级(YOLOv2训练过的预训练重量)   - http://host.robots.ox.ac.uk/pascal/VOC/)。

最好有平衡的课程,这意味着每个课程的图像数量必须接近,这样训练更容易。

你为什么要自己训练网络?您是否可以使用一些预先训练的模型,删除FC图层并插入自己的类?这种方式要快得多,而且你不需要那么多的图片。