INNER CODE UNIT · Python

bottom

FoundationVision/Liquid · data_process/convert_imagepair_cc512.py:31

    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    

    input_image = center_crop_image(ori_image)

View source record →