#!/bin/bash
FROZEN_GRAPH=$1
WORKING_DIR=$2
mkdir -p $WORKING_DIR
UFF_FILE=$WORKING_DIR/00000000.uff

bazel run //vision/inference/tensor_rt:tensorflow_to_uff -- \
    --frozen_graph=$FROZEN_GRAPH --output_uff=$UFF_FILE \
    --output_layers=tower_0/md/depth_value_output,tower_0/md/depth_residual_output,tower_0/md/depth_valid_output --input_layers=tower_0/input/common/im


