Skip to content

Task for Pattern Recognition course taken in BUAA SASEE 2024. Applying CausalTAD from OpenTAD toolbox for UCF-Crime dataset

License

Notifications You must be signed in to change notification settings

sunk3nstar/OpenTAD

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

56 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CausalTAD for UCF-Crime

This repository contains the full OpenTAD toolbox and configs for CausalTAD to work with UCF-Crime dataset.

Our method is extracting features from tailored dataset with VideoMAE and training with CausalTAD model. The training command is

  torchrun --nnodes=1 --nproc_per_node=1 --rdzv_backend=c10d --rdzv_endpoint=localhost:0 tools/train.py configs/causaltad/ucf_crime_videomae.py

We achieved a highest average mAP of 32.26%.

The following contents are inherited from the Original OpenTAD repo.

OpenTAD: An Open-Source Temporal Action Detection Toolbox.

OpenTAD is an open-source temporal action detection (TAD) toolbox based on PyTorch.

🥳 What's New

📖 Major Features

  • Support SoTA TAD methods with modular design. We decompose the TAD pipeline into different components, and implement them in a modular way. This design makes it easy to implement new methods and reproduce existing methods.
  • Support multiple TAD datasets. We support 9 TAD datasets, including ActivityNet-1.3, THUMOS-14, HACS, Ego4D-MQ, EPIC-Kitchens-100, FineAction, Multi-THUMOS, Charades, and EPIC-Sounds Detection datasets.
  • Support feature-based training and end-to-end training. The feature-based training can easily be extended to end-to-end training with raw video input, and the video backbone can be easily replaced.
  • Release various pre-extracted features. We release the feature extraction code, as well as many pre-extracted features on each dataset.

🌟 Model Zoo

One Stage Two Stage DETR End-to-End Training

The detailed configs, results, and pretrained models of each method can be found in above folders.

🛠️ Installation

Please refer to install.md for installation.

📝 Data Preparation

Please refer to data.md for data preparation.

🚀 Usage

Please refer to usage.md for details of training and evaluation scripts.

📄 Updates

Please refer to changelog.md for update details.

🤝 Roadmap

All the things that need to be done in the future is in roadmap.md.

🖊️ Citation

[Acknowledgement] This repo is inspired by OpenMMLab project, and we give our thanks to their contributors.

If you think this repo is helpful, please cite us:

@misc{2024opentad,
    title={OpenTAD: An Open-Source Toolbox for Temporal Action Detection},
    author={Shuming Liu, Chen Zhao, Fatimah Zohra, Mattia Soldan, Carlos Hinojosa, Alejandro Pardo, Anthony Cioppa, Lama Alssum, Mengmeng Xu, Merey Ramazanova, Juan León Alcázar, Silvio Giancola, Bernard Ghanem},
    howpublished = {\url{https://github.com/sming256/opentad}},
    year={2024}
}

If you have any questions, please contact: shuming.liu@kaust.edu.sa.

About

Task for Pattern Recognition course taken in BUAA SASEE 2024. Applying CausalTAD from OpenTAD toolbox for UCF-Crime dataset

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 98.0%
  • Cuda 1.1%
  • Other 0.9%