Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Unsupported Ops of type: SplitV

See original GitHub issue

Relevant information from pip freeze:

coremltools==3.0b3
tensorflow==1.12.0
tfcoreml==0.4.0b1

Script used:

import tfcoreml as tf_converter

tf_graph_path = '/wayfair/home/ns242e/od_133/tiny/tiny-v1/train-v1_best.pb'
coreml_graph_path = tf_graph_path.replace('.pb', '.mlmodel')

coreml_model = tf_converter.convert(
    tf_model_path=tf_graph_path,
    mlmodel_path=coreml_graph_path,
    output_feature_names=['output_boxes:0'],
    input_name_shape_dict={'inputs:0': [1, 416, 416, 3]},
)

Script output:

Loading the TF graph...
2019-07-25 14:03:50.019583: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 AVX512F
Graph Loaded.
Now finding ops in the TF graph that can be dropped for inference
Collecting all the 'Const' ops from the graph, by running it....
Done.
Traceback (most recent call last):
  File "/wayfair/home/ns242e/repos/od-yolo/translation/tensorflow_to_coreml.py", line 18, in <module>
    input_name_shape_dict={'inputs:0': [1, 416, 416, 3]},
  File "/wayfair/home/ns242e/repos/od-yolo/venv/lib/python3.6/site-packages/tfcoreml/_tf_coreml_converter.py", line 621, in convert
    custom_conversion_functions=custom_conversion_functions)
  File "/wayfair/home/ns242e/repos/od-yolo/venv/lib/python3.6/site-packages/tfcoreml/_tf_coreml_converter.py", line 294, in _convert_pb_to_mlmodel
    _check_unsupported_ops(OPS, output_feature_names, effectively_constant_ops + unused_ops)
  File "/wayfair/home/ns242e/repos/od-yolo/venv/lib/python3.6/site-packages/tfcoreml/_tf_coreml_converter.py", line 123, in _check_unsupported_ops
    ','.join(unsupported_op_types)))
NotImplementedError: Unsupported Ops of type: SplitV

You can find the frozen graph used here.

I understand that SplitV is not explicitly converted, but could the SplitV operation in the graph be converted to a Split operation, which is supported?