rPPG packaging project

rPPG custom project

Dataset

VIPL-HR-V1 Personal

Quick setup

# clone project
git clone https://github.com/DimensionSTP/rppg-project.git
cd rppg-project

# [OPTIONAL] create conda environment
conda create -n myenv python=3.10 -y
conda activate myenv

# install requirements
pip install -r requirements.txt

.env file setting

PROJECT_DIR={PROJECT_DIR}
CONNECTED_DIR={CONNECTED_DIR}
DEVICES={DEVICES}
HF_HOME={HF_HOME}
USER_NAME={USER_NAME}

Model Hyper-Parameters Tuning

• end-to-end

python main.py mode=tune is_tuned=untuned num_trials={num_trials}

Training

• end-to-end

python main.py mode=train is_tuned={tuned or untuned} num_trials={num_trials}

Testing

• end-to-end

python main.py mode=test is_tuned={tuned or untuned} num_trials={num_trials} epoch={ckpt epoch}

Prediction

• end-to-end

python main.py mode=predict is_tuned={tuned or untuned} num_trials={num_trials} epoch={ckpt epoch}

Examples of shell scipts

• train

bash scripts/train.sh

• test

bash scripts/test.sh

If you want to change main config, use --config-name={config_name}.

Also, you can use --multirun option.

You can set additional arguments through the command line.