From 99240eb484463ab8ff1b35b724dec2dad0e9dbf2 Mon Sep 17 00:00:00 2001 From: liuzhian Date: Mon, 7 Jun 2021 15:36:58 +0800 Subject: [PATCH] fix ColorJetter bug as in issue #8 --- torchvideotransforms/video_transforms.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchvideotransforms/video_transforms.py b/torchvideotransforms/video_transforms.py index 96be114..fa9f5d8 100644 --- a/torchvideotransforms/video_transforms.py +++ b/torchvideotransforms/video_transforms.py @@ -487,8 +487,8 @@ def __call__(self, clip): jittered_clip = [] for img in clip: for func in img_transforms: - jittered_img = func(img) - jittered_clip.append(jittered_img) + img = func(img) + jittered_clip.append(img) else: raise TypeError('Expected numpy.ndarray or PIL.Image' +