如何让Tensorflow对象检测api使用灰度图像进行训练（输入张量只有1个通道）？

Question

我只想在使用由tensorflow物体检测api训练的模型时获得实时速度，输入张量具有形状[1，宽度，高度，3]，它是3个通道，但我想如果我可以使用1通道训练我的模型，它只需要灰色图像作为输入，因此，这可以降低计算复杂度，这对我的应用来说，速度非常重要。

Answer 1

我们没有任何预制模型可以制作一个频道图像。如果您对创建新模型感兴趣，请考虑添加新的flexbox。

话虽这么说，你可能看不到你模型的切实加速。仅使用一个通道仅影响网络的输入层，而不影响中间层。如果性能是您最关心的问题，最好考虑使用更薄的Mobilenet功能提取器。