Docker容器中的CustomVision分类:无法输入形状值

时间:2019-03-07 14:13:03

标签: azure-cognitive-services

我已经在CustomVision中创建了一个分类模型,并将其导出到Dockerfile(Linux)。从CustomVision GUI进行测试时,该模型运行良好,但是在连接到Docker容器并像调用它一样

curl -X POST http://127.0.0.1/image -F imageData=@some_file_name.jpg

我总是会收到类似

的错误
  

“错误:无法对图像进行预处理以进行预测。无法为形状为((?,224,224,3)'的Tensor'占位符:0'馈入形状(1,227,227,3)的值”

甚至当some_file_name.jpg是模型从中训练的文件之一时

观察:我们在18年8月创建并导出到Dockerfiles的模型运行良好。现在修改这些模型(例如,从训练数据中删除文件)并重建模型时,如上所述会失败。与8月份相比,导出模型时创建的zip文件现在的大小几乎增加了一倍。尚未更改任何配置,并且该模型仍在同一数据中心上构建。

任何提示/帮助都非常感谢。

1 个答案:

答案 0 :(得分:0)

在导出的应用文件夹中,有一个文件Forecast.py。更改行,

network_input_size = 227 至 network_input_size = 224

然后我重建并重新运行了docker容器,它成功了。