-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathbuild.sh
executable file
·28 lines (28 loc) · 2.09 KB
/
build.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
./cleanup.sh
echo "======================================================================================================="
echo " Preparing Dataset"
echo "======================================================================================================="
python model/src/data_preparation.py \
--original-dataset-dir /home/nammathalle/work/retail_product_recognition_system/model/original_dataset \
--new_dataset-dir /home/nammathalle/work/retail_product_recognition_system/model/dataset \
--model-data-dir /home/nammathalle/work/retail_product_recognition_system/model/model_data \
--templates-dir /home/nammathalle/work/retail_product_recognition_system/model/templates
echo "======================================================================================================="
echo ""
echo ""
echo "======================================================================================================="
echo " Training the ResNet-50 model"
echo "======================================================================================================="
python model/src/ResNet-PyTorch/train.py \
--dataset-dir /home/nammathalle/work/retail_product_recognition_system/model/dataset \
--model-data-dir /home/nammathalle/work/retail_product_recognition_system/model/model_data
echo "======================================================================================================="
echo ""
echo ""
echo "======================================================================================================="
echo " Evaluating the trained ResNet-50 model"
echo "======================================================================================================="
python model/src/ResNet-PyTorch/test.py \
--dataset-dir /home/nammathalle/work/retail_product_recognition_system/model/dataset \
--model-data-dir /home/nammathalle/work/retail_product_recognition_system/model/model_data
echo "======================================================================================================="