Skip to content

This repository contains the code and dataset for activity recognition using mmWave radar sensors. The dataset was collected in two different environments with 23 subjects and includes range and micro-Doppler inputs. Additionally, there is an alternative task of human micro-Doppler generation using GANs.

License

Notifications You must be signed in to change notification settings

aidinattar/radio-wave-activity-detection

Repository files navigation

Patient Activity Recognition with Radio Waves

This is a research project on contactless sensing of human activities using mmWave radar devices. The aim is to monitor human movement in healthcare, security, and monitoring applications with high accuracy while preserving privacy.

The project involves two environments - a homelab and a real hospital - with 23 subjects involved. There are two types of input: range and Doppler, including micro Doppler. The dataset contains 2 sets of activities - in room and in bed - with file sizes varying from 1k to 2k samples per class.

In addition to activity recognition, the project also explores human micro Doppler generation with GANs, an active research topic. The dataset can be used for generating realistic micro Doppler examples, reducing the need for collecting real data.

The repository includes code for data preprocessing, model training, and evaluation. Please refer to the individual files for more details.

  radio-wave-activity-detection/
  │
  ├── /clustering
  │   ├── clustering.py       # Class meant to separate activities agnostically [TODO]
  │
  ├── /models
  │   ├── classifier.py          # class to manage training methods
  │   ├── GAN.py                 # generator and discriminator for generative adversial network
  │   ├── cGAN.py                # generator and discriminator for conditional GAN
  │   ├── ResNet18.py            # resnet model with 18 layers
  │   ├── ResNet50.py            # resnet model with 50 layers
  │   ├── inception_v4.py        # inception-v4 model
  │   ├── Inception_ResNet_v2.py # a trial of putting together models
  │   ├── Inception_v3.py        # inception-v3 model
  │   ├── cnn_md.py              # cnn-md model with 2d kernels
  │   ├── cnn_rd.py              # cnn-rd model with 3d kernels
  │   ├── custom_cnns.py         # general models to use custom numbers of layers
  │   ├── cnn_md_baseline.py     # cnn-md with 2 conv layers
  │   ├── cnn_md_inception.py    # cnn-md with inception module as unit
  │   ├── cnn_md_resnet.py       # cnn-md with skip connections
  │   └── pretrained_models.py   # trial for using pretrained models
  │
  ├── /preprocessing
  │   ├── DataReader.py  # class to read and perform first steps of preprocessing
  │   ├── DataCutter.py  # class to perform part of preprocessing
  │   ├── DataProcess.py # class to perform last part of preprocessing
  │   └── dataset.py     # classes to manage data in pytorch models
  │
  ├── /utils
  │   ├── augmentation.py   # methods to perform augmentation on 2D or 3D maps
  │   ├── clustering.py     # separate the activities using clustering techniques
  │   ├── constants.py      # some constants to import
  │   ├── cutting.py        # cutting utilities
  │   ├── plotting.py       # plotting methods
  │   ├── rescaling.py      # normalization methods
  │   └── torch.py          # class for early stopping 
  │
  ├── exceptions.py             # custom exceptions
  ├── features_extraction.ipynb # notebook to plot features extracted from the trained models
  ├── make_dataset.py           # main to split dataset and augment training samples
  ├── preprocess.py             # main to preprocess the dataset and produce a h5 file
  ├── requirements.txt          # requirements file
  ├── scraping.py               # download the data from PARrad
  ├── train.py                  # main to train the classification models
  ├── train_gan.py              # main to train GAN models
  ├── train_cgan.py             # main to train cGAN models
  ├── LICENSE          # License file
  └── README.md        # Project README file

About

This repository contains the code and dataset for activity recognition using mmWave radar sensors. The dataset was collected in two different environments with 23 subjects and includes range and micro-Doppler inputs. Additionally, there is an alternative task of human micro-Doppler generation using GANs.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published