温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

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

发布时间:2020-07-22 10:27:24 来源:亿速云 阅读:406 作者:小猪 栏目:开发技术

这篇文章主要讲解了Pytorch如何使用PIL和Numpy将单张图片转为Pytorch张量,内容清晰明了,对此有兴趣的小伙伴可以学习一下,相信大家阅读完之后会有帮助。

我就废话不多说了,大家还是直接看代码吧!

#!/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) 

补充知识:pytorch mxnet 多GPU训练 batch 的区别

区别有很多,这是一点点不一样

1、mxnet 的batch ,设置的是单卡的,多卡是 batch*多卡,多一张卡,速度可以提升一倍

2、pytorch 是 batch/多卡数量,分配给每个gpu,设置的batch是总的batch

看完上述内容,是不是对Pytorch如何使用PIL和Numpy将单张图片转为Pytorch张量有进一步的了解,如果还想学习更多内容,欢迎关注亿速云行业资讯频道。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI