Tensorflow Detection 中 SSD 检测器的灰度输入图像 API
Grayscale input image for SSD detector in Tensorflow Detection API
我正在创建图像数据集以使用 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 作为模型的有效输入图像
我正在创建图像数据集以使用 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 作为模型的有效输入图像