def conv3x3(in_planes, out_planes, stride=1):
"3x3 convolution with padding"
return nn.Conv2d(in_planes, out_planes, kernel_size=3, stride=stride,
padding=1, bias=False)
和
def conv3x3_2(in_planes, out_planes, stride=1):
"3x3 convolution with padding"
return DeformConvPack(in_planes, out_planes, kernel_size=3, stride=stride,
padding=1, bias=False)
换成这个之后
self.conv2 = DeformConvPack(planes, planes, kernel_size=3, stride=stride,
padding=1, bias=False)