chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-03 09:30:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update blip_processors.py

Browse Source
This commit is contained in:
Jun Chen 2023-11-08 10:21:09 -08:00 committed by GitHub
parent 23671f9861
commit 68d81af31c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
minigpt4/processors/blip_processors.py
View File

@ -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)
return cls(image_size=image_size, mean=mean, std=std)