Skip to content

Commit 29e7bc3

Browse files
committed
Fix std/mean values in transforms.
1 parent 5d71b7c commit 29e7bc3

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

torchvision/models/video/mvit.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -434,7 +434,13 @@ def _mvit(
434434
class MViT_V1_B_Weights(WeightsEnum):
435435
KINETICS400_V1 = Weights(
436436
url="https://download.pytorch.org/models/mvit_v1_b-a3d8bcb8.pth",
437-
transforms=partial(VideoClassification, crop_size=(224, 224), resize_size=(256,)),
437+
transforms=partial(
438+
VideoClassification,
439+
crop_size=(224, 224),
440+
resize_size=(256,),
441+
mean=(0.45, 0.45, 0.45),
442+
std=(0.225, 0.225, 0.225),
443+
),
438444
meta={
439445
"min_size": (224, 224),
440446
"min_temporal_size": 16,

0 commit comments

Comments
 (0)