16位图像(PNG)

时间:2018-02-12 05:27:21

标签: tensorflow object-detection

我的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)

1 个答案:

答案 0 :(得分:1)

目前包含在TensorFlow object detection API中的预训练模型均可在3通道RGB图像上运行,因此您将无法在灰度图像上使用它们。

为此,在调整模型定义以对单通道图像进行操作后,您可能需要训练自己的模型。