我正在尝试获取有关我拥有的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挂了;我猜数据太多了?
因此,不确定获取预处理信息的正确方法是什么。
谢谢。