hmrishavbandy / SketchINR-Code

SketchINR: A First Look into Sketches as Implicit Neural Representations [CVPR 2024]

Before training, change the root directory here

SketchINR-Code/single_fit.py

Line 248 in 707bf1f

root = "/vol/research/hmriCV2/Datasets/"

to your dataset folder. The folder (for QuickDraw) should look like
```
.
└── Datasets
    └── QuickDraw
         ├── QuickDraw_Keys.pickle
         ├── QuickDraw_TestData/
         ├── QuickDraw_TrainData/
         └── QuickDraw_ValidData/
```
Download dataset from https://huggingface.co/datasets/Hmrishav/QuickDraw
Create conda environment: conda env create --name sketchinr --file=environments.yml
Training: python3 single_fit.py