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

返回列表

求一个数的平方根函数sqrt(int num) ,在大多数语言中都提供实现。那么要求一个数的平方根，是怎么实现的呢？实际上求平方根的算法方法主要有两种：二分法(binary searc...

本文实例为大家分享了python多线程同步之文件读写控制的具体代码，供大家参考，具体内容如下 1、实现文件读写的文件ltz_schedule_times.py #! /usr/bin...

Python socket C/S结构的聊天室应用服务端： #!/usr/bin/env python #coding:utf8 import socket,select def...

模块概述如果说模块是按照逻辑来组织 Python 代码的方法, 那么文件便是物理层上组织模块的方法。因此, **一个文件被看作是一个独立模块, 一个模块也可以被看作是一个文件。模...

解码邮件 python自带的email模块是个很有意思的东西，它可以对邮件编码解码，用来处理邮件非常好用。处理邮件是一个很细致的工作，尤其是解码邮件，因为它的格式变化太多了，下面先看看...

宜配屋