如何在视力训练包的训练中标准化图像?

时间:2019-06-07 16:59:35

标签: tensorflow google-aiy

我遵循了makers guide并训练了一个自定义移动网络模型,并将其加载到我的视觉套件中。为了测试模型,我使用位于以下位置的测试脚本:

/AIY-projects-python/src/examples/vision/mobilenet_based_classifier.py

但出现以下错误:

Nothing detected when threshold=.1

我整个上午都在浏览代码,以试图解决这个问题。

首先,我将阈值降低到0,但这没有用。它只是给出:当阈值= 0时未检测到任何东西。为此,我认为图像肯定有问题。

因此,我在脚本中插入了一行,以捕捉来自相机的图像并保存。我看了看,看起来还不错。但是我想这并不是说脚本使用的帧很好,因为我拍摄了自定义图像而不是保存脚本使用的帧。...

但是后来我注意到图像正在被规范化,而在训练模型时却忘记了这样做。 mean和std的默认值为128。我打开inference.py确认如何进行标准化,并看到公式为:

(图片-128)/ 128。

因此,我经过培训回到了绘图板上,并根据上述公式对图像进行了归一化。但是随后该模型根本无法做出准确的预测。

现在我什至不知道该问什么。我该如何工作?是什么给我我的门槛问题?我是否需要以视觉套件对其进行标准化的相同方式对训练数据进行标准化以使其起作用?

谢谢

0 个答案:

没有答案