对比tensorflow,从0开始学pytorch(二)--多尺度实现

今天尝试着想实现了多尺度,终于发现了pytorch的好处。

一开始还吐槽为什么pytorch没有像tensorflow一样提供GAP和GMP这种最基本的层,才发现有AdaptiveAvgPool2d和AdaptiveMaxPool2d这种功能更加强大的层。

首先,用如下方式可以实现GAP或GMP

其次,SPP也可以用类似的方式实现,如下代码相当于spp的[2,3,4]

以后,拼接下隐层特征就行

代码有点丑,太长了,不知道是不是写的不正规,flatten层前写了一堆代码。GSM是N年前费好大劲想出的创新点,没想到pytorch实现起来如此简单。

改造完毕后,发现有点用,比flatten得到的结果好,原生LeNet 的结果96.xx%,有一点提升。GMS层=传递特征+显著特征+细节特征的拼接。其他经典神经网络就不一一对比了。

相关推荐
OpenBayes1 小时前
VibeVoice-Realtime TTS重构实时语音体验;覆盖9大真实场景,WenetSpeech-Chuan让模型听懂川话
人工智能·深度学习·数据集·图像识别·语音合成·图像生成·视频生成
光羽隹衡1 小时前
机器学习——线性回归
人工智能·机器学习·线性回归
zhongerzixunshi1 小时前
创新型中小企业申报条件详解
人工智能
Dev7z1 小时前
基于颜色特征与模板匹配融合决策的智能硬币识别系统
人工智能
市象1 小时前
WPS润色AI半成品
人工智能
老蒋新思维1 小时前
创客匠人 2025 万人峰会洞察:AI 时代创始人 IP 的能力进化 —— 知识变现的核心竞争力重构
网络·人工智能·tcp/ip·重构·数据挖掘·创始人ip·创客匠人
微三云、小叶1 小时前
酒店 RWA 模式深度拆解:资产轻量化与投资普惠化的双重突破路径
大数据·人工智能·区块链
AI科技星1 小时前
观察者与宇宙:描述如何创造物理实在
数据结构·人工智能·算法·机器学习·重构
ZKNOW甄知科技1 小时前
低代码 ITSM 知识管理平台:驱动企业数智化运维的新引擎
运维·服务器·人工智能·低代码·网络安全·自动化