TAL3x3

install

conda create --name rku python=3.8 -y
conda activate rku
conda install pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=10.2 -c pytorch  # **This** command will automatically install the latest version PyTorch and cudatoolkit, please check whether they match your environment.
pip install -U openmim
mim install mmengine
mim install mmcv==2.0.0
mim install mmdet
pip install einops
pip install numpy==1.23.5

pre_processing

All preprocessed dataset data is located in the pre_processing folder.

main function

rku_TAL.py is the main program. This file can be retrained according to the configuration.

python rku_TAL.py \
    --world_size 4 \
    --batch_size 8 \
    --epochs 20 \
    --use_bbox_feat \
    --use_skeleton_feat \
    --use_position_feat \
    --use_trajectory_feat