#!/usr/bin/env sh
MY=/dev/watermark_improvement
TRAIN_DATA_ROOT="/"
VAL_DATA_ROOT="/"
echo "Create train lmdb.."
rm -rf $MY/img_train_lmdb
/opt/caffe/build/tools/convert_imageset \
--shuffle \
--resize_height=256 \
--resize_width=256 \
$TRAIN_DATA_ROOT \
$MY/watermark-JD/train.txt \
$MY/img_train_lmdb
echo "Create test lmdb.."
rm -rf $MY/img_val_lmdb
/opt/caffe/build/tools/convert_imageset \
--shuffle \
--resize_height=256 \
--resize_width=256 \
$VAL_DATA_ROOT \
$MY/watermark-JD/val.txt \
$MY/img_val_lmdb
echo "All Done.."