# img_retouch

**Repository Path**: ferry_zhou/img_retouch

## Basic Information

- **Project Name**: img_retouch
- **Description**: 人像精修,ps.AI
用开源算法实现影楼级人像精修，探索「AI 修图接单」的可行性。当前以传统图像处理为主（OpenCV + PIL），MediaPipe 用于瘦脸；GFPGAN、Real-ESRGAN 等深度学习模型已预留接口，待 GPU 环境就绪后接入。
- **Primary Language**: Unknown
- **License**: MulanPSL-2.0
- **Default Branch**: ferry
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2026-06-26
- **Last Updated**: 2026-07-08

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README


# img_retouch — AI 人像精修

用开源算法实现影楼级人像精修，探索「AI 修图接单」的可行性。当前已集成稳定的修图工作流（含瘦脸、磨皮、调色），以及 AI 模型（背景去除），并支持完整算法集成（人脸增强、高清放大等）。

## 快速开始

### 使用推荐工作流（最稳定）

```bash
# 1. 激活 Python 3.9 环境
conda activate img_retouchPY39

# 2. 运行修复版修图脚本
python 修复版修图.py
```

输出位置：`output_fixed/`，推荐查看 `06_final_natural_studio.jpg`。

---

### 完整 AI 集成修图（含所有算法检测）

```bash
# 使用全算法集成脚本
conda activate img_retouchPY39
python 所有AI算法集成.py
```

输出位置：`output_all_ai/`，包含算法检测和可用的 AI 功能。

---

## 项目结构

```
img_retouch/
├── input/                      # 输入原片
├── output/                     # 基础修图输出
├── output_studio/              # 影楼精修输出
├── output_fixed/               # 修复版修图输出（推荐）
├── output_complete_ai/         # 完整AI修图输出
├── output_all_ai/              # 全算法集成输出
├── 所有AI算法集成.py          # ★ 新：全算法集成 + 检测
├── 修复版修图.py              # ★ 推荐：自然瘦脸 + 温和精修
├── 完整AI修图.py              # 完整工作流+AI背景去除
├── studio_retouch_final.py     # 影楼精修稳定版
├── studio_retouch_simple.py    # 简化稳定版
├── retouch_portrait.py         # 基础修图
├── check_env.py               # 环境检测
├── README_AI_models.md         # AI 模型调研报告
├── 开源算法总结.md            # 算法集成状态与环境问题
└── AI对话.md                  # 需求与迭代记录
```

---

## 脚本说明

| 脚本 | 输出目录 | 说明 |
|------|----------|------|
| `所有AI算法集成.py` | `output_all_ai/` | **新：全算法集成** + 自动检测可用库 |
| `修复版修图.py` | `output_fixed/` | **推荐**：自然瘦脸 + 温和精修 |
| `完整AI修图.py` | `output_complete_ai/` | 完整工作流 + AI 背景去除 |
| `studio_retouch_final.py` | `output_studio/` | 完整影楼流程 |

---

## 修图流程

### 全算法集成流程（`所有AI算法集成.py`）

1. **自动检测 AI 库**：PyTorch、GFPGAN、Real-ESRGAN、rembg、MediaPipe
2. 加载原图
3. **自然瘦脸**
4. **Real-ESRGAN 高清放大**（如可用）
5. **GFPGAN 人脸增强**（如可用）
6. 传统精修（白平衡、磨皮、去油光）
7. **rembg AI 背景去除**（如可用）
8. 最终影楼风格输出

### 推荐流程（`修复版修图.py`）

1. 加载原图
2. **自然瘦脸**（强度可调整）
3. **白平衡** + **自然色调曲线**
4. **肤色检测** + **温和磨皮**
5. **高光抑制**（去油光）
6. **智能锐化**（皮肤保护）
7. 自然影楼风格输出

---

## 使用的开源算法

| 算法 | 状态 | 用途 |
|------|------|------|
| 优化局部形变 | ✅ 已用 | **瘦脸**（自然强度，人脸中心定位） |
| 自动白平衡 | ✅ 已用 | 色彩校正 |
| 肤色检测 | ✅ 已用 | RGB 阈值 + 人脸区域 |
| 多层磨皮 | ✅ 已用 | 高斯 + 中值滤波混合，保留质感 |
| 高光抑制 | ✅ 已用 | 去油光 |
| 智能锐化 | ✅ 已用 | 皮肤区域保护不锐化 |
| **rembg (U2-Net)** | ✅ 已用 | **AI 背景去除** |
| 色调曲线 + 暗角 | ✅ 已用 | 影楼风格调色 |
| **GFPGAN (腾讯)** | ⏳ 待完善 | 人脸增强（需 PyTorch ≤ 2.0） |
| **Real-ESRGAN** | ⏳ 待完善 | 高清放大（需 PyTorch ≤ 2.0） |
| **LaMa / LaMa-cleaner** | ⏳ 待安装 | 瑕疵修补 / 智能填充 |
| **MediaPipe Face Mesh** | ✅ 已安装 | 人脸关键点（API 兼容性优化中） |
| **CodeFormer** | ⏳ 待集成 | 自然人脸修复 |
| **MODNet** | ⏳ 待集成 | 人像分割 / 抠图 |

---

## 环境配置

### 推荐环境（已配置：`img_retouchPY39`）

- Python 3.9
- NumPy 1.26.4（<2.0，兼容性好）
- PyTorch 2.7.1+cu118
- rembg 2.0.61
- OpenCV 4.9.0
- Pillow 11.3.0
- MediaPipe 0.10.35

### 完整环境配置（含所有算法）

```bash
# 创建并激活环境
conda create -n img_retouchPY39 python=3.9 -y
conda activate img_retouchPY39

# 安装基础依赖
pip install numpy<2.0 pillow opencv-python tqdm pyyaml

# 安装 PyTorch 2.0（兼容 GFPGAN/Real-ESRGAN）
pip install torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu118

# 安装 AI 修图库
pip install gfpgan realesrgan basicsr facexlib rembg mediapipe onnxruntime

# 安装 LaMa 瑕疵修补
pip install lama-cleaner
```

---

## 路线图

- [x] 传统影楼精修工作流
- [x] 自然瘦脸算法
- [x] AI 背景去除（rembg）
- [x] 全算法集成框架（自动检测）
- [ ] GFPGAN / CodeFormer 人脸修复
- [ ] Real-ESRGAN 超分辨率
- [ ] LaMa 瑕疵修补
- [ ] Gradio Web 界面

---

## 相关文档

- [README_AI_models.md](README_AI_models.md) — 开源模型调研报告
- [开源算法总结.md](开源算法总结.md) — 环境与兼容性说明
- [AI对话.md](AI对话.md) — 项目迭代记录