path=$1
train=$2

cd $path/$train
# delete checkpoint 0
if [ -f "$path/$train/model.ckpt-0.meta" ]; then
    rm $path/$train/model.ckpt-0.meta
    rm $path/$train/model.ckpt-0.index
    rm $path/$train/model.ckpt-0.data-00000-of-00001
fi
# start exporting graphs
for i in model.ckpt-*.meta
do 
  SUBSTRING=$(echo $i| cut -d'.' -f 2)
  num=$(echo $SUBSTRING| cut -d'-' -f 2)
  echo $num
  python /media/local_data/saschroeder/catkin_ws/src/object_recognition/externals/models/research/object_detection/export_inference_graph.py --input_type image_tensor --pipeline_config_path $path/ssd.config --trained_checkpoint_prefix $path/$train/model.ckpt-$num --output_directory $path/$train/$num
  if [ -f "$path/$train/$num/frozen_inference_graph.pb" ]; then
    echo "successfully exported model $num, delete model"
    rm $path/$train/model.ckpt-$num.meta
    rm $path/$train/model.ckpt-$num.index
    rm $path/$train/model.ckpt-$num.data-00000-of-00001
  else
    echo "something went wrong while exporting graph $num"
  fi
done