Fast-Neural-Style-Transfer-master

所属分类:其他
开发工具:Python
文件大小:12371KB
下载次数:2
上传日期:2020-05-21 16:01:24
上 传 者sadman2233
说明:  图像风格转换,python 深度学习基础项目
(Image style conversion)

文件列表:
LICENSE (1075, 2019-04-13)
assets (0, 2019-04-13)
assets\celeba_mosaic.gif (7652551, 2019-04-13)
assets\stylized-celtics.gif (3313321, 2019-04-13)
assets\zurich.jpg (1023105, 2019-04-13)
images (0, 2019-04-13)
images\content (0, 2019-04-13)
images\content\zurich.jpeg (334870, 2019-04-13)
images\styles (0, 2019-04-13)
images\styles\cuphead.jpg (122757, 2019-04-13)
images\styles\mosaic.jpg (77378, 2019-04-13)
images\styles\starry_night.jpg (315236, 2019-04-13)
models.py (3341, 2019-04-13)
test_on_image.py (1235, 2019-04-13)
test_on_video.py (1531, 2019-04-13)
train.py (5766, 2019-04-13)
utils.py (1756, 2019-04-13)

# Fast Neural Style Transfer in PyTorch

PyTorch implementation of [Fast Neural Style Transfer](https://cs.stanford.edu/people/jcjohns/eccv16/) ([official Lua implementation](https://github.com/jcjohnson/fast-neural-style)). I have made some trained models available [here](https://drive.google.com/drive/folders/1aRD6zakhcDImN2Y54qAT6f4801iLcCLB?usp=sharing). ## Train ``` python3 train.py --dataset_path \ --style_image \ --epochs 1 \ --batch_size 4 \ --image_size 256 ```

Figure: Training progress over the first 10,000 batches.

## Test on Video ``` python3 test_on_video.py --video_path \ --checkpoint_model \ ```

## Test on Image ``` python3 test_on_image.py --image_path \ --checkpoint_model \ ```

近期下载者

相关文件


收藏者