Pytorch使用PIL和Numpy将单张图片转为Pytorch张量

Song1830 次浏览0个评论2018年07月11日

这里会讲解一下如何通过PIL读取单张图片将其转换为Pytorch张量 [code]#!/usr/bin/python

-- coding: UTF-8 --

from PIL import Image import numpy as np import torch

pil_img = Image.open('/Users/songlu/Desktop/code.jpg') img = np.array(pil_img) print torch.from_numpy(img)[/code]

提交评论

请登录后评论

用户评论

    当前暂无评价,快来发表您的观点吧...

更多相关好文

    当前暂无更多相关好文推荐...

    公告提示

    Pytorch是Facebook的AI研究团队发布了一个Python工具包,是Python优先的深度学习框架。作为numpy的替代品;使用强大的GPU能力,提供最大的灵活性和速度,实现了机器学习框架Torch在Python语言环境的执行,基于python且具备强大GPU加速的张量和动态神经网络。