如何检查mlmodel(coreml)的预处理参数?

时间:2018-08-15 04:11:16

标签: ios neural-network coreml coremltools

我正在尝试获取有关我拥有的coreml模型的预处理信息。

我正在使用CoreML 2.01b。 我尝试遵循此代码,但由于预处理属性似乎已不存在,因此它似乎已过时。

spec = coremltools.models.utils.load_spec(coreml_model_file)
if spec.WhichOneof('Type') == 'neuralNetworkClassifier':
  nn = spec.neuralNetworkClassifier
if spec.WhichOneof('Type') == 'neuralNetwork':
  nn = spec.neuralNetwork  
if spec.WhichOneof('Type') == 'neuralNetworkRegressor':
  nn = spec.neuralNetworkRegressor

preprocessing = nn.preprocessing[0].scaler   #out of range error
print( 'channel scale: ', preprocessing.channelScale)
print( 'blue bias: ', preprocessing.blueBias)
print( 'green bias: ', preprocessing.greenBias)
print( 'red bias: ', preprocessing.redBias)

我尝试了nn.ListFields,但是Python shell挂了;我猜数据太多了?

因此,不确定获取预处理信息的正确方法是什么。

谢谢。

0 个答案:

没有答案