【网络结构复现】之ACNet
网络出处请看这篇博客:基于注意力机制的深度学习路面裂缝检测
直接上代码了,不会的地方可以留言
AFM.py
from torch import nn
class ConvG1(nn.Module): #ConvG1采用 7×7 和 3×3 步长均为 2 的 2 种卷积核
def __init__(self):
super(ConvG1, self).__init__()
self.conv1 = nn.Conv2d(512,128,7,2,3)
原创
2022-04-12 20:52:02 ·
1920 阅读 ·
0 评论