First, we need to get the dataset first.
Use the following command we build for downloading dataset.
bash ./get_dataset.sh
The shell script will automatically download the dataset and store it as name data
.
Simply type the following command.
python train_improved_DANN.py
There are four different case for training, if you want to switch other training case, you can alter the following lines in the code:
source1 = 'quickdraw'
source2 = 'infograph'
source3 = 'real'
target = 'sketch'
To predict the target image, type the following command that will produce pred.csv
file
python test.py
To evaluate the accuracy, type the following command thae will check the accuracy based on the previous output pred.csv
python check_accuracy.py
Below is a list of packages we used to implement this project:
CUDA
: 10.1
python
: 3.6.9torch
: 1.4.0
numpy
: 1.18.2
pandas
: 0.25.1PIL
: 6.1.0torchvision
: 0.4.0cv2
,matplotlib
The Python Standard Librarytqdm