Giter VIP home page Giter VIP logo

actionmae's Introduction

ActionMAE

Pytorch code for our AAAI 2023 paper "Towards Good Practices for Missing Modality Robust Action Recognition".

Action Recognition with Missing Modality

Missing Modality Action Recognition
Standard multi-modal action recognition assumes that the modalities used in the training stage are complete at inference time: (a) → (b). We address the action recognition problem in situations where such assumption is not established, i.e., when modalities are incomplete at inference time: (a) → (c). Our goal is to maintain performance in the absence of any input modality.

Get Started

$ git clone https://github.com/sangminwoo/ActionMAE.git
$ cd ActionMAE

Dependencies

  • Pytorch 1.11.0
  • CUDA Toolkit 11.3
  • NVIDIA Apex

Environment Setup

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch
  • Goto NVIDIA Apex, and follow the instruction.

  • See requirements.txt for all python dependencies, and you can install them using the following command.

$ pip install -r requirements.txt

Train & Eval

$ ./train_val_actionmae_multigpu.sh

See/modify configurations in ActionMAE/lib/configs.py

Citation

@inproceedings{woo2023towards,
  title={Towards Good Practices for Missing Modality Robust Action Recognition},
  author={Woo, Sangmin and Lee, Sumin and Park, Yeonju and Nugroho, Muhammad Adi and Kim, Changick},
  booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
  volume={37},
  number={1},
  year={2023}
}

Acknowledgement

We appreciate much the nicely organized codes developed by MAE and pytorch-image-models. Our codebase is built on them.

actionmae's People

Contributors

sangminwoo avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar

Forkers

mondalanindya

actionmae's Issues

Any training suggestions?

I am working on the reproduction of this paper, and I found that using the baseline model to train the NTU dataset with RGB+Depth two modalities, the highest accuracy I achieved was acc==77, I don’t know why.
In addition, my data set reading speed is very slow.
I wonder if the author or other friends can give me some training experience and suggestions.

I used the dataset downloaded from the official website and used the script to extract frames from the RGB modality videos, 16 frames per sample.
I made very few changes to the source code, and the parameters are basically the default parameters of the source code.

Thank you very much.

Some questions regarding paper

Thank you for the interesting work! I just had a few questions from the paper...

In Section 3.2 "Note that only memory token and dummy tokens are affected by the reconstruction loss (i.e., reconstruction loss is not computed for the remaining tokens) during training." Does this mean the reconstruction loss is not backpropagated to the encoders of the "healthy" (retained) modalities? If so, why might this be done/were you able to do experiments ablating this?

Also, I was wondering if you had run experiments on simply dropping out a modality during training for the baseline to help the model get used to missing modalities.

Thank you!

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.