SSD Mobilenet v1 coco - 我应该在标记它们进行培训之前调整图像大小吗?

时间:2018-06-16 20:23:58

标签: object tensorflow detection

我使用的是戴尔服务器,配备2个Nvidia V100 GPU,Ubuntu 16.04,Tensorflow 1.7。

我想弄清楚何时调整图片大小。在ssd_mobilenet_v1_coco.config中,它有

image_resizer {   fixed_shape_resizer {     身高:300     宽度:300   } }

我在很多很多图像上绘制了边界框,总共有10个类。图像大小不同,有些大,有些小。我对物体的检测表现非常糟糕。我尝试将一堆图像调整为300x300,然后重新绘制边界框,但这没有帮助,图像质量很差,缩放不正确。

我的问题是:在标记之前,我是否需要将图像缩放到300x300?如果我不这样做,SSD Mobilenet会调整大小,我的边界框不会排列,或者它们是否与图像一起适当调整大小?我不想在大图像上绘制边界框,然后在SSD Mobilenet调整大小时让它们不正确。

提前致谢。

1 个答案:

答案 0 :(得分:0)

由于我不知道您如何学习SSD MobileNet,所以我无法真正回答您的问题。

我假设您想使用自己的数据集并将其转换为tfrecord,对吧?

  • 如果是这样,您将不得不自行调整大小
  • 如果您要使用COCO数据集,则可以使用pycocotools来创建所需的tfrecord

最好的问候, 瓦茨拉夫