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张量方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
您可能感兴趣的文章:
相关文章
今天在使用 Pandas 的 to_sql 方法时,遇到一些问题,所以下面这篇文章主要给大家介绍了关于Python pandas中to_sql的使用及问题的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下2022-12-12
这篇文章主要介绍了使用python实现将两个实数矩阵合并为一个复数矩阵的操作，具有很好的参考价值，希望对大家有所帮助。2021-05-05
这篇文章将通过两个小问题：年龄计算、斐波那契数列，带领大家深入浅出的理解两种语言的基本语法，并用以实际场景，需要的可以参考一下2022-03-03
这篇文章主要给大家介绍了关于Python matplotlib实用绘图技巧的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-05-05
这篇文章主要介绍了Python Learning-列表的更多操作,需要的朋友可以参考下2018-08-08
这篇文章主要介绍了通过案例解析python鸭子类型相关原理,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-10-10
这篇文章主要介绍了python paramiko远程服务器终端操作过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-12-12
小编今天将为大家详细介绍如何利用Python语言制作一个UI界面，大家可以通过一键点击就实现头像照片转化成动漫风格的功能，感兴趣的可以动手尝试一下2022-07-07
这篇文章主要介绍了python爬虫自动创建文件夹的功能，非常不错，具有一定的参考借鉴价值，需要的朋友参考下吧2018-08-08
遗传算法是模仿自然界生物进化机制发展起来的随机全局搜索和优化方法,它借鉴了达尔文的进化论和孟德尔的遗传学说,其本质是一种高效、并行、全局搜索的方法,本文给大家介绍了使用Python实现遗传算法的详细步骤,需要的朋友可以参考下2023-11-11
最新评论