diff --git a/README.md b/README.md index da2af5b..84edc1e 100644 --- a/README.md +++ b/README.md @@ -4,6 +4,41 @@ 本仓库包含了 Gaussian Haircut 的官方实现,这是一种基于股线的人体头发重建方法,用于单目视频。 +## 预览 + +重建过程包括以下主要阶段: + +1. **预处理阶段** + - 视频帧提取和整理 + - COLMAP相机重建 + - 头发和身体分割 + - 图像质量评估和筛选 + - 方向图计算 + - 人脸关键点检测 + - FLAME头部模型拟合 + +2. **重建阶段** + - 3D高斯体重建 + - FLAME网格拟合 + - 场景裁剪和优化 + - 头发股线重建 + +3. **可视化阶段** + - 导出重建的股线 + - Blender渲染可视化 + - 生成结果视频 + +预期输出: +``` +[your_scene_folder]/ +├── raw.mp4 # 输入视频 +├── 3d_gaussian_splatting/ # 3D高斯体重建结果 +├── flame_fitting/ # FLAME头部模型拟合结果 +├── strands_reconstruction/ # 头发股线重建中间结果 +├── curves_reconstruction/ # 最终头发股线结果 +└── visualization/ # 渲染结果和视频 +``` + ## 入门指南 ### Linux 平台 diff --git a/install.bat b/install.bat index fc6bbef..20a5d33 100644 --- a/install.bat +++ b/install.bat @@ -58,13 +58,6 @@ IF %ERRORLEVEL% NEQ 0 ( exit /b 1 ) -REM 检查CUDA版本 -nvcc --version 2>nul | findstr "release 11.8" >nul -IF %ERRORLEVEL% NEQ 0 ( - echo ERROR: CUDA 11.8 not found or version mismatch - exit /b 1 -) - REM 检查COLMAP是否可用 colmap -h >nul 2>nul IF %ERRORLEVEL% NEQ 0 ( @@ -73,6 +66,13 @@ IF %ERRORLEVEL% NEQ 0 ( exit /b 1 ) +REM 检查CUDA版本 +nvcc --version 2>nul | findstr "release 11.8" >nul +IF %ERRORLEVEL% NEQ 0 ( + echo ERROR: CUDA 11.8 not found or version mismatch + exit /b 1 +) + REM 创建缓存目录 mkdir cache\gdown 2>nul mkdir cache\torch 2>nul