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



我正在创建一个图像数据集,以使用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作为模型的有效输入图像

最新更新