GeneFace Plus Plus 对硬盘有要求,且建议优先使用固态硬盘(SSD)。
该项目涉及大量视频、音频数据的预处理以及模型权重的存储,对硬盘的读写速度和容量都有一定要求。以下是具体的配置建议:
GeneFace Plus Plus 的流程包含多个阶段(数据预处理、训练、渲染),会生成大量中间文件。建议预留 100GB - 200GB 的可用空间。
以下是各阶段大致的空间占用参考:
| 阶段 | 内容 | 预估占用空间 |
|---|---|---|
| 原始数据 | 训练用的视频(几分钟到几十分钟) | 1GB - 5GB |
| 预处理数据 | 提取的人脸图像、音频特征、深度信息 | 30GB - 50GB (取决于视频长度和分辨率) |
| 模型权重 | 训练好的 NeRF 模型、音频编码器 | 5GB - 10GB |
| 结果输出 | 生成的视频文件 | 1GB - 5GB |
| 环境依赖 | Conda 环境、PyTorch、CUDA 等 | 10GB - 20GB |
虽然你问的是硬盘,但为了保证 GeneFace Plus Plus 能跑起来,还需要注意以下硬件配置:
| 硬件 | 建议配置 | 备注 |
|---|---|---|
| 显卡 (GPU) | NVIDIA GPU (显存 ≥ 12GB) | 推荐 3090/4090。显存不足会导致训练直接 OOM (显存溢出)。 |
| 内存 (RAM) | ≥ 32GB | 数据预处理阶段(特别是提取 3DMM 参数)非常吃内存。 |
| CPU | 多核 CPU | 用于数据预处理和视频编解码。 |