From 04ad14359ed742da14507efb251adfe460827137 Mon Sep 17 00:00:00 2001 From: coenarrow <31496917+coenarrow@users.noreply.github.com> Date: Fri, 21 Jun 2024 07:52:57 +0800 Subject: [PATCH] Update mtts_can.py Added missing imports of os and requests --- pyVHR/deepRPPG/mtts_can.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pyVHR/deepRPPG/mtts_can.py b/pyVHR/deepRPPG/mtts_can.py index a06c696..61c937f 100644 --- a/pyVHR/deepRPPG/mtts_can.py +++ b/pyVHR/deepRPPG/mtts_can.py @@ -8,6 +8,8 @@ import scipy.io from scipy.sparse import spdiags import h5py +import os +import requests def preprocess_raw_video(frames, fs=30, dim=36): """A slightly different version from the original: @@ -112,4 +114,4 @@ def MTTS_CAN_deep(frames, fs, model_checkpoint=None, batch_size=100, dim=36, img [b_pulse, a_pulse] = butter(1, [0.75 / fs * 2, 2.5 / fs * 2], btype='bandpass') #[b_pulse, a_pulse] = butter(1, [0.65 / fs * 2, 2.5 / fs * 4], btype='bandpass') pulse_pred = scipy.signal.filtfilt(b_pulse, a_pulse, np.double(pulse_pred)) - return pulse_pred \ No newline at end of file + return pulse_pred