Update to 0.5.1

README.md

@@ -32,6 +32,12 @@ which is heavily depended on by this toolbox.
 
 This project is released under the [GPLv3 license](LICENSE).
 
+## Updates
+
+v0.5.1 (20/10/2018)
+- Add BBoxAssigner and BBoxSampler, the `train_cfg` field in config files are restructured.
+- `ConvFCRoIHead` / `SharedFCRoIHead` are renamed to `ConvFCBBoxHead` / `SharedFCBBoxHead` for consistency.
+
 ## Benchmark and model zoo
 
 We provide our baseline results and the comparision with Detectron, the most

configs/fast_mask_rcnn_r50_fpn_1x.py

@@ -20,7 +20,7 @@
         out_channels=256,
         featmap_strides=[4, 8, 16, 32]),
     bbox_head=dict(
-        type='SharedFCRoIHead',
+        type='SharedFCBBoxHead',
         num_fcs=2,
         in_channels=256,
         fc_out_channels=1024,

configs/fast_rcnn_r50_fpn_1x.py

@@ -20,7 +20,7 @@
         out_channels=256,
         featmap_strides=[4, 8, 16, 32]),
     bbox_head=dict(
-        type='SharedFCRoIHead',
+        type='SharedFCBBoxHead',
         num_fcs=2,
         in_channels=256,
         fc_out_channels=1024,

configs/faster_rcnn_r50_fpn_1x.py

@@ -30,7 +30,7 @@
         out_channels=256,
         featmap_strides=[4, 8, 16, 32]),
     bbox_head=dict(
-        type='SharedFCRoIHead',
+        type='SharedFCBBoxHead',
         num_fcs=2,
         in_channels=256,
         fc_out_channels=1024,

configs/mask_rcnn_r50_fpn_1x.py

@@ -30,7 +30,7 @@
         out_channels=256,
         featmap_strides=[4, 8, 16, 32]),
     bbox_head=dict(
-        type='SharedFCRoIHead',
+        type='SharedFCBBoxHead',
         num_fcs=2,
         in_channels=256,
         fc_out_channels=1024,

mmdet/models/bbox_heads/__init__.py

@@ -1,4 +1,4 @@
 from .bbox_head import BBoxHead
-from .convfc_bbox_head import ConvFCRoIHead, SharedFCRoIHead
+from .convfc_bbox_head import ConvFCBBoxHead, SharedFCBBoxHead
 
-__all__ = ['BBoxHead', 'ConvFCRoIHead', 'SharedFCRoIHead']
+__all__ = ['BBoxHead', 'ConvFCBBoxHead', 'SharedFCBBoxHead']

mmdet/models/bbox_heads/convfc_bbox_head.py

@@ -4,7 +4,7 @@
 from ..utils import ConvModule
 
 
-class ConvFCRoIHead(BBoxHead):
+class ConvFCBBoxHead(BBoxHead):
     """More general bbox head, with shared conv and fc layers and two optional
     separated branches.
 
@@ -24,7 +24,7 @@ def __init__(self,
                  fc_out_channels=1024,
                  *args,
                  **kwargs):
-        super(ConvFCRoIHead, self).__init__(*args, **kwargs)
+        super(ConvFCBBoxHead, self).__init__(*args, **kwargs)
         assert (num_shared_convs + num_shared_fcs + num_cls_convs + num_cls_fcs
                 + num_reg_convs + num_reg_fcs > 0)
         if num_cls_convs > 0 or num_reg_convs > 0:
@@ -116,7 +116,7 @@ def _add_conv_fc_branch(self,
         return branch_convs, branch_fcs, last_layer_dim
 
     def init_weights(self):
-        super(ConvFCRoIHead, self).init_weights()
+        super(ConvFCBBoxHead, self).init_weights()
         for module_list in [self.shared_fcs, self.cls_fcs, self.reg_fcs]:
             for m in module_list.modules():
                 if isinstance(m, nn.Linear):
@@ -162,11 +162,11 @@ def forward(self, x):
         return cls_score, bbox_pred
 
 
-class SharedFCRoIHead(ConvFCRoIHead):
+class SharedFCBBoxHead(ConvFCBBoxHead):
 
     def __init__(self, num_fcs=2, fc_out_channels=1024, *args, **kwargs):
         assert num_fcs >= 1
-        super(SharedFCRoIHead, self).__init__(
+        super(SharedFCBBoxHead, self).__init__(
             num_shared_convs=0,
             num_shared_fcs=num_fcs,
             num_cls_convs=0,

setup.py

@@ -12,7 +12,7 @@ def readme():
 
 MAJOR = 0
 MINOR = 5
-PATCH = 0
+PATCH = 1
 SUFFIX = ''
 SHORT_VERSION = '{}.{}.{}{}'.format(MAJOR, MINOR, PATCH, SUFFIX)