--- a +++ b/scripts/convert_to_pb.sh @@ -0,0 +1,19 @@ +[ -a ./build_image_data.py ] || \ + curl -O https://raw.githubusercontent.com/tensorflow/models/f87a58cd96d45de73c9a8330a06b2ab56749a7fa/research/inception/inception/data/build_image_data.py + +DATADIR=../data/data_1024/infl_split + +echo "normal" > $DATADIR/labels.txt +echo "infl" >> $DATADIR/labels.txt + +rm -rf $DATADIR/protobuf +mkdir -p $DATADIR/protobuf + +python3 build_image_data.py \ + --train_directory $DATADIR/train \ + --validation_directory $DATADIR/val \ + --output_directory $DATADIR/protobuf \ + --train_shards 1 \ + --validation_shards 1 \ + --num_threads 1 \ + --labels_file $DATADIR/labels.txt