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

相关推荐
yumgpkpm4 分钟前
Acceldata ODP(Open Data Platform)3.3.6.4(RHEL9)保姆级完整安装手册
大数据·人工智能·hive·hadoop·kafka·hbase·cloudera
邓工说电16 分钟前
智慧断路器安全吗?数据加密、离线保护与合规认证全解读
大数据·数据库·人工智能·智能断路器·炜晔科技
阿里云大数据AI技术18 分钟前
Lance 数据检索怎么选,当然阿里云 Milvus 向量湖
人工智能
出海客20 分钟前
跨境电商多语言客服知识库怎么建:资料结构、检索边界与人工升级
大数据·人工智能
xsd2024111837 分钟前
从自主导航到视觉读表:一台工业巡检机器人的全栈技术链路拆解
人工智能
袁哥大话安全42 分钟前
巡隐WEBSHELL扫描软件
人工智能·安全·web
论文复现现场1 小时前
8卡4090能跑70B吗?Llama-2显存预算、QLoRA与通信瓶颈
人工智能·深度学习·分布式训练·llama·显存·qlora·算家云
酷虎软件1 小时前
如何用AI创作AI歌曲AI音乐
人工智能
小宋10211 小时前
A2UI从零实战:Agent安全生成交互界面与事件回传
javascript·人工智能·安全·交互
罗小罗同学1 小时前
谷歌团队在Nat Med最新发表的医学多模态模型,4B参数的模型性能逼近671B的DeepSeek
人工智能·医学图像处理·医工交叉·医学ai