From 459d5ebca256129cd2101cac9459b3aa7cb73afd Mon Sep 17 00:00:00 2001 From: Kai Chen <chenkaidev@gmail.com> Date: Fri, 12 Oct 2018 16:06:31 +0800 Subject: [PATCH] minor fix --- mmdet/models/builder.py | 3 ++- mmdet/models/rpn_heads/rpn_head.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/mmdet/models/builder.py b/mmdet/models/builder.py index bdf0ac3..ee5ae0b 100644 --- a/mmdet/models/builder.py +++ b/mmdet/models/builder.py @@ -2,7 +2,7 @@ from mmcv.runner import obj_from_dict from torch import nn from . import (backbones, necks, roi_extractors, rpn_heads, bbox_heads, - mask_heads, detectors) + mask_heads) __all__ = [ 'build_backbone', 'build_neck', 'build_rpn_head', 'build_roi_extractor', @@ -48,4 +48,5 @@ def build_mask_head(cfg): def build_detector(cfg, train_cfg=None, test_cfg=None): + from . import detectors return build(cfg, detectors, dict(train_cfg=train_cfg, test_cfg=test_cfg)) diff --git a/mmdet/models/rpn_heads/rpn_head.py b/mmdet/models/rpn_heads/rpn_head.py index e67d7ae..61e6e19 100644 --- a/mmdet/models/rpn_heads/rpn_head.py +++ b/mmdet/models/rpn_heads/rpn_head.py @@ -48,8 +48,8 @@ class RPNHead(nn.Module): self.anchor_scales = anchor_scales self.anchor_ratios = anchor_ratios self.anchor_strides = anchor_strides - self.anchor_base_sizes = anchor_strides.copy( - ) if anchor_base_sizes is None else anchor_base_sizes + self.anchor_base_sizes = list( + anchor_strides) if anchor_base_sizes is None else anchor_base_sizes self.target_means = target_means self.target_stds = target_stds self.use_sigmoid_cls = use_sigmoid_cls -- GitLab