Skip to content

Temporally smooth online action detection using cycle-consistent future anticipation

Notifications You must be signed in to change notification settings

YHKimGithub/FATSnet

Repository files navigation

Future Anticipation and Temporally Smoothing network (FATSnet)

Temporally Smooth Online Action Detection using Cycle-consistent Future Anticipation
Young Hwi Kim, Seonghyeon Nam, Seon Joo Kim
[arXiv]

Updates

25 Nov, 2021: Initial update

Installation

Prerequisites

  • Ubuntu 16.04
  • Python 2.7.17
  • CUDA 10.0

Requirements

  • pytorch==1.4.0
  • numpy==1.16.6
  • h5py==2.10.0

Training

Input Features

We provide the Kinetics pre-trained feature of THUMOS'14 dataset. The extracted features can be downloaded from here. Files should be located in 'data/'.
The feature that is pre-trained on Activitynet can be downloaded from here.

Trained Model

The trained models that used Kinetics pre-trained feature can be downloaded from here. Files should be located in 'checkpoints/'. The Activitynet version can be downloaded from here.

Training Model

For Kinetics pre-trained input feature,

python train.py --gen_feature_len=12

For Activitynet pre-trained input feature,

python train.py --gen_feature_len=8 --feature_size=3072

Testing

For Kinetics pre-trainedd input feature,

python prediction.py
python eval_map.py

For Activitynet pre-trained input feature,

python prediction.py --feature_size=3072
python eval_map.py
Dataset Feature mAP
THUMOS'14 TwoStream-Anet 51.6
THUMOS'14 TwoStream-Kinetics 59.0

Citing FATSnet

Please cite our paper in your publications if it helps your research:

@article{kim2021temporally,
  title={Temporally smooth online action detection using cycle-consistent future anticipation},
  author={Kim, Young Hwi and Nam, Seonghyeon and Kim, Seon Joo},
  journal={Pattern Recognition},
  volume={116},
  pages={107954},
  year={2021},
  publisher={Elsevier}
}

About

Temporally smooth online action detection using cycle-consistent future anticipation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages