使用协同工具从Caffe转换后，mlmodel无法正常工作

Question

我想把这个NSFW模型转换成CoreML模型。我所做的：

coremltools

• Convert

1. 下载Anaconda 2.7
2. 安装这个雅虎nsfw模式- https://github.com/yahoo/open_nsfw/tree/master/nsfw_model，但我不确定它是Caffe v1，因为苹果文档中说只有这个版本支持。不管怎样，…
3. --我使用此命令进行转换，并在没有任何警告的情况下进行转换。

coremltools.converters.caffe.convert(('resnet_50_1by2_nsfw.caffemodel'，'deploy.prototxt')，coreml_model _coreml_model_coreml_model=‘data’)将这个模型导入到我的项目中，这一切看起来都很好。

• I编写了224x224图像，并使用了VNImageRequestHandler和cgImage等视觉框架，以及

。

但!

所有图像都返回相同的结果。

[<VNCoreMLFeatureValueObservation: 0x281b1daa0> 2E00F417-95C0-4AA1-A621-A0945BB5E095 requestRevision=1 confidence=1.000000 "prob" - "MultiArray : Double 1 x 1 x 2 x 1 x 1 array" (1.000000)]

我如何调试这个问题并找出问题所在？

​

​

Answer 1

也许你只是在看淘气的照片？

可能是图像预处理。您没有指定任何预处理选项，而Caffe模型通常使用ImageNet平均值/std进行规范化。有关更多信息，请参阅我的博客文章：https://machinethink.net/blog/help-core-ml-gives-wrong-output/

但是，我在您的deploy.prototxt中没有看到任何规范化选项，所以可能不是这样。

我将如何调试它:从Caffe模型中移除第一层之外的所有内容，并转换为Core。在Caffe和Core中运行这个单层模型，并比较输出。如果它们不同，则与加载或预处理输入数据的方式有关。