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 = [