Tensorflow对象检测:ValueError:无法重塑数组

时间:2017-12-07 16:09:08

标签: tensorflow object-detection

我已经从Tensorflow培训过“带有Mobilenet的SSD”模型。训练进展顺利。 现在,当我尝试通过在图像上运行object_detection_tutorial.ipynb来测试推理图的性能时,出现以下错误:

ValueError:无法将大小为X的数组重塑为形状(a,b,c)

X,a,b,c是不同测试图像的不同值。

我不认为图像尺寸会导致问题,因为模型必须独立于输入图像尺寸执行。事实上,即使使用我用于训练的图像,我也会收到此错误。

请在这里帮忙。

1 个答案:

答案 0 :(得分:1)

@Mandroid建议,以编程方式将输入图像更改为3通道可能是可行的方法,但这就是我最终解决问题的方式。

注意:我不确定从图像中删除Alpha可能会带来一些后果。但是,这是某种信息丢失。

image = Image.open(<image_path>)代替image = Image.open(<image_path>).convert('RGB')对我来说很有效。