【问题标题】:Grayscale input image for SSD detector in Tensorflow Detection APITensorflow 检测 API 中 SSD 检测器的灰度输入图像
【发布时间】:2018-07-11 13:04:37
【问题描述】:
我正在创建一个图像数据集,以使用 Tensoflow 检测 API (SSD/MobileNet) 训练检测器。
图像是灰度的,但似乎输入应该是 RGB 图像。
我是否需要通过将第一个通道复制到其他两个通道来将灰度图像转换为三通道 RGB? (如果是,是否有任何软件可以做到这一点?)或者另外两个通道应该是空的? (有什么软件可以做到吗?)
最好的问候。
【问题讨论】:
标签:
tensorflow
conv-neural-network
training-data
【解决方案1】:
是的,您必须将灰度图像转换为 RGB 图像。
一种可能的解决方案是使用 OpenCV:
import cv2
# suppose that gray_img is your grayscale image
input = cv2.cvtColor(gray_img, cv2.COLOR_GRAY2RGB)
现在您可以使用input 作为模型的有效输入图像