mirror of
https://github.com/Vision-CAIR/MiniGPT-4.git
synced 2025-04-04 18:10:47 +00:00
Update blip_processors.py
This commit is contained in:
parent
23671f9861
commit
68d81af31c
@ -75,9 +75,8 @@ class Blip2ImageTrainProcessor(BlipImageBaseProcessor):
|
|||||||
|
|
||||||
self.transform = transforms.Compose(
|
self.transform = transforms.Compose(
|
||||||
[
|
[
|
||||||
transforms.RandomResizedCrop(
|
transforms.Resize(
|
||||||
image_size,
|
(image_size,image_size),
|
||||||
scale=(min_scale, max_scale),
|
|
||||||
interpolation=InterpolationMode.BICUBIC,
|
interpolation=InterpolationMode.BICUBIC,
|
||||||
),
|
),
|
||||||
transforms.ToTensor(),
|
transforms.ToTensor(),
|
||||||
|
Loading…
Reference in New Issue
Block a user