Awesome
RhythmMamba
:wrench: Setup
STEP1: bash setup1.sh
STEP2: conda activate toolbox
STEP3: bash setup2.sh
The codes are based on rPPG-toolbox
:computer: 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
:computer: 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