INNER CODE UNIT · Python

left

FoundationVision/Liquid · data_process/convert_imagepair_cc512.py:28

    left = (resize_width - tgt_width)//2
    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 :

View source record →