TraDeS 项目使用教程

1. 项目目录结构及介绍

TraDeS 项目的目录结构如下：

TraDeS/
├── experiments/
│   ├── crowdhuman.sh
│   ├── mot17_test.sh
│   ├── mot17_train.sh
│   ├── nuScenes_test.sh
│   ├── nuScenes_train.sh
├── models/
├── readme/
├── src/
│   ├── demo.py
│   ├── ...
├── videos/
├── .gitignore
├── LICENSE
├── NOTICE
├── README.md
├── mot_switch_version.sh
├── nuscenes_switch_version.sh
├── requirements.txt

目录介绍

• experiments/: 包含项目的实验脚本，用于训练和测试模型。
• models/: 存放预训练模型文件。
• readme/: 可能包含项目的额外文档或说明。
• src/: 项目的源代码目录，包含主要的 Python 脚本。
• videos/: 存放用于演示的视频文件。
• .gitignore: Git 忽略文件配置。
• LICENSE: 项目的开源许可证文件。
• NOTICE: 项目通知文件。
• README.md: 项目的主文档文件。
• mot_switch_version.sh: 用于切换 MOT 数据集 API 版本的脚本。
• nuscenes_switch_version.sh: 用于切换 nuScenes 数据集 API 版本的脚本。
• requirements.txt: 项目依赖的 Python 包列表。

2. 项目启动文件介绍

项目的启动文件主要位于 src/ 目录下，其中 demo.py 是用于运行演示的主要脚本。

src/demo.py

demo.py 是用于运行 2D 和 3D 跟踪演示的脚本。可以通过命令行参数指定不同的模型和视频文件进行演示。

示例命令

• 2D 跟踪演示:

  python demo.py tracking --dataset mot --load_model ../models/mot_half.pth --demo ../videos/mot_mini.mp4 --pre_hm --ltrb_amodal --pre_thresh 0.5 --track_thresh 0.4 --inference --clip_len 3 --trades --save_video --resize_video --input_h 544 --input_w 960
  

• 3D 跟踪演示:

  python demo.py tracking_ddd --dataset nuscenes --load_model ../models/nuscenes.pth --demo ../videos/nuscenes_mini.mp4 --pre_hm --track_thresh 0.1 --inference --clip_len 2 --trades --save_video --resize_video --input_h 448 --input_w 800 --test_focal_length 633
  

3. 项目的配置文件介绍

项目的配置文件主要包括 requirements.txt 和 experiments/ 目录下的脚本。

requirements.txt

requirements.txt 文件列出了项目运行所需的 Python 依赖包。可以使用以下命令安装这些依赖：

pip install -r requirements.txt

experiments/ 目录下的脚本

experiments/ 目录下的脚本用于配置和运行不同的实验，包括训练和测试模型。

• 训练 MOT17 数据集:

  sh experiments/mot17_train.sh
  

• 测试 MOT17 数据集:

  sh experiments/mot17_test.sh
  

• 训练 nuScenes 数据集:

  sh experiments/nuScenes_train.sh
  

• 测试 nuScenes 数据集:

  sh experiments/nuScenes_test.sh
  

这些脚本会根据项目的配置文件和数据集进行相应的训练和测试操作。