Human Activity Recognition from Wearable Sensor Data Using Self-Attention

Tensorflow 2.x implementation of "Human Activity Recognition from Wearable Sensor Data Using Self-Attention", 24th European Conference on Artificial Intelligence, ECAI 2020 by Saif Mahmud and M. Tanjid Hasan Tonmoy et al.

[ arXiV ] [ IOS Press ]

Installation

To install the dependencies in python3 environment, run:

pip install -r requirements.txt

Dataset Download

To download dataset and place it under data directory for model training and inference, run the script dataset_download.py with following commad:

python dataset_download.py --dataset DATASET --unzip

Here, the name of dataset in command line argument DATASET of this project will be as follows:

DATASET = < pamap2 / opp / uschad / skoda >

For example, to download PAMAP2 dataset and unzip under data directory, run the following command from project root:

python dataset_download.py --dataset pamap2 --unzip

Pretrained Models

The saved_model directory contains pretrained models for PAMAP2, Opportuninty, USC-HAD and Skoda dataset. These models can be used directly for inference and performance evaluation as described in the following section.

Training and Evaluation

Python script main.py will be used for model training, inference and performance evaluation. The arguments for this script are as follows:

-h, --help         show this help message and exit 
--train            Training Mode 
--test             (Testing / Evaluation) Mode
--epochs EPOCHS    Number of Epochs for Training
--dataset DATASET  Name of Dataset for Model Training or Inference

For example, in order to train model for 75 epochs on PAMAP2 dataset and evaluate model performance, run the following command:

TF_CPP_MIN_LOG_LEVEL=3 python main.py --train --test --epochs 75 --dataset pamap2

If the pretrained weights are stored in saved_model directory and to infer with that, run the following command:

TF_CPP_MIN_LOG_LEVEL=3 python main.py --test --dataset pamap2

Citation

@inproceedings{ECAI2020HAR-SaifTanjid,
  title={Human Activity Recognition from Wearable Sensor Data Using Self-Attention},
  author={Saif Mahmud and M. T. H. Tonmoy and Kishor Kumar Bhaumik and A. M. Rahman and M. A. Amin and M. Shoyaib and Muhammad Asif Hossain Khan and A. Ali},
  booktitle = {{ECAI} 2020 - 24th European Conference on Artificial Intelligence, 29 August-8 September 2020, Santiago de Compostela, Spain},
  year={2020}
}