conda env create -f conda.yamlconda activate sketchdetectionpoetry installapt install p7zip-full- If poetry stalls at any point, try
poetry config keyring.enabled false
- Download the Sketchy dataset and put it in the
datadirectory 7z x data/rendered_256x256.7z -o./data/- (Optional) cleanup:
rm data/rendered_256x256.7zrm -rf data/256x256/photo
python train.py --config configs/sample-config.json --checkpoint checkpoints/resnet152-1e-3.pth
python gui.py --port 60001 --architecture ResNet152 --checkpoint checkpoints/resnet152-1e-3.pth