Neural Strokes: Stylized Line Drawing of 3D Shapes

This repository contains the PyTorch implementation for ICCV 2021 Paper "Neural Strokes: Stylized Line Drawing of 3D Shapes" by Difan Liu, Matthew Fisher, Aaron Hertzmann, Evangelos Kalogerakis.

Dependency

The project is developed on Ubuntu 16.04 with cuda10.0 + cudnn7.6. The code has been tested with PyTorch 1.2.0 (GPU version) and Python 3.7.6.
Python packages:
- diffvg
- OpenCV (tested with 4.3.0)
- PyYAML (tested with 5.3.1)
- scikit-image (tested with 0.17.2)
- scipy (tested with 1.4.1)

Dataset and Weights

Pre-trained model is available here, please put it in weights:
```
cd weights
unzip weights.zip
```
Dataset is available here, please put it in datasets:
```
cd datasets
unzip datasets.zip
```

Preprocessing

Preprocess raw data:
```
python preprocess.py -s datasets/style_01
```
-s: path to the data of a single style.

Test

Test:
```
python test.py -d datasets/style_01/test_2 -g weights/style_01/SG.pth -t weights/style_01/ST.pth -s results/style_01__test_2.png
```
-d: path to testing data.
-g: path to Stroke Geometry checkpoint.
-t: path to Stroke Texture checkpoint.
-s: path to save the synthesized image.

Training

Start the Stroke Geometry (SG) training:

python train_SG.py -d datasets/style_01/train -n style_01__SG

After the Stroke Geometry (SG) training is finished, start the Stroke Texture (ST) training:
```
python train_ST.py -d datasets/style_01/train -n style_01__ST
```
-d: path to training data.
-n: name of the experiment.

Cite:

@InProceedings{Liu_2021_ICCV,
author={Liu, Difan and Fisher, Matthew and Hertzmann, Aaron and Kalogerakis, Evangelos},
title={Neural Strokes: Stylized Line Drawing of 3D Shapes},
booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
year = {2021}
}

Contact

To ask questions, please email.

DifanLiu / NeuralStrokes