池化层【马赛克】

最大池化,也叫下采样。

公式:

参数:

celling的意思:

例子:

尺寸:

code:

复制代码
import torchimport torchvision.datasetsfrom tensorboardX import SummaryWriterfrom torch import nnfrom torch.nn import MaxPool2dfrom torch.utils.data import DataLoader​dataset = torchvision.datasets.CIFAR10("data",train=False,download=True,transform=torchvision.transforms.ToTensor())dataloader = DataLoader(dataset,batch_size=64)​input = torch.tensor([[1,2,0,3,1],                      [0,1,2,3,1],                      [1,2,1,0,0],                      [5,2,3,1,1],                      [2,1,0,1,1]],dtype=torch.float32)​input = torch.reshape(input,(-1,1,5,5))print(input.shape)​class Tudui(nn.Module):    def __init__(self):        super(Tudui,self).__init__()        self.maxpool1 = MaxPool2d(kernel_size=3,ceil_mode = True)    def forward(self,input):        output = self.maxpool1(input)        return  output​tudui = Tudui()output = tudui(input)print(output)​writer = SummaryWriter("logs_maxpool")step = 0for data in dataloader:    imgs,targets = data    writer.add_images("input",imgs,step)    output=tudui(imgs)    writer.add_images("output",output,step)    step = step+1writer.close()
相关推荐
Python图像识别6 分钟前
63_基于深度学习的草莓病害检测识别系统(yolo11、yolov8、yolov5+UI界面+Python项目源码+模型+标注好的数据集)
python·深度学习·yolo
十六点五1 小时前
Java NIO的底层原理
java·开发语言·python
跟橙姐学代码1 小时前
不要再用 print() 了!Python logging 库才是调试的终极武器
前端·python
小叶lr2 小时前
python 从pycharm部署到新环境
开发语言·python·pycharm
2301_763471032 小时前
Python单元测试(unittest)实战指南
python
xiaoxiongip6662 小时前
动态ip适合挂什么项目
网络·爬虫·python·网络协议·tcp/ip·ip
Q_Q5110082852 小时前
springboot+python+uniapp基于微信小程序的旅游服务系统景点信息展示 路线推荐 在线预约 评论互动系统
spring boot·python·微信小程序·django·flask·uni-app
天下无敌笨笨熊2 小时前
一些常用的激活函数及绘图
深度学习
E___V___E3 小时前
设计模式--装饰器模式
python·设计模式·装饰器模式
格林威3 小时前
短波红外相机在机器视觉检测方向的应用
运维·人工智能·深度学习·数码相机·计算机视觉·视觉检测