export PROJECT_DIR=<YOUR PROJECT DIRECTORY>
export MODEL_DIR=$PROJECT_DIR/models/<YOUR SSD MODEL>

export TF_API_DIR=$PROJECT_DIR/tf-models/research
export MODEL_CONFIG=$MODEL_DIR/pipeline.config
export CHECKPOINT_PATH=$MODEL_DIR/model.ckpt-<CHECKPOINT NUMBER>
export OUTPUT_DIR=$MODEL_DIR/tmp/tflite

# Run the export_tflite_ssd_graph.py script to get the frozen graph
# Make sure PYTHONPATH is updated
cd $TF_API_DIR
python object_detection/export_tflite_ssd_graph.py \
--pipeline_config_path=$MODEL_CONFIG \
--trained_checkpoint_prefix=$CHECKPOINT_PATH \
--output_directory=$OUTPUT_DIR \
--add_postprocessing_op=true