- 博客(1)
- 收藏
- 关注
原创 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
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人