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

yipeiwu_com5年前 (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设计】。

返回列表

这几天用到了raw socket，用python写了些demo程序，这里记录下。首先我们看一个简单的sniffer程序：复制代码代码如下:#! /usr/bin/python# c...

我们经常在处理字符串时遇到有很多空格的问题，一个一个的去手动删除不是我们程序员应该做的事情，今天这篇技巧的文章【听图阁-专注于Python设计】就来给大家讲一下，如何用Python去除字...

Python的标准库中的os模块包含普遍的操作系统功能。如果你希望你的程序能够与平台无关的话，这个模块是尤为重要的。即它允许一个程序在编写后不需要任何改动，也不会发生任何问题，就可以在L...

一：安装PyQt5 pip install pyqt5 二：PyQt5简单使用 1：使用PyQt5创建一个简单窗口 import sys from PyQt5 import...

我就废话不多说了，直接上代码吧！ import torch import torch.nn as nn import torch.nn.functional as F clas...

宜配屋