diff --git a/lib/model/faster_rcnn/resnet.py b/lib/model/faster_rcnn/resnet.py index 5457724b3..45d38d878 100644 --- a/lib/model/faster_rcnn/resnet.py +++ b/lib/model/faster_rcnn/resnet.py @@ -106,7 +106,7 @@ class ResNet(nn.Module): def __init__(self, block, layers, num_classes=1000): self.inplanes = 64 super(ResNet, self).__init__() - self.conv1 = nn.Conv2d(3, 64, kernel_size=7, stride=2, padding=3, + self.conv1 = nn.Conv2d(3, self.inplanes, kernel_size=7, stride=2, padding=3, bias=False) self.bn1 = nn.BatchNorm2d(64) self.relu = nn.ReLU(inplace=True)