如何讓Tensorflow對象檢測API使用灰度圖像進行訓練（輸入張量只有1個通道）？

Question

我只是想通過張量流對象檢測api訓練模型來獲得實時速度，輸入張量具有形狀[1，寬度，高度，3]，它是3個通道，但我認爲如果我可以只使用1個通道訓練我的模型，它只需要灰色圖像作爲輸入，因此，這可以降低計算複雜度，對於我的應用程序來說，速度非常重要。

Answer 1

我們沒有任何預訓練模型可以做一個通道圖像。如果您有興趣創建新模型，請考慮添加新的SSD feature extractor。

這就是說，你可能沒有看到有形的加速到你的模型。僅使用一個通道只會影響網絡的輸入層，而不會影響中間層。如果性能是您最關心的問題，那麼考慮使用更薄的Mobilenet功能提取器可能會更好。