Update app.py
Browse files
app.py
CHANGED
@@ -176,7 +176,7 @@ class CreateDatasetProd():
|
|
176 |
# Read and process Videos
|
177 |
video = self.read_video(video_paths)
|
178 |
video = torch.from_numpy(video.asnumpy())
|
179 |
-
video = transforms.v2.functional.resize(video.
|
180 |
video = self.add_landmarks(video)
|
181 |
# Data Preperation for ML Model without Augmentation
|
182 |
video = self.transform_prod(video.permute(0, 3, 1, 2))
|
|
|
176 |
# Read and process Videos
|
177 |
video = self.read_video(video_paths)
|
178 |
video = torch.from_numpy(video.asnumpy())
|
179 |
+
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
|
180 |
video = self.add_landmarks(video)
|
181 |
# Data Preperation for ML Model without Augmentation
|
182 |
video = self.transform_prod(video.permute(0, 3, 1, 2))
|