目前,我正在训练图像检测模型,我想知道每个类需要多少图像,我是否需要每个对象的数量相同。
我需要一些建议。
我使用Tensorflow和Yolo v2模型。
谢谢,
答案 0 :(得分:1)
你需要尽可能多的数量,但至少如果你从头开始训练网络,你肯定需要数万个数量级(YOLOv2训练过的预训练重量) - http://host.robots.ox.ac.uk/pascal/VOC/)。
最好有平衡的课程,这意味着每个课程的图像数量必须接近,这样训练更容易。
你为什么要自己训练网络?您是否可以使用一些预先训练的模型,删除FC图层并插入自己的类?这种方式要快得多,而且你不需要那么多的图片。