当前位置:   article > 正文

Python图片、视频风格转换(fast-style-transfer)教程_通过风格迁移让照片变成视频

通过风格迁移让照片变成视频

Github项目网址:https://github.com/lengstrom/fast-style-transfer

在Anaconda Prompt中输入以下命令,创建一个名为“tf-gpu”的新python环境,并激活切换到此环境下:

  1. conda create -n tf-gpu tensorflow-gpu=2.1.0
  2. conda activate tf-gpu

之后github项目中说需要安装jupyterlab再安装moviepy这个包或者直接使用命令“conda install moviepy”安装最新版本moviepy包。选择后者更为简便。

之后就可以使用git命令将项目clone到本地或者直接使用下载项目zip压缩包然后到本地解压即可:

然后下载网络上别人已经训练好的模型到自己指定位置。

 之后输入命令:

  1. python evaluate.py --checkpoint path/to/style/model.ckpt \
  2. --in-path dir/of/test/imgs/ \
  3. --out-path dir/for/results/
  4. 比如(依次为model地址,源图片所在地址,图片预计输出地址):
  5. python evaluate.py --checkpoint ./model/udnie.ckpt --in-path ./rawed/tang2.jpg --out-path ./finished/tang2_udnie.jpg

关于视频的转换是类似的,在转换前需要在环境下安装ffmpeg这个包。命令为:“conda install ffmpeg”。

这一步骤可能出现的bug:

  1. File "[mycondapath]\tf-gpu\lib\site-packages\moviepy\video\io\ffmpeg_writer.py", line 136, in write_frame
  2. self.proc.stdin.write(img_array.tobytes())
  3. BrokenPipeError: [Errno 32] Broken pipe
  4. During handling of the above exception, another exception occurred:
  5. Traceback (most recent call last):
  6. File "transform_video.py", line 55, in <module>
  7. main()
  8. File "transform_video.py", line 51, in main
  9. evaluate.ffwd_video(opts.in_path, opts.out, opts.checkpoint, opts.device, opts.batch_size)
  10. File "[myrepopath]\fast-style-transfer\evaluate.py", line 62, in ffwd_video
  11. style_and_write(frame_count)
  12. File "[myrepopath]\fast-style-transfer\evaluate.py", line 55, in style_and_write
  13. video_writer.write_frame(np.clip(_preds[i], 0, 255).astype(np.uint8))
  14. File "[mycondapath]\tf-gpu\lib\site-packages\moviepy\video\io\ffmpeg_writer.py", line 180, in write_frame
  15. raise IOError(error)
  16. OSError: [Errno 32] Broken pipe
  17. MoviePy error: FFMPEG encountered the following error while writing file out/guitar/fox.mp4:
  18. b''```

通过此方法可以解决:

其他可能出现的问题在项目的issue中可以自行浏览。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/530327
推荐阅读
相关标签
  

闽ICP备14008679号