Tensorflow Detection API中用于SSD检测器的灰度输入图像

时间:2018-07-11 13:04:38

标签: tensorflow conv-neural-network training-data

我正在创建图像数据集,以使用Tensoflow Detection API(SSD / MobileNet)训练探测器。

图像是灰度图像,但似乎输入应该是RGB图像。

是否仅需将第一个通道复制到其他两个通道,即可将灰度图像转换为三通道RGB? (如果是,是否有任何软件可以执行此操作?)或另外两个通道应该为空? (是否有用于执行此操作的软件?)

最诚挚的问候。

1 个答案:

答案 0 :(得分:1)

是的,您必须将灰度图像转换为RGB图像。

可能的解决方案是使用OpenCV:

import cv2

# suppose that gray_img is your grayscale image
input = cv2.cvtColor(gray_img, cv2.COLOR_GRAY2RGB)

现在您可以将input用作模型的有效输入图像