mshukor commited on
Commit
9c36953
·
1 Parent(s): 53fbbb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -196,6 +196,9 @@ pad_idx = general_task.src_dict.pad()
196
 
197
  # Video process
198
 
 
 
 
199
  type_transform = transforms.Lambda(lambda x: x.float().div(255.0))
200
  patch_video_resize_transform = transforms.Compose([
201
  transforms.CenterCrop(video_caption_cfg.task.patch_frame_size),
 
196
 
197
  # Video process
198
 
199
+ mean = [0.5, 0.5, 0.5]
200
+ std = [0.5, 0.5, 0.5]
201
+
202
  type_transform = transforms.Lambda(lambda x: x.float().div(255.0))
203
  patch_video_resize_transform = transforms.Compose([
204
  transforms.CenterCrop(video_caption_cfg.task.patch_frame_size),