pytorch-style-transfer-fast
usage
- train
- run
python3 train.py
to train the model and generate the model parameters for TransformerNet. - the generated binary file will be save at
binaries/
- test with camera
- run
python3 test_with_camera.py
to test the style transfer algorithm. The stylized image will be presented every 5 seconds.