我想将此 NSFW 模型转换为 CoreML 模型。我做了什么:
- 下载 Anaconda 2.7
- 安装 coremltools
- 从这里转换这个 yahoo nsfw 模型 - https://github.com/yahoo/open_nsfw/tree/master/nsfw_model但我不确定它是 Caffe v1,因为 Apple 文档说只支持这个版本。无论如何……</li>
我使用此命令进行转换,它在没有任何警告的情况下进行了转换。
coreml_model = coremltools.converters.caffe.convert(('resnet_50_1by2_nsfw.caffemodel', 'deploy.prototxt'), image_input_names='data') coreml_model.save('nsfw2.mlmodel')
- 我将这个模型导入到我的项目中,一切看起来都很好。
- 我准备了 224x224 图像并使用 Vision 框架,如 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)]
如何调试此问题并找出问题所在?