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

yipeiwu_com6年前 (2020-03-06)Python基础

在使用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设计】。

返回列表

首先我们要在邮箱的设置中开通那个POP3 然后我们要导入这些包 import poplib from datetime import datetime import jieba im...

前面我们介绍了cookies，主要应用在用户登录上，保存用户登录状态，不过cookies直接放在了浏览器上，安全性较低，所以我们便引出了session功能与cookies相同，不同的是它...

今天我们讲一下用Python写的GUI小程序。一个小闹钟（只是屏幕提示，没有声音哦）让我们先介绍这个闹钟如何奇葩。需要通过命令行启动。没有标题栏。没菜单。甚至没有关闭按钮。没...

django加载本地html from django.shortcuts import render from django.http import HttpResponse fro...

这次主要记录在windows下嵌入 python 解释器的过程，程序没有多少，主要是头文件与库文件的提取。程序平台：windows10 64 bit、 Qt 5.5.1 M...

宜配屋