Created by: yeyupiaoling

在做图片分类时,使用https://github.com/PaddlePaddle/models/blob/develop/image_classification/reader.py#L13读取自定义的图片,下面的是224 * 224的,如果我的图片是长方形的的200 * 300 的,该怎么办

def train_mapper(sample):
    '''
    map image path to type needed by model input layer for the training set
    '''
    img, label = sample
    img = paddle.image.load_image(img)
    img = paddle.image.simple_transform(img, 256, 224, True)
    return img.flatten().astype('float32'), label