Sketch Recognition

Sketch Recognition on TU Berlin Dataset with pre-trained ConvNets and Transfer Learning

Usage

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