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 \
```
近期下载者:
相关文件:
收藏者: