WCT2 项目使用与启动教程

WCT2 项目使用与启动教程

WCT2 Software that can perform photorealistic style transfer without the need of any post-processing steps. 项目地址: https://gitcode.com/gh_mirrors/wc/WCT2

1. 项目目录结构及介绍

WCT2 项目是一个基于 PyTorch 的图像风格转换开源项目。项目目录结构如下：

WCT2/
├── examples/           # 示例图片和标签文件夹
│   ├── content/        # 内容图片
│   ├── style/          # 风格图片
│   ├── content_segment # 内容图片的标签图
│   └── style_segment   # 风格图片的标签图
├── model_checkpoints/  # 预训练模型存储目录
├── utils/             # 工具函数和类
├── LICENSE.md          # 项目许可证信息
├── README.md           # 项目说明文件
├── requirements.txt    # 项目依赖列表
├── transfer.py         # 图像风格转换主程序
└── model.py            # 模型定义文件

• examples/：包含用于演示和测试的图像数据，包括内容图片、风格图片以及对应的标签图。
• model_checkpoints/：存储预训练模型权重文件的目录。
• utils/：包含项目所需的辅助函数和类。
• LICENSE.md：项目的开源许可证。
• README.md：提供项目描述、安装步骤、使用方法和依赖信息。
• requirements.txt：列出项目运行所需的 Python 包。
• transfer.py：执行图像风格转换的脚本。
• model.py：定义用于风格转换的模型架构。

2. 项目的启动文件介绍

项目的启动文件是 transfer.py。该脚本负责加载模型、处理输入图像以及执行风格转换过程。以下是一些基本的命令行参数：

• --content：指定内容图片的文件夹路径。
• --style：指定风格图片的文件夹路径。
• --content_segment：指定内容图片标签图的文件夹路径（如果有）。
• --style_segment：指定风格图片标签图的文件夹路径（如果有）。
• --output：指定输出图片的文件夹路径。
• --image_size：指定输出图片的大小。
• --alpha：控制内容与风格特征的混合比例。
• --option_unpool：选择不同的模型版本（sum 或 cat5）。
• --transfer_at_encoder、--transfer_at_decoder、--transfer_at_skip：指定在模型的哪个部分进行风格转换。
• --transfer_all：在所有可能的组合中进行风格转换。
• --cpu：在 CPU 上运行模型（如果 GPU 不可用）。
• --verbose：打印详细的运行信息。

示例命令：

python transfer.py --option_unpool cat5 -a --content ./examples/content --style ./examples/style --content_segment ./examples/content_segment --style_segment ./examples/style_segment/ --output ./outputs/ --verbose --image_size 512

3. 项目的配置文件介绍

该项目的主要配置是通过命令行参数来完成的，并没有一个单独的配置文件。所有的参数都可以在 transfer.py 脚本中通过命令行传递。如果有需要，可以通过编写一个额外的配置脚本或使用 JSON、YAML 等格式的配置文件来简化参数的传递和管理。不过，目前项目默认的方式是直接通过命令行进行配置。

WCT2 Software that can perform photorealistic style transfer without the need of any post-processing steps. 项目地址: https://gitcode.com/gh_mirrors/wc/WCT2