From 6d19f3c0c804928048b8f5b206d3e130572b5285 Mon Sep 17 00:00:00 2001 From: czk32611 Date: Sat, 1 Jun 2024 22:23:47 +0800 Subject: [PATCH] Remove crop_audio_window from DataLoader.py --- train_codes/DataLoader.py | 14 +------------- 1 file changed, 1 insertion(+), 13 deletions(-) diff --git a/train_codes/DataLoader.py b/train_codes/DataLoader.py index 8dacbde..0d7436e 100644 --- a/train_codes/DataLoader.py +++ b/train_codes/DataLoader.py @@ -118,18 +118,6 @@ class Dataset(object): return window - - def crop_audio_window(self, spec, start_frame): - if type(start_frame) == int: - start_frame_num = start_frame - else: - start_frame_num = self.get_frame_id(start_frame) - start_idx = int(80. * (start_frame_num / float(hparams.fps))) - - end_idx = start_idx + syncnet_mel_step_size - - return spec[start_idx : end_idx, :] - def prepare_window(self, window): # 1 x H x W x 3 x = np.asarray(window) / 255. @@ -249,4 +237,4 @@ if __name__ == "__main__": ref_image, image, masked_image, mask, audio_feature = data print("ref_image: ", ref_image.shape) - \ No newline at end of file +