From 60555f6b27c8396fb92b50c10902347fb0addcc1 Mon Sep 17 00:00:00 2001 From: Erik Date: Mon, 13 Jan 2020 23:01:23 +0100 Subject: [PATCH] noticed hardcoded inplanes under ResNet implementation If you compare this with the PyTorch implementation of ResNet, you notice the hardcoding --- lib/model/faster_rcnn/resnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)