模块出处
[link] [code] [PR 23] Cross-level Feature Aggregation Network for Polyp Segmentation
模块名称
Cross-level Feature Fusion (CFF)
模块作用
双级特征融合
模块结构
模块代码
import torch
import torch.nn as nn
class BasicConv2d(nn.Module):
def __init__(self, in_planes, out_planes, kernel_size, stride=1, padding=0, dilation=1):
super(BasicConv2d, self).__init__()
self.conv = nn.Conv2d(in_planes, out_planes,
kernel_size=kernel_size, stride=stride,
padding=padding, dilation=dilation, bias=False)
self.bn = nn.BatchNorm2d(out_planes)
self.relu = nn.ReLU(inplace=True)
def