zizheng-guo / RhythmMamba

RhythmMamba

RhythmMamba

🔧 Setup

STEP1: bash setup1.sh

STEP2: conda activate rppg-toolbox

STEP3: bash setup2.sh

The codes are based on rPPG-toolbox

💻 Example of Using Pre-trained Models

Please use config files under ./configs/infer_configs

For example, if you want to run the pre-trained model for intra-dataset on MMPD, use python main.py --config_file ./configs/infer_configs/MMPD_RHYTHMMAMBA.yaml

💻 Examples of Neural Network Training

Please use config files under ./configs/train_configs

Intra-dataset on MMPD With RhythmMamba

STEP 1: Download the MMPD raw data by asking the paper authors

STEP 2: Modify ./configs/train_configs/intra/0MMPD_RHYTHMMAMBA.yaml

STEP 3: Run python main.py --config_file ./configs/train_configs/intra/0MMPD_RHYTHMMAMBA.yaml

Cross-dataset - Training on PURE and testing on UBFC-rPPG With RhythmMamba

STEP 1: Download the PURE raw data by asking the paper authors.

STEP 2: Download the UBFC-rPPG raw data via link

STEP 3: Modify ./configs/train_configs/cross/PURE_UBFC-rPPG_RHYTHMMAMBA.yaml

STEP 4: Run python main.py --config_file ./configs/train_configs/cross/PURE_UBFC-rPPG_RHYTHMMAMBA.yaml