PIEPredict 项目安装与使用教程

PIEPredict 项目安装与使用教程

PIEPredict PIE: A Large-Scale Dataset and Models for Pedestrian Intention Estimation and Trajectory Prediction 项目地址: https://gitcode.com/gh_mirrors/pi/PIEPredict

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

PIEPredict 项目的目录结构如下：

PIEPredict/
├── pie_predict/
│   ├── __init__.py
│   ├── pie_data.py
│   ├── train_test.py
│   └── ...
├── PIE_dataset/
│   ├── annotations/
│   │   ├── set01/
│   │   ├── set02/
│   │   └── ...
│   ├── PIE_clips/
│   │   ├── set01/
│   │   ├── set02/
│   │   └── ...
│   └── ...
├── requirements.txt
├── README.md
└── ...

目录结构介绍

• pie_predict/: 包含项目的主要代码文件，如数据接口 pie_data.py 和训练测试脚本 train_test.py。
• PIE_dataset/: 存放 PIE 数据集的目录，包括标注文件 annotations/ 和视频剪辑 PIE_clips/。
• requirements.txt: 列出了项目所需的 Python 依赖库。
• README.md: 项目的说明文档。

2. 项目的启动文件介绍

项目的启动文件主要是 train_test.py，该文件用于训练和测试模型。

train_test.py

该脚本用于从零开始训练所有模型，并在测试数据上进行评估。使用方法如下：

python train_test.py 1

该命令将启动模型的训练和测试过程。

3. 项目的配置文件介绍

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

环境变量配置

1. 设置 PIE 数据根目录：

   export PIE_PATH=/path/to/PIE/data/root
   

2. 设置 PIE 数据剪辑目录：

   export PIE_RAW_PATH=/path/to/PIE/data/PIE_clips/
   

依赖库安装

建议使用虚拟环境安装依赖库：

1. 安装虚拟环境：

   sudo apt-get install virtualenv
   

2. 创建虚拟环境：

   virtualenv --system-site-packages -p python3 /venv
   source venv/bin/activate
   

3. 安装依赖库：

   pip3 install -r requirements.txt
   

通过以上步骤，您可以成功配置并启动 PIEPredict 项目。

PIEPredict PIE: A Large-Scale Dataset and Models for Pedestrian Intention Estimation and Trajectory Prediction 项目地址: https://gitcode.com/gh_mirrors/pi/PIEPredict