# stegface

**Repository Path**: sheenashea/stegface

## Basic Information

- **Project Name**: stegface
- **Description**: 2024-2025 神经网络大作业，利用stable diffusion完成人脸图片的隐写任务
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-01-12
- **Last Updated**: 2025-01-14

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

## 基于 stable diffusion 实现面部图像隐写任务
该项目是天津大学2024-2025秋季神经网络与机器学习期末大作业。

### 项目环境
1. autodl 云服务器平台
2. 镜像:  
PyTorch  2.5.1  
Python  3.12(ubuntu22.04)  
Cuda  12.4  
3. GPU:  
RTX 4090(24GB) * 1  
4. CPU:  
16 vCPU Intel(R) Xeon(R) Platinum 8352V CPU @ 2.10GHz  
5. 内存: 120GB  
硬盘-系统盘: 30 GB  

### 项目结构
```
.
├── Readme.md
├── data                           # 100 张高清人脸图像
├── image_metrics                  # 评估指标
│   ├── README.md
│   ├── cal_lpips.py               # lpips
│   ├── fid.py                     # fid
│   ├── inception.py
│   ├── inception_score_torch.py
│   ├── niqe.py                    # niqe
│   ├── pre-train-models
│   ├── psnr_ssim.py               # psnr + ssim
│   └── utils.py
├── main.py                        # 入口文件
├── tuning.py                      # 参数调优
├── output                         # 输出文件夹
├── prompt.csv                     # 对应 prompt
├── utils                          # 辅助作图
├── run.sh                         # 执行脚本
└── src
    ├── ParamsTuning.py            # 模型参数调优
    ├── SchedulerTuning.py         # 调度器调优
    └── steganography.py           # 隐写类实现
```

### 运行项目

实现隐写任务
```bash
pip install -r requirements.txt
bash run.sh
```
进行生成图像质量评估
```bash
cd image_metrics
python xxx.py
```
参数调优(需要在代码中修改调优参数)
```bash
python tuning.py
```