utils.py 文件源码

def img2tensor(img, img_shape):
    '''
        Transforms and preprocesses image for vgg model
        Inputs:
            img: numpy array, rgb image
        Outputs:
            tensor
    '''
    img = imresize(img, img_shape)
    img = image.img_to_array(img)    
    img = np.expand_dims(img,axis=0)
    img = preprocess_input(img)

    return img