search

onnx / onnxmltools

ONNXMLTools enables conversion of models to ONNX
https://onnx.ai
Apache License 2.0
992 stars 180 forks source link

caffe_coreml_onnx script #389

Open lucasncabral opened 4 years ago

lucasncabral commented 4 years ago

I am trying to convert a caffe model to tensorflow, but I found the following problems in the script:

  1. The conversion method is assigned a variable not previously defined. To solve this, I believe that the "caffe_model" variable is the path to my .caffemodel (defined as input_caffe_path);

Change: coreml_model = coremltools.converters.caffe.convert((caffe_model, proto_file)) to: coreml_model = coremltools.converters.caffe.convert((input_caffe_path, proto_file))

  1. After making this modification, I come across the following error:

Captura de tela de 2019-10-23 10-48-58

vinitra-zz commented 4 years ago

Thank you for your feedback. The coreml-converter script has been fixed. It looks like this is a bug in the converter, so I've transferred it to onnxmltools. Can you provide the model for debugging?