Pytorch中的variable, tensor与numpy相互转化的方法

 更新时间:2019年10月10日 10:05:55   作者:pengge0433  
这篇文章主要介绍了Pytorch中的variable, tensor与numpy相互转化的方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧

在使用pytorch作为深度学习的框架时,经常会遇到变量variable、张量tensor与矩阵numpy的类型的相互转化的问题,本章结合这实际图像对此转化方法进行实现。

1.加载需要用到的模块

import torch
from torch.autograd import Variable
import matplotlib.pyplot as plt
import matplotlib.image as mpimg

2.显示图片与图片中的一部分区域

test_img = mpimg.imread('example1.jpg')
i_x = 20
i_y = 85
sub_img = test_img[i_y:i_y + 100,i_x:i_x + 100,:] #numpy类型

3.将numpy矩阵转换为Tensor张量

sub_ts = torch.from_numpy(sub_img)  #sub_img为numpy类型

4.将Tensor张量转化为numpy矩阵

sub_np1 = sub_ts.numpy()       #sub_ts为tensor张量

5.将numpy转换为Variable

sub_va = Variable(torch.from_numpy(sub_img))

6.将Variable张量转化为numpy

sub_np2 = sub_va.data.numpy()

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。

相关文章

  • Python中集合的内建函数和内建方法学习教程

    Python中集合的内建函数和内建方法学习教程

    这篇文章主要介绍了Python中集合的内建函数和内建方法学习教程,包括工厂函数和仅用于可变集合的方法等知识点,需要的朋友可以参考下
    2015-08-08
  • Micropython固件使用Pico刷固件并配置VsCode开发环境的方法

    Micropython固件使用Pico刷固件并配置VsCode开发环境的方法

    这篇文章主要介绍了Micropython固件使用Pico刷固件并配置VsCode开发环境的方法,本文通过图文并茂的形式给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下吧
    2021-07-07
  • Python各种相似度计算方法详解

    Python各种相似度计算方法详解

    这篇文章主要介绍了Python各种相似度计算方法详解,在日常开发中有需要计算相似度的需求,那么今天我们就来看一下各种计算相似度的方法,对日常开发非常有帮助,需要的朋友可以参考下
    2023-07-07
  • Python中的十大图像处理工具(小结)

    Python中的十大图像处理工具(小结)

    这篇文章主要介绍了Python中的十大图像处理工具,本文主要介绍了一些简单易懂最常用的Python图像处理库,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2019-06-06
  • Python中的time模块与datetime模块用法总结

    Python中的time模块与datetime模块用法总结

    Python中内置的各项时间日期函数几乎都来自于time和datetime这两个模块,下面整理了Python中的time模块与datetime模块用法总结,需要的朋友可以参考下
    2016-06-06
  • python如何实现TF-IDF算法

    python如何实现TF-IDF算法

    这篇文章主要介绍了python如何实现TF-IDF算法问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2023-11-11
  • 使用Python批量修改文件名的代码实例

    使用Python批量修改文件名的代码实例

    今天小编就为大家分享一篇关于使用Python批量修改文件名的代码实例,小编觉得内容挺不错的,现在分享给大家,具有很好的参考价值,需要的朋友一起跟随小编来看看吧
    2019-01-01
  • Python实现抓取HTML网页并以PDF文件形式保存的方法

    Python实现抓取HTML网页并以PDF文件形式保存的方法

    这篇文章主要介绍了Python实现抓取HTML网页并以PDF文件形式保存的方法,结合实例形式分析了PyPDF2模块的安装及Python抓取HTML页面并基于PyPDF2模块生成pdf文件的相关操作技巧,需要的朋友可以参考下
    2018-05-05
  • kali中python版本的切换方法

    kali中python版本的切换方法

    今天小编就为大家分享一篇kali中python版本的切换方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-07-07
  • Python彻底删除文件夹及其子文件方式

    Python彻底删除文件夹及其子文件方式

    今天小编就为大家分享一篇Python彻底删除文件夹及其子文件方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-12-12

最新评论