Skip to content

chihina/GAFL-CVPR2024

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Intro

Our codes are based on https://github.com/JacobYuan7/DIN-Group-Activity-Recognition-Benchmark.
I deeply appreciate their efforts. This is the official repository for the following paper:

Chihiro Nakatani, Hiroaki Kawashima, Norimichi Ukita
Learning Group Activity Features Through Person Attribute Prediction, CVPR2024
Project page: https://toyota-ti.ac.jp/Lab/Denshi/iim/ukita/selection/CVPR2024-GAFL.html

Top page

Environment

python 3.10.2
ROIAlign (https://github.com/longcw/RoIAlign.pytorch)

And you can use requirements.txt

pip install -r requirements.txt

Data preparation

1. Download dataset

You can download daatset from the following url.
These dataset are required to place in data/ in the repository as follows:

2. Training

2.1 Volleyball dataset

  • Ours
python scripts/train_volleyball_stage2_gr.py

The following folder contains the trained models.

  1. GAFL_PAC_VOL (GAFL-PAC)
  2. GAFL_PAF_VOL (GAFL-PAF)

2.2 Collective Activity dataset

  • Ours
python scripts/train_collective_stage2_gr.py

The following folder contains the trained models.

  1. GAFL_PAC_CAD (GAFL-PAC)
  2. GAFL_PAF_CAD (GAFL-PAF)

3. Evaluation

3.1 Volleyball dataset

You can choose the model that you would like to evaluate in the bash file script.

  • Ours
bash ./evaluation_vol.bash

3.2 Collective Activity dataset

  • Ours
bash ./evaluation_cad.bash

Releases

No releases published

Packages

No packages published