I2S

HAIGEN

[Part 02: Image-to-Sketch Local Module]

---

Our model weights is avialable checkpoint

01 Data Preparation

For datasets that have paired sketch-image data, the path should be formatted as:

./data/rgb/  # training reference (image)
./data/skt/  # training ground truth (sketch)

Our Image-to-Sketch Generation dataset is available Clothes-V1.

02 Train and Test

1. Train

python main.py

2. Test

python evaluate.py

Note: you should change your checkpoint path.

3. Result