"...utils/git@developer.sourcefind.cn:modelzoo/S3Diff.git" did not exist on "76b9024b282206200ee82b441dbe08c2d9299f2f"
Commit 459d5ebc authored by Kai Chen's avatar Kai Chen
Browse files

minor fix

parent 04444895
...@@ -2,7 +2,7 @@ from mmcv.runner import obj_from_dict ...@@ -2,7 +2,7 @@ from mmcv.runner import obj_from_dict
from torch import nn from torch import nn
from . import (backbones, necks, roi_extractors, rpn_heads, bbox_heads, from . import (backbones, necks, roi_extractors, rpn_heads, bbox_heads,
mask_heads, detectors) mask_heads)
__all__ = [ __all__ = [
'build_backbone', 'build_neck', 'build_rpn_head', 'build_roi_extractor', 'build_backbone', 'build_neck', 'build_rpn_head', 'build_roi_extractor',
...@@ -48,4 +48,5 @@ def build_mask_head(cfg): ...@@ -48,4 +48,5 @@ def build_mask_head(cfg):
def build_detector(cfg, train_cfg=None, test_cfg=None): 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)) return build(cfg, detectors, dict(train_cfg=train_cfg, test_cfg=test_cfg))
...@@ -48,8 +48,8 @@ class RPNHead(nn.Module): ...@@ -48,8 +48,8 @@ class RPNHead(nn.Module):
self.anchor_scales = anchor_scales self.anchor_scales = anchor_scales
self.anchor_ratios = anchor_ratios self.anchor_ratios = anchor_ratios
self.anchor_strides = anchor_strides self.anchor_strides = anchor_strides
self.anchor_base_sizes = anchor_strides.copy( self.anchor_base_sizes = list(
) if anchor_base_sizes is None else anchor_base_sizes anchor_strides) if anchor_base_sizes is None else anchor_base_sizes
self.target_means = target_means self.target_means = target_means
self.target_stds = target_stds self.target_stds = target_stds
self.use_sigmoid_cls = use_sigmoid_cls self.use_sigmoid_cls = use_sigmoid_cls
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment