ActivityNet 开源项目教程

1. 项目目录结构及介绍

ActivityNet 项目的目录结构如下：

ActivityNet/
├── data/
│   ├── annotations/
│   ├── videos/
│   └── ...
├── scripts/
│   ├── download_videos.py
│   ├── extract_features.py
│   └── ...
├── src/
│   ├── models/
│   ├── utils/
│   └── main.py
├── config/
│   ├── config.yaml
│   └── ...
├── README.md
├── requirements.txt
└── ...

目录结构介绍

• data/: 存放数据集的目录，包括视频文件和标注文件。
  • annotations/: 存放视频的标注文件，通常是 JSON 格式。
  • videos/: 存放下载的视频文件。
• scripts/: 存放用于数据处理和特征提取的脚本。
  • download_videos.py: 用于下载视频的脚本。
  • extract_features.py: 用于提取视频特征的脚本。
• src/: 存放项目的源代码。
  • models/: 存放各种模型的实现代码。
  • utils/: 存放各种工具函数和辅助代码。
  • main.py: 项目的启动文件。
• config/: 存放项目的配置文件。
  • config.yaml: 主要的配置文件，包含项目的各种参数设置。
• README.md: 项目的说明文档，通常包含项目的简介、安装方法、使用说明等。
• requirements.txt: 项目依赖的 Python 包列表。

2. 项目的启动文件介绍

项目的启动文件是 src/main.py。该文件是整个项目的入口，负责初始化配置、加载数据、调用模型进行训练或推理等操作。

main.py 文件的主要功能

• 初始化配置: 从 config/config.yaml 文件中读取配置参数。
• 加载数据: 根据配置文件中的路径加载数据集。
• 模型初始化: 根据配置文件中的参数初始化模型。
• 训练或推理: 根据配置文件中的设置，选择进行模型训练或推理。

3. 项目的配置文件介绍

项目的配置文件是 config/config.yaml。该文件采用 YAML 格式，包含了项目的各种配置参数，如数据路径、模型参数、训练参数等。

config.yaml 文件的主要内容

data:
  annotations_path: "data/annotations"
  videos_path: "data/videos"

model:
  name: "resnet50"
  num_classes: 203

training:
  batch_size: 32
  epochs: 50
  learning_rate: 0.001

inference:
  use_gpu: true
  threshold: 0.5

配置文件介绍

• data: 数据相关的配置，包括标注文件路径和视频文件路径。
• model: 模型相关的配置，包括模型名称和类别数量。
• training: 训练相关的配置，包括批量大小、训练轮数和学习率。
• inference: 推理相关的配置，包括是否使用 GPU 和置信度阈值。

通过修改 config.yaml 文件，可以灵活地调整项目的运行参数，以适应不同的需求和环境。