Update app.py
Browse files
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),
|