Mamba-MSQNet: A Fast and Efficient Model for Animal Action Recognition

The required Python packages can be installed with the following Bash command:

pip install -r requirements.txt

Then, for using correctly VideoMamba (after downloading it form the GitHub repo) you must:

pip install -e VideoMamba/causal-conv1d
pip install -e VideoMamba/mamba

To run the code simply fire the main.py file.

python mamba-MSQNet/main.py --dataset='animalkingdom' --model='videomambaclipinitvideoguidemultilayermamba' --total_length=16 --num_workers=2 --batch_size=8 --videomamba_version='m' 

Note: in case of using BaboonLand, change in_features (line 93, videomambaclipvideoguidemultilayermamba.py) with 29, as indicated in the comment.

To cite our article:

@article{fazzari2024selective,
  title={Selective state models are what you need for animal action recognition},
  author={Fazzari, Edoardo and Romano, Donato and Falchi, Fabrizio and Stefanini, Cesare},
  journal={Ecological Informatics},
  pages={102955},
  year={2024},
  publisher={Elsevier}
}

@inproceedings{fazzari2025artemis2,
  title={ARTEMIS2 \& ARTEMIS2-Mini: From Full-Scale to Distilled Real-Time Animal Behavior Recognition in a Robotic Dog},
  author={Fazzari, Edoardo and Romano, Donato and Falchi, Fabrizio and Stefanini, Cesare},
  booktitle={2025 5th International Conference on Robotics, Automation, and Artificial Intelligence (RAAI)},
  pages={318--325},
  year={2025},
  organization={IEEE}
}

@article{fazzari2025artemis,
  title={ARTEMIS: animal recognition through enhanced multimodal integration system},
  author={Fazzari, Edoardo and Romano, Donato and Falchi, Fabrizio and Stefanini, Cesare},
  journal={International Journal of Machine Learning and Cybernetics},
  volume={16},
  number={9},
  pages={5877--5892},
  year={2025},
  publisher={Springer}
}

@article{fazzari2025real,
  title={Real-Time Behavior Recognition Using a Legged Robot for Animal--Robot Interaction},
  author={Fazzari, Edoardo and Romano, Donato and Falchi, Fabrizio and Stefanini, Cesare},
  journal={Journal of Field Robotics},
  year={2025},
  publisher={Wiley Online Library}
}