FasterSeg 项目使用教程

FasterSeg 项目使用教程

1. 项目目录结构及介绍

FasterSeg 项目的目录结构如下：

FasterSeg/
├── images/
├── latency/
├── search/
├── tools/
├── train/
├── .gitignore
├── LICENSE
├── README.md
├── requirements.txt

目录介绍：

• images/: 存放项目相关的图片文件。
• latency/: 包含与延迟测量相关的工具和脚本。
• search/: 包含用于搜索网络架构的脚本和配置文件。
• tools/: 包含项目使用的各种工具脚本。
• train/: 包含用于训练网络的脚本和配置文件。
• .gitignore: Git 忽略文件，指定哪些文件和目录不需要被 Git 管理。
• LICENSE: 项目的开源许可证文件。
• README.md: 项目的介绍文档，包含项目的概述、使用方法等信息。
• requirements.txt: 项目依赖的 Python 包列表。

2. 项目的启动文件介绍

FasterSeg 项目的主要启动文件位于 search/ 和 train/ 目录中。以下是主要的启动文件及其功能介绍：

search/

• train_search.py: 用于预训练超网络和搜索网络架构的启动文件。

train/

• train.py: 用于训练教师网络和学生网络的启动文件。

启动示例：

# 预训练超网络
CUDA_VISIBLE_DEVICES=0 python search/train_search.py

# 训练教师网络
CUDA_VISIBLE_DEVICES=0 python train/train.py --mode teacher

# 训练学生网络
CUDA_VISIBLE_DEVICES=0 python train/train.py --mode student

3. 项目的配置文件介绍

FasterSeg 项目的配置文件主要位于 search/ 和 train/ 目录中，用于配置训练和搜索过程中的各种参数。

search/

• config_search.py: 用于配置搜索过程中的参数，如预训练轮数、搜索轮数等。

train/

• config_train.py: 用于配置训练过程中的参数，如训练模式（教师或学生）、加载路径等。

配置示例：

# config_search.py
pretrain = True  # 是否进行预训练
pretrain_epochs = 20  # 预训练轮数
search_epochs = 30  # 搜索轮数

# config_train.py
mode = "teacher"  # 训练模式：教师或学生
load_path = "search-224x448_F12_L16_batch2-20200102-123456"  # 加载路径
teacher_path = "train-512x1024_teacher_batch12-20200103-234501"  # 教师网络路径

通过以上配置文件，用户可以灵活地调整训练和搜索过程中的各种参数，以满足不同的需求。