简体   繁体   中英

Tensorflow .pb file to coreml model: 'Unsupported Ops of type: AddV2'

I am trying to convert my tensorflow model(.pb file) obtained by retraining the mobilenet architecture to coreml model. Initially I had tensorflow 1.14.0 everything worked perfectly. But when I upgraded the tensorflow to 2.0.0 and retrained the model, because of deprecations python is adding 'AddV2' operation instead of 'Add'. Because of this I am getting an error during convertion as

Unsupported Ops of type: AddV2

What can I do about this? Should I have to change the operation or tweak conversion code to support the operation.

Although the tfcoreml==1.1.0 says it has added AddV2 operation to support tf 2.0.0 it has not.

https://github.com/tf-coreml/tf-coreml/releases/tag/1.1.0

Proof:

https://github.com/tf-coreml/tf-coreml/blob/master/tfcoreml/_ops_to_layers.py

This can be solved by adding AddV2 which does elementwise operation in the above script.

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM