百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 博客教程 > 正文

转置卷积层和卷积层是一样的吗?使用PyTorch进行概念试验

connygpt 2024-08-28 14:13 10 浏览

在这篇文章中,我将使用常规卷积操作测试和比较神经网络的分类任务,并通过用转置卷积替换卷积来比较相同的网络。

转置卷积运算

转置卷积操作现在是众所周知的,并且已经在许多需要上采样的模型中使用。它与卷积运算非常相似,只是卷积矩阵被转置。因此结果是输出增长而不是减少(取决于使用的padding,步幅和内核大小)。首先,让我们对输出在随机张量上的转置卷积的输出方式进行一些测试。

首先,我导入PyTorch并将其封装为变量,从批量大小为1、通道尺寸为1、高度为5、宽度为5的正态分布中抽取的随机张量如下图所示。然后我定义转置卷积运算,以得到正确的输入,内核大小为3x3,最大步长为1,padding为0。将这些转换为PyTorch代码(Python实现):

import torch

import torch.nn as nn

from torch.autograd import Variable

random_tensor = Variable(torch.randn(1, 1, 5, 5))

upsample_tnsr = nn.ConvTranspose2d(in_channels = 1, out_channels = 1, kernel_size = 3, stride = 1, padding = 0)

在继续之前,我想指定一些表示法,以方便地遵循这些定义。

一些设置:

  • $ latex n_W = $宽度的输入尺寸,$ latex n_H = $高度的输入尺寸。由于张量将是正方形,我们将参数n代表两者。
  • p =是应用于张量的填充
  • f =内核的大小
  • s =是将使用卷积运算应用的步幅。
  • c =是张量的通道维数

如果我们对张量应用常规卷积,结果可以描述如下:

这将是张量的宽度和高度的输出,并且它将是相同的,因为我们可以假设输入将具有相同的高度和宽度尺寸。

作为一个例子,假设我考虑上面指定的张量,在padding p = 1,内核大小f = 3和stride s = 1的情况下对其应用常规卷积运算,我们的输出将与输入n相同。

上图中:批量大小= 1,通道= 1,高度,宽度= 5的随机张量上运行卷积运算。内核f = 3,stride = 1 padding = 1的卷积输出在高度和宽度上保持不变。

现在,如果我们想对此运行转置卷积,那么我们期望的结果将通过以下等式描述:

假设在相同的张量上,我们运行转置卷积,其中stride s = 2,内核f = 3,padding = 1.然后我们期望的输出将是output = 9。

现在,如果我们使用stride = 1,内核大小f = 3并且padding = 1,那么输出的结果将与输入相同。

您可以在下面看到结果:

用转置卷积构建网络

现在我利用转置卷积的输出,其中正确的设置保持与输入相同。这将是网络的隐藏层,并且由于平均池操作,到网络的输入图片将仅通过过渡层减少。

所以网络的 Dense Block用Python实现看起来像这样:

class Dense_Block_Transposed(nn.Module):

def __init__(self, in_channels):

super(Dense_Block_Transposed, self).__init__()

self.relu = nn.LeakyReLU(inplace = True)

self.bn = nn.BatchNorm2d(num_features = in_channels)

self.conv1 = nn.ConvTranspose2d(in_channels = in_channels, out_channels = 32, kernel_size = 3, stride = 1, padding = 1)

self.conv2 = nn.ConvTranspose2d(in_channels = 32, out_channels = 32, kernel_size = 3, stride = 1, padding = 1)

self.conv3 = nn.ConvTranspose2d(in_channels = 64, out_channels = 32, kernel_size = 3, stride = 1, padding = 1)

self.conv4 = nn.ConvTranspose2d(in_channels = 96, out_channels = 32, kernel_size = 3, stride = 1, padding = 1)

self.conv5 = nn.ConvTranspose2d(in_channels = 128, out_channels = 32, kernel_size = 3, stride = 1, padding = 1)

def forward(self, x):

bn = self.bn(x)

conv1 = self.relu(self.conv1(bn))

conv2 = self.relu(self.conv2(conv1))

c2_dense = self.relu(torch.cat([conv1, conv2], 1))

conv3 = self.relu(self.conv3(c2_dense))

c3_dense = self.relu(torch.cat([conv1, conv2, conv3], 1))

conv4 = self.relu(self.conv4(c3_dense))

c4_dense = self.relu(torch.cat([conv1, conv2, conv3, conv4], 1))

conv5 = self.relu(self.conv5(c4_dense))

c5_dense = self.relu(torch.cat([conv1, conv2, conv3, conv4, conv5], 1))

return c5_dense

这里,图像的高度和宽度尺寸不会改变,只有通道尺寸增加。

网络的下一个构建块是Transition Layer。

所以Transition Layer的Python代码如下所示:

class Transition_Layer_Transposed(nn.Module):

def __init__(self, in_channels, out_channels):

super(Transition_Layer_Transposed, self).__init__()

self.relu = nn.LeakyReLU(inplace = True)

self.bn = nn.BatchNorm2d(num_features = in_channels)

self.conv = nn.ConvTranspose2d(in_channels = in_channels, out_channels = out_channels, kernel_size = 1, bias = False)

self.avg_pool = nn.AvgPool2d(kernel_size = 2, stride = 2, padding = 0)

def forward(self, x):

bn = self.bn(self.relu(self.conv(x)))

out = self.avg_pool(bn)

return out

与常规的卷积

现在,我要在CIFAR10上训练这个模型,用随机梯度下降和完全相同的超参数训练这两个网络。我对60个epochs进行了优化,并在训练数据上画出了损失。

对每个类的精度分别进行测量

现在让我们看看这些网络对测试数据的表现,以及它们在每个类中的准确性:

结论

通过上述实验,转置卷积似乎可以学习有意义的结果,并且具有与常规卷积相当的准确度,即使对于分类任务也是如此。

但是对于转置卷积更多地用于上采样目的,例如在生成对抗网络中。但正如本文所建议的那样,更好地使用上采样层,例如运行双线性插值以进行上采样,然后进行卷积。

相关推荐

3分钟让你的项目支持AI问答模块,完全开源!

hello,大家好,我是徐小夕。之前和大家分享了很多可视化,零代码和前端工程化的最佳实践,今天继续分享一下最近开源的Next-Admin的最新更新。最近对这个项目做了一些优化,并集成了大家比较关注...

干货|程序员的副业挂,12个平台分享

1、D2adminD2Admin是一个完全开源免费的企业中后台产品前端集成方案,使用最新的前端技术栈,小于60kb的本地首屏js加载,已经做好大部分项目前期准备工作,并且带有大量示例代码,助...

Github标星超200K,这10个可视化面板你知道几个

在Github上有很多开源免费的后台控制面板可以选择,但是哪些才是最好、最受欢迎的可视化控制面板呢?今天就和大家推荐Github上10个好看又流行的可视化面板:1.AdminLTEAdminLTE是...

开箱即用的炫酷中后台前端开源框架第二篇

#头条创作挑战赛#1、SoybeanAdmin(1)介绍:SoybeanAdmin是一个基于Vue3、Vite3、TypeScript、NaiveUI、Pinia和UnoCSS的清新优...

搭建React+AntDeign的开发环境和框架

搭建React+AntDeign的开发环境和框架随着前端技术的不断发展,React和AntDesign已经成为越来越多Web应用程序的首选开发框架。React是一个用于构建用户界面的JavaScrip...

基于.NET 5实现的开源通用权限管理平台

??大家好,我是为广大程序员兄弟操碎了心的小编,每天推荐一个小工具/源码,装满你的收藏夹,每天分享一个小技巧,让你轻松节省开发效率,实现不加班不熬夜不掉头发,是我的目标!??今天小编推荐一款基于.NE...

StreamPark - 大数据流计算引擎

使用Docker完成StreamPark的部署??1.基于h2和docker-compose进行StreamPark部署wgethttps://raw.githubusercontent.com/a...

教你使用UmiJS框架开发React

1、什么是Umi.js?umi,中文可发音为乌米,是一个可插拔的企业级react应用框架。你可以将它简单地理解为一个专注性能的类next.js前端框架,并通过约定、自动生成和解析代码等方式来辅助...

简单在线流程图工具在用例设计中的运用

敏捷模式下,测试团队的用例逐渐简化以适应快速的发版节奏,大家很早就开始运用思维导图工具比如xmind来编写测试方法、测试点。如今不少已经不少利用开源的思维导图组件(如百度脑图...)来构建测试测试...

【开源分享】神奇的大数据实时平台框架,让Flink&Spark开发更简单

这是一个神奇的框架,让Flink|Spark开发更简单,一站式大数据实时平台!他就是StreamX!什么是StreamX大数据技术如今发展的如火如荼,已经呈现百花齐放欣欣向荣的景象,实时处理流域...

聊聊规则引擎的调研及实现全过程

摘要本期主要以规则引擎业务实现为例,陈述在陌生业务前如何进行业务深入、调研、技术选型、设计及实现全过程分析,如果你对规则引擎不感冒、也可以从中了解一些抽象实现过程。诉求从硬件采集到的数据提供的形式多种...

【开源推荐】Diboot 2.0.5 发布,自动化开发助理

一、前言Diboot2.0.5版本已于近日发布,在此次发布中,我们新增了file-starter组件,完善了iam-starter组件,对core核心进行了相关优化,让devtools也支持对IAM...

微软推出Copilot Actions,使用人工智能自动执行重复性任务

IT之家11月19日消息,微软在今天举办的Ignite大会上宣布了一系列新功能,旨在进一步提升Microsoft365Copilot的智能化水平。其中最引人注目的是Copilot...

Electron 使用Selenium和WebDriver

本节我们来学习如何在Electron下使用Selenium和WebDriver。SeleniumSelenium是ThoughtWorks提供的一个强大的基于浏览器的开源自动化测试工具...

Quick 'n Easy Web Builder 11.1.0设计和构建功能齐全的网页的工具

一个实用而有效的应用程序,能够让您轻松构建、创建和设计个人的HTML网站。Quick'nEasyWebBuilder是一款全面且轻巧的软件,为用户提供了一种简单的方式来创建、编辑...