目录
一、ASPP
理论部分
论文地址:https://arxiv.org/pdf/1606.00915.pdf
ASPP网络结构
代码
class ASPP(nn.Module):
def __init__(self, in_channel, depth=256):
super(ASPP, self).__init__()
self.mean = nn.AdaptiveAvgPool2d((1, 1)) # (1,1)means ouput_dim