Pytorch使用PIL和Numpy将单张图片转为Pytorch张量方式
更新时间：2020年05月25日 17:17:31 作者：tony2278
这篇文章主要介绍了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张量方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
您可能感兴趣的文章:
相关文章
这篇文章主要介绍了解决python运行启动报错问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06
这篇文章主要给大家介绍了关于Python利用myqr库创建自己的二维码的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-11-11
Python自动化操作Excel方法详解(xlrd,xlwt)
Excel是Windows环境下流行的、强大的电子表格应用。本文将详解用Python利用xlrd和xlwt实现自动化操作Excel的方法详细，需要的可以参考一下2022-06-06
Python tkinter实现的图片移动碰撞动画效果【附源码下载】
这篇文章主要介绍了Python tkinter实现的图片移动碰撞动画效果,涉及Python tkinter模块操作图片基于坐标动态变换与数值判定实现移动、碰撞检测等相关操作技巧,需要的朋友可以参考下2018-01-01
在本篇文章里小编给大家整理的是关于python数据爬下来保存的位置，需要的朋友们可以参考下。2020-02-02
这篇文章主要介绍了Python如何获取Win7，Win10系统缩放大小,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-01-01
随着数字化时代的到来，我们经常需要从图像中提取文本信息,本文将介绍如何使用Python编程语言和wxPython图形用户界面库构建一个简单的中文OCR截图工具,需要的可以参考下2023-08-08
python 使用GDAL实现栅格tif转矢量shp的方式小结
今天通过本文给大家分享python 使用GDAL实现栅格tif转矢量shp的方式小结，计划是使用栅格转矢量的方式，将栅格数据转为矢量shp文件，然后进行矢量切片，使用Mapbox进行前端动态渲染，具体内容详情跟随小编一起看看吧2021-08-08
这篇文章主要为大家详细介绍了python实现126邮箱发送邮件，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2020-05-05
今天小编就为大家分享一篇关于Python饼状图的绘制实例，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧2019-01-01
最新评论