Non-local block 即插即用
import torchimport torch.nn as nnimport torchvisionclass NonLocalBlock(nn.Module): def __init__(self, channel): super(NonLocalBlock, self).__init__() self.inter_channel = channel // 2 self.conv_phi = nn.Con...
原创
2021-04-20 20:51:41 ·
2720 阅读 ·
1 评论