使用Tensorflow对象检测API时是否所有图像都必须具有相同大小？

Question

我有一组分辨率不同的图像（已标记），我希望使用对象检测api进行训练。通过将配置文件中的min_dimension和max_dimension指定为最大宽度和最小高度，如果我所有图像的大小都不同，是否可以训练它？

Answer 1

否，在使用tensorflow对象检测api时，所有图像的大小都不必相同。

您不需要调整训练图像的大小。脚本本身会注意这一点。无论您的图片尺寸是多少，它们都会由配置文件在内部调整大小。

这是ssdmobilenet配置文件的示例

image_resizer {
  fixed_shape_resizer {
    height: 300
    width: 300
  }

image resizer将所有图像的大小调整为300x300，并进行训练。

如果您想获得更高的精度，可以按如下方式从配置文件中更改这些值（请注意，这样做需要更多时间来训练模型）

image_resizer {
  fixed_shape_resizer {
    height: 600
    width: 800
  }

无论您选择添加到图像缩放器的任何值，请确保所有图像尺寸均为> = image resizer尺寸。否则，您将得到张量形状不匹配错误。