Tensorflow对象检测:更改模型的输入分辨率

时间:2018-02-26 21:44:33

标签: tensorflow object-detection

我正在使用Tensorflow的对象检测API,使用repository中提供的代码来分析高分辨率图像(4K)。我已经测试了Model Zoo中提供的一些预先训练的模型。模型配置为将输入图像缩小到其config file中指定的特定大小。由于大幅调整输入图像的大小,因此在我的图像中未检测到小对象。

是否有办法更改输入图像的分辨率,以便为预训练模型调整大小?

示例:ssd_mobilenet_v1_coco预先训练的模型在其配置文件中包含以下行:

image_resizer {
  fixed_shape_resizer {
    height: 300
    width: 300
  }
}

我尝试更改配置文件以调整大小为600x600并在预先训练的model.ckpt上重新运行export_inference_graph.py以获取新的推理图。虽然这确实导出了一个新图形,但在将其用于对象检测时,结果并不像预期的那样。结果不准确,当它确实正确地检测到一个物体时,边界框太大了2倍。显然,这不是这样做的方法。

我要问的是:改变预训练模型的输入分辨率的正确方法是什么?

0 个答案:

没有答案