0

我正在使用 tensorflow 和 tflite 来检测对象。我使用的模型是来自https://github.com/tensorflow/models/tree/master/research/object_detection的 mobilenet_ssd(版本 2)

检测的输入图像大小固定为300*300,在模型中是硬编码的。我想输入1280*720的图像进行检测,怎么做?我没有分辨率为 1280*720 的训练图像数据集。我只有 pascal 和 coco 数据集。

如何修改模型以接受 1280*720 图像(不缩放图像)进行检测?

4

2 回答 2

1

要改变图像的输入尺寸,需要重新设计锚框位置。因为锚点是固定在输入图像分辨率上的。一旦将锚点位置更改为 720P,移动网络就可以接受 720p 作为输入。

于 2020-08-18T06:49:14.573 回答
0

通常的做法是在将数据输入 TensorFlow / TensorFlow Lite 之前缩放输入图像。

注意:训练数据集中的图像原本不是 300*300。原始分辨率可能更大且非正方形,并且缩小到 300*300。这意味着将 1280*720 图像缩小到 300*300 图像是完全可以的,它应该可以正常工作。

你介意尝试缩放并看看它是否有效吗?

于 2018-06-14T19:23:10.957 回答