From f2a516dbd81716784f2d70e77df1c68b02760421 Mon Sep 17 00:00:00 2001 From: Elias Schwalme <132968734+eliasschwalme@users.noreply.github.com> Date: Sun, 11 Feb 2024 00:26:18 +0100 Subject: [PATCH] Update augmentation.py --- augmentation.py | 1 + 1 file changed, 1 insertion(+) diff --git a/augmentation.py b/augmentation.py index df77004..31ae0d2 100644 --- a/augmentation.py +++ b/augmentation.py @@ -19,6 +19,7 @@ def crop_clip(clip, min_h, min_w, h, w): if isinstance(clip[0], np.ndarray): cropped = [img[min_h:min_h + h, min_w:min_w + w, :] for img in clip] + cropped = [img[min_h:min_h + h, min_w:min_w + w, :] elif isinstance(clip[0], PIL.Image.Image): cropped = [