Merge pull request #73 from hellock/hotfix

bug fix for init sampler during testing

mmdet/datasets/loader/build_loader.py

@@ -25,13 +25,13 @@ def build_dataloader(dataset,
         batch_size = imgs_per_gpu
         num_workers = workers_per_gpu
     else:
-        sampler = GroupSampler(dataset, imgs_per_gpu)
+        if not kwargs.get('shuffle', True):
+            sampler = None
+        else:
+            sampler = GroupSampler(dataset, imgs_per_gpu)
         batch_size = num_gpus * imgs_per_gpu
         num_workers = num_gpus * workers_per_gpu
 
-    if not kwargs.get('shuffle', True):
-        sampler = None
-
     data_loader = DataLoader(
         dataset,
         batch_size=batch_size,

mmdet/models/bbox_heads/convfc_bbox_head.py

@@ -11,7 +11,7 @@ class ConvFCBBoxHead(BBoxHead):
                                 /-> cls convs -> cls fcs -> cls
     shared convs -> shared fcs
                                 \-> reg convs -> reg fcs -> reg
-    """
+    """  # noqa: W605
 
     def __init__(self,
                  num_shared_convs=0,

mmdet/models/rpn_heads/rpn_head.py

@@ -30,7 +30,7 @@ class RPNHead(nn.Module):
         target_stds (Iterable): Std values of regression targets.
         use_sigmoid_cls (bool): Whether to use sigmoid loss for classification.
             (softmax by default)
-    """
+    """  # noqa: W605
 
     def __init__(self,
                  in_channels,