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

返回列表

最简单的模式，C/S模式实现聊天室从半双工开始，何谓半双工？半双工即是说双方可以互发消息，但一次只能一个用户发送。只要稍微会点socket编程的人都会觉得很简单，所以过过...

如下所示： sum = 0 n = 99 while n > 0: sum = sum + n n = n - 2 print(sum) 只要条件满足，就不断循环，条...

如下所示： enc = OneHotEncoder(categorical_features=np.array([0,1,2]),n_values=[5,4,2]) enc.f...

Python 实现链表实例代码前言算法和数据结构是一个亘古不变的话题，作为一个程序员，掌握常用的数据结构实现是非常非常的有必要的。实现清单实现链表，本质上和语言是无关的。但是灵活...

本文实例为大家分享了python3.5仿微软计算器的具体代码，供大家参考，具体内容如下 from tkinter import * from math import * root =...

宜配屋