#!/bin/bash
echo"########################################################"
echo"Start Check whether the train compressed package exists#"
echo"########################################################"

if [! -f "ILSVRC2012_img_train.tar"];then
echo "ILSVRC2012_img_train.tar no exist"
exit
fi
echo"########################################################"
echo"              check tran.tar finish                     "
echo"########################################################"


echo"########################################################"
echo" Start Check whether the val compressed package exists  "
echo"########################################################"

echo"---------Start Check whether the val compressed package exists ----------"
if [! -f "ILSVRC2012_img_val.tar"];then
echo "ILSVRC2012_img_val.tar no exist"
exit
fi
echo"########################################################"
echo"              check val.tar finish                      "
echo"########################################################"


echo"########################################################"
echo"          Start Check train picture number              "
echo"########################################################"

if [-d "train"];then
train_number=$(ls -lR train/ |grep "^-" |wc -l)
echo "current train picture number : "$train_number
train_pic_number=1281167
	if test $[train_number] -eq $[train_pic_number]
	then
		echo "current train picture number no right"
		rm -rf ./train
fi
echo"########################################################"
echo"              check train picture number finish         "
echo"########################################################"

echo"########################################################"
echo"          Start Check val picture number              "
echo"########################################################"

if [-d "val"];then
val_number=$(ls -lR val/ |grep "^-" |wc -l)
echo "current val picture number : "$train_number
val_pic_number=1281167
	if test $[val_number] -eq $[val_pic_number]
	then
		echo "current val picture number no right"
		rm -rf ./val
fi
echo"########################################################"
echo"              check val picture number finish           "
echo"########################################################"


echo"########################################################"
echo"                  strat unzip the package               "
echo"########################################################"

mkdir -p train val bbox imagenet_tf

echo"########################################################"
echo"                  strat unzip train package             "
echo"########################################################"
tar -xvf ILSVRC2012_img_train.tar -C train/
echo"########################################################"
echo"                  unzip train package finish            "
echo"########################################################"
echo"                                                        "
echo"                                                        "
echo"########################################################"
echo"                  strat unzip val package               "
echo"########################################################"
tar -xvf ILSVRC2012_img_val.tar -C val/
echo"########################################################"
echo"                  unzip val package finish              "
echo"########################################################"
echo"                                                        "
echo"                                                        "
echo"########################################################"
echo"                  strat unzip bbox package              "
echo"########################################################"
tar -xvf ILSVRC2012_bbox_train_v2.tar.gz -C bbox/

echo"########################################################"
echo"                  unzip bbox package finish             "
echo"########################################################"
echo"                                                        "
echo"                                                        "
echo"########################################################"
echo"                 strat unzip train package  again       "
echo"########################################################"
cd train/
find . -name "*.tar" | while read LINE ; do mkdir -p "${LINE%.tar}"; tar -xvf "${LINE}" -C "${LINE%.tar}"; rm -f "${LINE}"; done
echo"########################################################"
echo"           unzip train package again finish             "
echo"########################################################"
echo"                                                        "
echo"                                                        "
echo"########################################################"
echo"             strat label val                            "
echo"########################################################"
cd ..
tar_train_number=$(ls -lR train/ |grep "^-" |wc -l)
echo "new train picture number"$tar_train_number
python3 preprocess_imagenet_validation_data.py val/ imagenet_2012_validation_synset_labels.tx

echo"########################################################"
echo"                    label val  finish                   "
echo"########################################################"
echo"########################################################"
echo"                                                        "
echo"                                                        "
echo"########################################################"
python3 build_imagenet_data.py --output_directory=imagenet_tf/ --train_directory=train/ --validation_directory=val/