我正在创建图像数据集,以使用Tensoflow Detection API(SSD / MobileNet)训练探测器。
图像是灰度图像,但似乎输入应该是RGB图像。
是否仅需将第一个通道复制到其他两个通道,即可将灰度图像转换为三通道RGB? (如果是,是否有任何软件可以执行此操作?)或另外两个通道应该为空? (是否有用于执行此操作的软件?)
最诚挚的问候。
答案 0 :(得分:1)
是的,您必须将灰度图像转换为RGB图像。
可能的解决方案是使用OpenCV:
import cv2
# suppose that gray_img is your grayscale image
input = cv2.cvtColor(gray_img, cv2.COLOR_GRAY2RGB)
现在您可以将input
用作模型的有效输入图像