Skip to content

gmh1627/Pytorch-style-transfer

Repository files navigation

运用pytorch和深度学习实现图像风格迁移

运行main.py训练模型,运行stylize.py进行风格迁移。

环境要求

  • 操作系统:Windows、macOS 或 Linux
  • Python 版本:Python 3.6 或更高版本
  • 依赖库:
    • NumPy
    • tqdm
    • PIL
    • torch(PyTorch)、torchvision(根据cuda版本下载,cuda12.2能兼容cuda12.1)

About

运用pytorch和深度学习实现图像风格迁移

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages