2017-09-25 113 views
2

是否有运行物体检测API中可用的物体检测网络的最佳尺寸?该API似乎接受所有尺寸的图像,但是在向网络馈送数据之前,我不清楚图像是如何重新缩放的。 请你澄清一下吗?物体检测的图像大小API

谢谢!

回答

2

答案取决于您正在运行的模型。对于我们的SSD型号,我们会将图像重塑为300x300像素。对于FasterRCNN或RFCN,我们将在600-1024像素之间重塑。

用户应该添加到TFRecord中的图像可以是任何尺寸,但我们建议用户保持尽可能小的尺寸(即,对于SSD,每个尺寸最大〜400-600px,或对于FasterRCNN,每个尺寸最大〜1500px;或者R-FCN)出于记忆原因。

+0

对于SavedModel上的预测,我们应该首先重塑图像还是将其馈入image_tensor:0原样?我看到更快的rcnn模型预测时间像45秒/图像。试图调试。 – bw4sz

+0

但是,简单的调整大小会扭曲所有输入图像(更多,更少),这不会对检测质量产生负面影响吗?我一直困惑这一段时间,你可能是正确的人问:) – Alex

1

有一个脚本名为preprocessor_builder负责。因此,无论何时将图像提供给网络,都必须经过此预处理,并确保根据网络配置文件正确调整图像大小以匹配网络。

实际调整大小正在发生here