Fix crop for small videos.

Author: datumbox

torchvision/transforms/_presets.py

@@ -113,13 +113,15 @@ def forward(self, vid: Tensor) -> Tensor:
         if self.crop_size is not None:
             if self.crop_mode == "center_crop":
                 vid = F.center_crop(vid, self.crop_size)
-            else:
+            elif vid.shape[-2] >= self.crop_size[-2] and vid.shape[-1] >= self.crop_size[-1]:
                 crops = (
                     list(F.five_crop(vid, self.crop_size))
                     if self.crop_mode == "five_crop"
                     else list(F.ten_crop(vid, self.crop_size))
                 )
                 vid = torch.cat(crops)
+            else:
+                vid = F.resize(vid, self.crop_size, interpolation=self.interpolation)
         vid = F.convert_image_dtype(vid, torch.float)
         vid = F.normalize(vid, mean=self.mean, std=self.std)
         H, W = vid.shape[-2:]