pipenv --python 3.6  # setup pipenv with python 3.6
pipenv install requests
pipenv install tensorflow
pipenv install deepomatic-api

export DEEPOMATIC_APP_ID=xxxxxxxxxxxx
export DEEPOMATIC_API_KEY=xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
pipenv run python

export DEEPOMATIC_VERSION_ID=xxxxx

import os
import json
from deepomatic.api.client import Client
​
def pretty_save_json_to_file(json_data, json_path):
    """Helper function to save json to file in readable fashion."""
    try:
        with open(json_path, 'w') as json_file:
            json.dump(json_data, json_file, indent=4, sort_keys=True)
    except:
        print(f"Could not save file {json_path} in json format.")
        raise
​
# Intialize client
app_id = os.getenv('DEEPOMATIC_APP_ID')
api_key = os.getenv('DEEPOMATIC_API_KEY')
client = Client(app_id, api_key)
​
# Retrieve the recognition version
version_id = os.getenv('DEEPOMATIC_VERSION_ID')
version = client.RecognitionVersion.retrieve(version_id)
version_data = version.data()
postprocess_file = 'postprocess.json'
pretty_save_json_to_file(version_data['post_processings'], postprocess_file)
print(f"Recognition Version number {version_id} saved to {postprocess_file}")
​
# Retrieve the network
network_id = version_data['network_id']
network = client.Network.retrieve(network_id)
network_data = network.data()
preprocess_file = 'preprocess.json'
pretty_save_json_to_file(network_data['preprocessing'], preprocess_file)
print(f"Network number {network_id} saved to {preprocess_file}")
​
# Retrieve the recognition specification
spec_id = version_data['spec_id']
spec = client.RecognitionSpec.retrieve(spec_id)
spec_data = spec.data()
outputs_file = 'outputs.json'
pretty_save_json_to_file(spec_data['outputs'], outputs_file)
print(f"Recognition Specification {spec_id} saved to {outputs_file}")

import zipfile
import requests
​
# Download the raw model
model_url = f"https://api.deepomatic.com/v0.7/networks/{network_id}/download"
print("Starting network download...")
r = requests.get(model_url, headers={'X-APP-ID': app_id, 'X-API-KEY': api_key})
model_file = 'network.zip'
with open(model_file, 'wb') as f:
    f.write(r.content)
print(f"Raw model saved to {model_file}")
​
# Extract network file
zip_ref = zipfile.ZipFile(model_file, 'r')
zip_ref.extractall('network')
zip_ref.close()

​
import tensorflow as tf
from tensorflow.python.platform import gfile
from tensorflow.core.protobuf import saved_model_pb2
from tensorflow.python.util import compat
​
# Open tensorflow model and list all nodes
model_filename = 'network/saved_model.pb'
nodes_file = 'nodes.json'
with tf.Session() as sess:
    with gfile.FastGFile(model_filename, 'rb') as f:
        # Convert pb file to tensorflow graph
        data = compat.as_bytes(f.read())
        sm = saved_model_pb2.SavedModel()
        sm.ParseFromString(data)
        g_in = tf.import_graph_def(sm.meta_graphs[0].graph_def)
​
        # Print all nodes in graph
        nodes = [n.name for n in tf.get_default_graph().as_graph_def().node]
        pretty_save_json_to_file(nodes, nodes_file)
        print(f"Models Nodes saved to {nodes_file}")
​