Kaushik066 commited on
Commit
b4a6d61
·
verified ·
1 Parent(s): 1594daa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.transpose(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))
 
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))