对比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层=传递特征+显著特征+细节特征的拼接。其他经典神经网络就不一一对比了。

相关推荐
爱可生开源社区5 分钟前
SCALE | 重构 AI 时代数据库能力的全新评估标准
人工智能
Jahzo26 分钟前
openclaw本地化部署体验与踩坑记录--飞书机器人配置
人工智能·开源
Narrastory26 分钟前
明日香 - Pytorch 快速入门保姆级教程(一)
人工智能·pytorch·深度学习
数据智能老司机27 分钟前
用于进攻性网络安全的智能体 AI——在 n8n 中构建你的第一个 AI 工作流
人工智能·安全·agent
数据智能老司机30 分钟前
用于进攻性网络安全的智能体 AI——智能体 AI 入门
人工智能·安全·agent
Narrastory31 分钟前
明日香 - Pytorch 快速入门保姆级教程(二)
人工智能·pytorch·深度学习
AI攻城狮32 分钟前
OpenClaw Session 管理完全指南:Context 压缩、重置与持久化
人工智能·云原生·aigc
中杯可乐多加冰2 小时前
OpenClaw到底能做什么?有什么用?先装这几个实用的Skills
人工智能
千寻girling2 小时前
一份不可多得的 《 Python 》语言教程
人工智能·后端·python
aircrushin4 小时前
从春晚看分布式实时协同算法与灵巧手工程实现
人工智能·机器人