对于TensorFlow对象检测API,最好具有多大的图像尺寸?

时间:2018-03-07 05:26:57

标签: tensorflow object-detection

我想知道何时使用预先训练的模型训练我自己的图像数据集(例如: ssd_mobilenet_v1_coco ssd_inception_v2_coco faster_rcnn_inception_v2_coco ... )在开始创建TF记录和培训之前,什么图像尺寸更适合调整图像大小?

我问这是因为当我收集图像时它有不同的大小,有些非常大(1920x1080...),但我不知道调整大小是否合适图片。

它是否取决于我想要选择的型号?如果有,那么每个型号的尺寸更好?

提前致谢!

1 个答案:

答案 0 :(得分:0)

您参考的大多数(全部?)模型都是在224 x 224尺寸的图像上进行训练。为获得最佳性能,您需要将图像缩放到该大小。除非您正在检测微小物体,否则将较大尺寸图像调整为模型所需的较小尺寸似乎不会影响性能。

如果您想要的物体很小并且您害怕失去分辨率,您可以将原始图像裁剪成较小的部分进行训练然后重新组装。这将是耗时但可以工作的。我在图上检测PSV符号的回购使用了这种技术:https://github.com/siddiqaa/psvcounter