model_utils.py 文件源码

python
阅读 25 收藏 0 点赞 0 评论 0

项目:SuperResolution 作者: bguisard 项目源码 文件源码
def image_loader(image_name, max_sz=256):
    """ forked from pytorch tutorials """
    r_image = Image.open(image_name)
    mindim = np.min((np.max(r_image.size[:2]), max_sz))

    loader = transforms.Compose([transforms.CenterCrop(mindim),
                                 transforms.ToTensor()])

    image = Variable(loader(r_image))

    return image.unsqueeze(0)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号