通过将ResNet结构中的3x3卷积替换成CoT模块,作者提出了CoTNet,并在分类、检测、分割任务上都做了实验,都取得了不错的性能。 本文通过PaddlePaddle复现CoTNet50,并取得不错的效果. 请点击此处查看本环境基本用法. Please click here for more detailed instructions.关于...
# 构建CoTNetLayerclass CoTNetLayer(nn.Layer): def __init__(self, dim=512, kernel_size=3): super().__init__() self.dim = dim self.kernel_size = kernel_size self.key_embed = nn.Sequential(# 通过K*K的卷积提取上下文信息,视作输入X的静态上下文表达nn.Conv2D(dim, dim, kernel_size=kern...