Sapiens 项目使用与配置指南

1. 项目目录结构及介绍

Sapiens 项目旨在为人类视觉任务提供高分辨率模型。项目目录结构如下：

sapiens/
├── _install/             # 安装脚本目录
├── assets/               # 资源文件目录
├── cv/                   # 计算机视觉相关代码
├── det/                  # 检测模块代码
├── docs/                 # 文档目录
├── engine/               # 引擎模块代码
├── lite/                 # 精简版模型代码
├── pose/                 # 姿态估计模块代码
├── pretrain/             # 预训练模块代码
├── seg/                  # 分割模块代码
├── .gitignore            # git 忽略文件
├── CODE_OF_CONDUCT.md    # 行为准则文件
├── CONTRIBUTING.md       # 贡献指南文件
├── LICENSE               # 许可证文件
├── README.md             # 项目说明文件
└── git_script.sh         # git 相关脚本

2. 项目的启动文件介绍

项目的启动主要是通过 _install/ 目录下的 conda.sh 脚本。以下是启动步骤：

1. 克隆仓库到本地：

   git clone https://github.com/facebookresearch/sapiens.git
   

2. 设置环境变量：

   export SAPIENS_ROOT=/path/to/sapiens
   

3. 运行安装脚本创建新的 conda 环境，并安装所需依赖：

   cd $SAPIENS_ROOT/_install
   ./conda.sh
   

4. 下载预训练模型权重文件，并放置在相应目录。

3. 项目的配置文件介绍

Sapiens 项目的配置主要通过环境变量来进行。以下是配置步骤：

1. 设置 SAPIENS_ROOT 环境变量，指向项目根目录。

2. 设置 SAPIENS_CHECKPOINT_ROOT 环境变量，指向预训练模型权重文件夹。

   export SAPIENS_CHECKPOINT_ROOT=/path/to/sapiens_host
   

3. 根据需要下载对应的预训练模型权重文件，并确保它们按照以下结构放置：

   sapiens_host/
   ├── detector/
   │   └── checkpoints/
   ├── pretrain/
   │   └── checkpoints/
   ├── pose/
   └── checkpoints/
   

确保正确设置这些配置文件和环境变量后，你就可以开始使用 Sapiens 项目进行相关任务的开发和研究。