Course Project of CSE545 Advanced Computer Vision, UNIST
-
Download the dataset from this link. You can select and download each object.
-
mkdir datasets
andcd datasets
-
Move the downloaded dataset to this location.
-
tar -xvf [file_name].tar.xz
- first, install all necessary packages to run
pip install -r requirements.txt
- In
script
folder, you can see shell file. Also, you can run file usingtrain.sh
ortest.sh
.
sh train.sh
sh test.sh
- When excuting the shell file, modify the argument such as path.
optional arguments:
-h, --help show this help message and exit
--dataset {toothbrush,bottle,capsule}
--dataroot DATAROOT
--mode {train,test/defective,test/poke,test/squeeze,test/broken_large,test/broken_small,test/contamination,test/crack}
--seed SEED
--epochs EPOCHS
--batch_size BATCH_SIZE
--lr LR
--constant CONSTANT constant for loss
--weight_path WEIGHT_PATH
used for test
--threshold THRESHOLD
using at calculating difference
--save_root SAVE_ROOT
--memo MEMO make folder with value of parameter at
`result/[dataset]/img`
toothbrush/test/defective/010.png