我正在创建图像数据集,以使用Tensoflow Detection API(SSD / MobileNet)训练检测器。
图像为灰度图像,但似乎输入应为RGB图像。
我是否需要通过仅将第一个通道复制到其他两个通道来将灰度图像转换为三通道RGB?(如果是,是否有任何软件可以执行此操作?)或另外两个通道应该为空?(是否有用于执行此操作的软件?)
最好的祝福。
是的,您必须将灰度图像转换为RGB图像。
一个可能的解决方案是使用OpenCV:
import cv2
# suppose that gray_img is your grayscale image
input = cv2.cvtColor(gray_img, cv2.COLOR_GRAY2RGB)
现在您可以将其input
用作模型的有效输入图像
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句