使用tf-coreml将Tensorflow转换为coreML时出错

时间:2019-03-14 16:54:38

标签: python tensorflow coreml

我想用coreML进行手部跟踪。

为此,我采用了现成的tensorFlow模型:https://github.com/timctho/convolutional-pose-machines-tensorflow

我正在尝试使用tf-coreml将其转换为coreML。但是在转换过程中出现错误。

(base) MBP-Alexey:Desktop alexeygorbunov$ python convert_tf_to_coreml.py

Loading the TF graph...
Traceback (most recent call last):   
File "convert_tf_to_coreml.py", line 5, in <module>
    input_name_shape_dict  = { ' input_placeholder: 0 ' : [ 1 , 256 , 256 , 3 ]})   
File "/Users/alexeygorbunov/anaconda3/lib/python3.7/site packages/tfcoreml/_tf_coreml_converter.py", line 586, in convert
    custom_conversion_functions=custom_conversion_functions)   
File "/Users/alexeygorbunov/anaconda3/lib/python3.7/site-packages/tfcoreml/_tf_coreml_converter.py", line 150, in _convert_pb_to_mlmodel
 gdef.ParseFromString(serialized)  
google.protobuf.message.DecodeError: Error parsing message

可能是什么错误以及如何解决?

0 个答案:

没有答案