diff --git a/minigpt4/processors/blip_processors.py b/minigpt4/processors/blip_processors.py index fd26160..ee3f694 100644 --- a/minigpt4/processors/blip_processors.py +++ b/minigpt4/processors/blip_processors.py @@ -75,9 +75,8 @@ class Blip2ImageTrainProcessor(BlipImageBaseProcessor): self.transform = transforms.Compose( [ - transforms.RandomResizedCrop( - image_size, - scale=(min_scale, max_scale), + transforms.Resize( + (image_size,image_size), interpolation=InterpolationMode.BICUBIC, ), transforms.ToTensor(), @@ -138,4 +137,4 @@ class Blip2ImageEvalProcessor(BlipImageBaseProcessor): mean = cfg.get("mean", None) std = cfg.get("std", None) - return cls(image_size=image_size, mean=mean, std=std) \ No newline at end of file + return cls(image_size=image_size, mean=mean, std=std)