shujakuin/cl-lmm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat✨: 增加图像尺寸调整逻辑以支持小于28的宽高

Browse Source
This commit is contained in:
YunyaoZhou 2025-01-07 22:40:36 +08:00
parent 7cf28e9851
commit 8d6e5d5416
Signed by: shujakuin
GPG Key ID: 418C3CA28E350CCF
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/dataset_library/OCRVQADataset.py
View File

@ -66,6 +66,13 @@ class OCRVQADataset(Dataset):
new_width = int(width * ratio) new_width = int(width * ratio)
new_height = int(height * ratio) new_height = int(height * ratio)
image = image.resize((new_width, new_height), Image.Resampling.BILINEAR) image = image.resize((new_width, new_height), Image.Resampling.BILINEAR)
if width < 28 or height < 28:
min_size = min(width, height)
ratio = 28 / min_size + 1
new_width = int(width * ratio)
new_height = int(height * ratio)
image = image.resize((new_width, new_height), Image.Resampling.BILINEAR)
question = sample["question"] question = sample["question"]
answer = sample["answer"] answer = sample["answer"]