From 7b80b7ed3394558846d2e93978ec462981e7789e Mon Sep 17 00:00:00 2001 From: Hatem Ibrahem <48507516+HatemHosam@users.noreply.github.com> Date: Wed, 27 Dec 2023 13:29:29 +0900 Subject: [PATCH] Update dataset.py --- dataloaders/dataset.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/dataloaders/dataset.py b/dataloaders/dataset.py index b5e01058..cad85eb6 100644 --- a/dataloaders/dataset.py +++ b/dataloaders/dataset.py @@ -125,8 +125,8 @@ def preprocess(self): file_path = os.path.join(self.root_dir, file) video_files = [name for name in os.listdir(file_path)] - train_and_valid, test = train_test_split(video_files, test_size=0.2, random_state=42) - train, val = train_test_split(train_and_valid, test_size=0.2, random_state=42) + #train_and_valid, test = train_test_split(video_files, test_size=0.2, random_state=42) + #train, val = train_test_split(train_and_valid, test_size=0.2, random_state=42) train_dir = os.path.join(self.output_dir, 'train', file) val_dir = os.path.join(self.output_dir, 'val', file) @@ -254,4 +254,4 @@ def crop(self, buffer, clip_len, crop_size): print(labels) if i == 1: - break \ No newline at end of file + break