Skip to content

Latest commit

 

History

History
10 lines (10 loc) · 358 Bytes

README.md

File metadata and controls

10 lines (10 loc) · 358 Bytes

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

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

环境要求

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