INNER CODE UNIT · Python
__init__
pprp/awesome-attention-mechanism-in-cv · Plug-and-Play-Module/ASPPBlock.py:40
def __init__(self, inplanes, planes, rate):
super(ASPP, self).__init__()
self.rate = rate
if rate == 1:
kernel_size = 1
padding = 0
else:
kernel_size = 3
padding = rate
#self.conv1 = nn.Conv2d(planes, planes, kernel_size=3, bias=False,padding=1)
self.conv1 = SeparableConv2d(planes, planes, 3, 1, 1)
self.bn1 = nn.BatchNorm2d(planes)
self.relu1 = nn.ReLU()
# self.atrous_convolution = nn.Conv2d(inplanes, planes, kernel_size=kernel_size,
# stride=1, padding=padding, dilation=rate, bias=False)
self.atrous_convolution = SeparableConv2d(inplanes, planes,
kernel_size, 1, padding,