8000 GitHub - Coder-Liuu/MixPool2d: 使用PyTorch完成混合池化层
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

Coder-Liuu/MixPool2d

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Torch 实现混合池化层

前言

该idea由我的师兄的idea改进而来,纯属自己DIY玩一玩,当时自己写这个遇到了不少困难,将算法开源希望能帮住到更多的人。

距离实现idea的日子已经过去一年多了,自己最近重新翻改了一下。

如果您也想实现一个自己的池化层,不妨先看一下最大池化层是怎么实现的,这将会为你自定义池化层提供一定的帮助。

算法思想

对于图片中平坦地区进行模糊化,对图片中的边缘地区进行尖锐化。

算法步骤

1. 计算整张图片的均值
2. 计算池化块的均值
3. 计算池化块的最大值
4. if 若池化区域的均值 > 图片的整体均值
      该池化区域进行均值池化
   else:
      该池化区域进行最大值池化

About

使用PyTorch完成混合池化层

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

0