INNER CODE UNIT · Python
top
FoundationVision/Liquid · data_process/convert_imagepair_cc512.py:29
top = (resize_height - tgt_height)//2
right = (resize_width + tgt_width)//2
bottom = (resize_height + tgt_height)//2
# Crop the center of the image
input_image = input_image.crop((left, top, right, bottom))
return input_image
def process_item(data,image_tokenizer):
image_path = data['image_path']
try:
ori_image = Image.open(image_path).convert("RGB")
except:
print('load error', image_path)
return None
original_ratio = max(ori_image.size) / min(ori_image.size) # remove images with ratio>2
if original_ratio >2 :
return None