search

zizheng-guo / RhythmMamba

RhythmMamba
28 stars 4 forks source link

Instructions for Inferencing on Custom mp4 files #6

Open ss8319 opened 1 month ago

ss8319 commented 1 month ago

Hi. Thank you for the excellent work.

I like to run some inferencing on custom mp4 files. My ground truth data comes from a pulse oximeter. Can you provide some sample code to run BPM estimation for an input mp4 video?

zizheng-guo commented 1 month ago

You can create a dataloader for the custom mp4 files similar to UBFCrPPGLoader.py (with label set to a fixed value), configure the yaml file to inference, and then export predictions in the trainer file.