Converting Caffe models to tensorflow 2.2 models in colab
Posting this quickly because I could not for the life of me figure out how to do it !mmconvert --srcFramework caffe --inputWeight /tmp/nst/resnext50.caffemodel --inputNetwork /tmp/nst/resnext50_deploy.prototxt --dstFramework tensorflow --outputModel /tmp/nst/out import tensorflow as tf imported = tf.saved_model.load('/tmp/nst/out') s = imported.signatures['serving_default'] print(s.inputs) # Tells you desired shape of inputs test_data = tf.zeros([1, 224, 224, 3]) s(tf.constant(test_data))