我的16位灰度图像(通道= 1)出错。在object_detection_tutorial文件jupyter笔记本中我有以下错误: 值误差:不能为Tensor' image_tensor:0'提供形状值(1,480,640,1),它具有形状'(?,?,?,3)'
为预训练模型提供16位1通道图像需要哪些步骤?
此行错误:
def load_image_into_numpy_array(image):
(im_width, im_height) = image.size
return np.array(image.getdata()).reshape(
(im_height, im_width,1)).astype(np.uint16)
答案 0 :(得分:1)
目前包含在TensorFlow object detection API中的预训练模型均可在3通道RGB图像上运行,因此您将无法在灰度图像上使用它们。
为此,在调整模型定义以对单通道图像进行操作后,您可能需要训练自己的模型。