Commit
·
bc40b6c
1
Parent(s):
ad54dd4
Update app.py
Browse files
app.py
CHANGED
@@ -169,7 +169,7 @@ class CreateDatasetProd():
|
|
169 |
|
170 |
def create_dataset(self, video_paths):
|
171 |
# Read and process Videos
|
172 |
-
video = self.read_video(
|
173 |
video = transforms.v2.functional.resize(video.permute(0, 3, 1, 2), size=(self.clip_size*2, self.clip_size*3)) # Auto converts to (F, C, H, W) format
|
174 |
video = self.add_landmarks(video)
|
175 |
# Data Preperation for ML Model without Augmentation
|
|
|
169 |
|
170 |
def create_dataset(self, video_paths):
|
171 |
# Read and process Videos
|
172 |
+
video = self.read_video(video_paths)
|
173 |
video = transforms.v2.functional.resize(video.permute(0, 3, 1, 2), size=(self.clip_size*2, self.clip_size*3)) # Auto converts to (F, C, H, W) format
|
174 |
video = self.add_landmarks(video)
|
175 |
# Data Preperation for ML Model without Augmentation
|