HiSup 开源项目使用教程

HiSup 开源项目使用教程

HiSup 项目地址: https://gitcode.com/gh_mirrors/hi/HiSup

1. 项目的目录结构及介绍

HiSup 项目的目录结构如下：

HiSup/
├── config-files/
│   └── ... # 配置文件
├── figures/
│   └── ... # 项目相关图片
├── hisup/
│   └── ... # 项目核心代码
├── scripts/
│   └── ... # 脚本文件
├── tools/
│   └── ... # 工具文件
├── .gitignore
├── LICENSE
├── README.md
├── citation.bib
├── requirements.txt
└── ...

目录结构介绍

• config-files/: 存放项目的配置文件，用于定义训练和测试的参数。
• figures/: 存放项目相关的图片，如模型结构图等。
• hisup/: 项目核心代码，包括模型定义、数据处理等。
• scripts/: 存放项目的脚本文件，如训练脚本、测试脚本等。
• tools/: 存放项目的工具文件，如数据转换工具等。
• .gitignore: Git 忽略文件配置。
• LICENSE: 项目许可证文件。
• README.md: 项目介绍文件。
• citation.bib: 项目引用文件。
• requirements.txt: 项目依赖库文件。

2. 项目的启动文件介绍

HiSup 项目的启动文件主要位于 scripts/ 目录下，常用的启动文件包括：

• scripts/demo.py: 用于快速启动预训练模型的演示脚本。
• scripts/train.py: 用于训练模型的脚本。
• scripts/test.py: 用于测试模型的脚本。

启动文件介绍

• demo.py: 该脚本用于加载预训练模型并进行快速演示。可以通过命令行参数指定数据集和图像路径。
• train.py: 该脚本用于训练模型。可以通过配置文件指定训练参数，如数据集、模型结构等。
• test.py: 该脚本用于测试训练好的模型。可以通过配置文件指定测试参数，如模型路径、数据集等。

3. 项目的配置文件介绍

HiSup 项目的配置文件主要位于 config-files/ 目录下，常用的配置文件包括：

• config-files/crowdai-small_hrnet48.yaml: 用于 AICrowd 数据集的小规模训练配置。
• config-files/crowdai_hrnet48.yaml: 用于 AICrowd 数据集的训练配置。

配置文件介绍

• crowdai-small_hrnet48.yaml: 该配置文件用于 AICrowd 数据集的小规模训练，定义了训练的超参数、数据路径、模型结构等。
• crowdai_hrnet48.yaml: 该配置文件用于 AICrowd 数据集的训练，定义了训练的超参数、数据路径、模型结构等。

通过这些配置文件，用户可以灵活地调整训练和测试的参数，以适应不同的需求和数据集。

HiSup 项目地址: https://gitcode.com/gh_mirrors/hi/HiSup