PhotoMaker 项目使用教程

PhotoMaker 项目使用教程

PhotoMaker 项目地址: https://gitcode.com/gh_mirrors/ph/PhotoMaker

1. 项目目录结构及介绍

PhotoMaker/
├── examples/
│   └── ...
├── feature-extractor/
│   └── ...
├── gradio_demo/
│   └── ...
├── inference_scripts/
│   └── ...
├── photomaker/
│   └── ...
├── .dockerignore
├── .gitignore
├── LICENSE
├── MacGPUEnv.md
├── README.md
├── README_pmv2.md
├── cog.yaml
├── photomaker_demo.ipynb
├── photomaker_style_demo.ipynb
├── predict.py
├── pyproject.toml
└── requirements.txt

目录结构说明

• examples/: 包含项目的示例文件，用于展示如何使用PhotoMaker生成图像。
• feature-extractor/: 包含特征提取器的相关代码。
• gradio_demo/: 包含Gradio演示的代码，用于启动本地演示界面。
• inference_scripts/: 包含推理脚本的代码，用于生成图像。
• photomaker/: 包含PhotoMaker的核心代码。
• .dockerignore: Docker构建时忽略的文件列表。
• .gitignore: Git版本控制时忽略的文件列表。
• LICENSE: 项目的开源许可证文件。
• MacGPUEnv.md: 在Mac上使用GPU的配置说明。
• README.md: 项目的介绍和使用说明。
• README_pmv2.md: PhotoMaker V2的介绍和使用说明。
• cog.yaml: Cog配置文件，用于定义项目的依赖和环境。
• photomaker_demo.ipynb: PhotoMaker的Jupyter Notebook演示文件。
• photomaker_style_demo.ipynb: PhotoMaker风格化生成的Jupyter Notebook演示文件。
• predict.py: 用于预测和生成图像的Python脚本。
• pyproject.toml: Python项目的配置文件。
• requirements.txt: 项目依赖的Python包列表。

2. 项目启动文件介绍

gradio_demo/app.py

该文件是启动Gradio演示界面的入口文件。通过运行该文件，可以在本地启动一个Web界面，用于交互式生成图像。

python gradio_demo/app.py

photomaker_demo.ipynb

这是一个Jupyter Notebook文件，提供了PhotoMaker的基本使用示例。通过运行该Notebook，可以了解如何加载模型、生成图像以及调整参数。

photomaker_style_demo.ipynb

这是另一个Jupyter Notebook文件，专注于PhotoMaker的风格化生成。通过运行该Notebook，可以了解如何使用不同的风格化模型生成图像。

3. 项目的配置文件介绍

pyproject.toml

该文件是Python项目的配置文件，定义了项目的元数据、依赖项和其他配置。

[project]
name = "PhotoMaker"
version = "0.1.0"
description = "Customizing Realistic Human Photos via Stacked ID Embedding"
authors = [
    { name="TencentARC", email="[email protected]" }
]
dependencies = [
    "torch>=2.0.0",
    "diffusers",
    "gradio",
    "huggingface_hub"
]

requirements.txt

该文件列出了项目运行所需的Python包及其版本。

torch>=2.0.0
diffusers
gradio
huggingface_hub

cog.yaml

Cog配置文件，用于定义项目的依赖和环境。

build:
  - python=3.10
  - pip install -r requirements.txt

通过以上配置文件，可以确保项目在不同的环境中正确安装和运行所需的依赖项。

PhotoMaker 项目地址: https://gitcode.com/gh_mirrors/ph/PhotoMaker