From 9b0bcb1c83c169bdcbd0a882d98e3517705eedf6 Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Sun, 23 Jun 2019 00:05:20 +0800
Subject: [PATCH] bug fix for x101 gcnet (#854)
---
mmdet/models/backbones/resnet.py | 18 +++++++++---------
mmdet/models/backbones/resnext.py | 4 ++--
2 files changed, 11 insertions(+), 11 deletions(-)
diff --git a/mmdet/models/backbones/resnet.py b/mmdet/models/backbones/resnet.py
index 3a8a613..abe1280 100644
--- a/mmdet/models/backbones/resnet.py
+++ b/mmdet/models/backbones/resnet.py
@@ -297,11 +297,11 @@ def make_res_layer(block,
layers = []
layers.append(
block(
- inplanes,
- planes,
- stride,
- dilation,
- downsample,
+ inplanes=inplanes,
+ planes=planes,
+ stride=stride,
+ dilation=dilation,
+ downsample=downsample,
style=style,
with_cp=with_cp,
conv_cfg=conv_cfg,
@@ -314,10 +314,10 @@ def make_res_layer(block,
for i in range(1, blocks):
layers.append(
block(
- inplanes,
- planes,
- 1,
- dilation,
+ inplanes=inplanes,
+ planes=planes,
+ stride=1,
+ dilation=dilation,
style=style,
with_cp=with_cp,
conv_cfg=conv_cfg,
diff --git a/mmdet/models/backbones/resnext.py b/mmdet/models/backbones/resnext.py
index b36fa9b..c0b42b2 100644
--- a/mmdet/models/backbones/resnext.py
+++ b/mmdet/models/backbones/resnext.py
@@ -11,12 +11,12 @@ from ..utils import build_conv_layer, build_norm_layer
class Bottleneck(_Bottleneck):
- def __init__(self, groups=1, base_width=4, *args, **kwargs):
+ def __init__(self, inplanes, planes, groups=1, base_width=4, **kwargs):
"""Bottleneck block for ResNeXt.
If style is "pytorch", the stride-two layer is the 3x3 conv layer,
if it is "caffe", the stride-two layer is the first 1x1 conv layer.
"""
- super(Bottleneck, self).__init__(*args, **kwargs)
+ super(Bottleneck, self).__init__(inplanes, planes, **kwargs)
if groups == 1:
width = self.planes
--
GitLab