File tree Expand file tree Collapse file tree 1 file changed +13
-10
lines changed
Expand file tree Collapse file tree 1 file changed +13
-10
lines changed Original file line number Diff line number Diff line change @@ -667,16 +667,19 @@ class ResizeImagesByLongerEdgeNode(ImageProcessingNode):
667667
668668 @classmethod
669669 def _process (cls , image , longer_edge ):
670- img = tensor_to_pil (image )
671- w , h = img .size
672- if w > h :
673- new_w = longer_edge
674- new_h = int (h * (longer_edge / w ))
675- else :
676- new_h = longer_edge
677- new_w = int (w * (longer_edge / h ))
678- img = img .resize ((new_w , new_h ), Image .Resampling .LANCZOS )
679- return pil_to_tensor (img )
670+ resized_images = []
671+ for image_i in image :
672+ img = tensor_to_pil (image_i )
673+ w , h = img .size
674+ if w > h :
675+ new_w = longer_edge
676+ new_h = int (h * (longer_edge / w ))
677+ else :
678+ new_h = longer_edge
679+ new_w = int (w * (longer_edge / h ))
680+ img = img .resize ((new_w , new_h ), Image .Resampling .LANCZOS )
681+ resized_images .append (pil_to_tensor (img ))
682+ return torch .cat (resized_images , dim = 0 )
680683
681684
682685class CenterCropImagesNode (ImageProcessingNode ):
You can’t perform that action at this time.
0 commit comments