Sketch Recognition on TU Berlin Dataset with pre-trained ConvNets and Transfer Learning
Download the weights for the model from this link and place the file in the root directory.
Install necessary requirements such as PyTorch, pandas, etc.
pip install -r requirements.txt
Run test.py
python test.py <src> <dst> <--ckpt>
- For
src
, provide the path to the folder containing image files - For
dst
, provide the name of the.csv
file to be created - (optional) For
--ckpt
, provide the name of the downloaded.ckpt
file (default is"./weights.pt"
)
Results are generated in the following format
image_name,id