我已经从Tensorflow培训过“带有Mobilenet的SSD”模型。训练进展顺利。 现在,当我尝试通过在图像上运行object_detection_tutorial.ipynb来测试推理图的性能时,出现以下错误:
ValueError:无法将大小为X的数组重塑为形状(a,b,c)
X,a,b,c是不同测试图像的不同值。
我不认为图像尺寸会导致问题,因为模型必须独立于输入图像尺寸执行。事实上,即使使用我用于训练的图像,我也会收到此错误。
请在这里帮忙。
答案 0 :(得分:1)
@Mandroid建议,以编程方式将输入图像更改为3通道可能是可行的方法,但这就是我最终解决问题的方式。
注意:我不确定从图像中删除Alpha可能会带来一些后果。但是,这是某种信息丢失。
用image = Image.open(<image_path>)
代替image = Image.open(<image_path>).convert('RGB')
对我来说很有效。