Skip to content

Commit 459d5eb

Browse files
committed
minor fix
1 parent 0444489 commit 459d5eb

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

mmdet/models/builder.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from torch import nn
33

44
from . import (backbones, necks, roi_extractors, rpn_heads, bbox_heads,
5-
mask_heads, detectors)
5+
mask_heads)
66

77
__all__ = [
88
'build_backbone', 'build_neck', 'build_rpn_head', 'build_roi_extractor',
@@ -48,4 +48,5 @@ def build_mask_head(cfg):
4848

4949

5050
def build_detector(cfg, train_cfg=None, test_cfg=None):
51+
from . import detectors
5152
return build(cfg, detectors, dict(train_cfg=train_cfg, test_cfg=test_cfg))

mmdet/models/rpn_heads/rpn_head.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,8 @@ def __init__(self,
4848
self.anchor_scales = anchor_scales
4949
self.anchor_ratios = anchor_ratios
5050
self.anchor_strides = anchor_strides
51-
self.anchor_base_sizes = anchor_strides.copy(
52-
) if anchor_base_sizes is None else anchor_base_sizes
51+
self.anchor_base_sizes = list(
52+
anchor_strides) if anchor_base_sizes is None else anchor_base_sizes
5353
self.target_means = target_means
5454
self.target_stds = target_stds
5555
self.use_sigmoid_cls = use_sigmoid_cls

0 commit comments

Comments
 (0)