add

detectron2/config/defaults.py

@@ -635,6 +635,15 @@ _C.MODEL.VGG.OUT_FEATURES = ["plain5"]
 _C.MODEL.VGG.CONV5_DILATION = 1
 _C.MODEL.VGG.FPN = False
 
+# detectron 的 VGG 的参数
+# _C.MODEL.VGG = CN()
+# _C.MODEL.VGG.DEPTH = 16
+# _C.MODEL.VGG.OUT_FEATURES = ["vgg_block5"]
+# # Options: FrozenBN, GN, "SyncBN", "BN"
+# _C.MODEL.VGG.NORM = "FrozenBN"
+# # Output channels of conv5 block
+# _C.MODEL.VGG.CONV5_OUT_CHANNELS = 512
+
 # VGG BOXHEAD 参数设置
 _C.MODEL.ROI_BOX_HEAD.DAN_DIM = [4096, 4096]
 

detectron2/data/detection_utils.py

@@ -596,6 +596,9 @@ def build_augmentation(cfg, is_train):
                 vertical=cfg.INPUT.RANDOM_FLIP == "vertical",
             )
         )
+        if cfg.SEMISUPNET.WEAK_AUG:
+            augmentation.append(T.RandomBrightness(1.0 / 1.5, 1.5))
+            augmentation.append(T.RandomSaturation(1.0 / 1.5, 1.5))
     return augmentation
 
 

detectron2/modeling/roi_heads/box_head_vgg.py

@@ -89,7 +89,7 @@ class DiscriminativeAdaptionNeck(nn.Module):
             for layer in self.fcs:
                 x = F.relu_(layer(x))
                 # TODO: 记得改回 p=0.5
-                x = F.dropout(x, p=0.5, training=self.training, inplace=False)
+                # x = F.dropout(x, p=0.5, training=self.training, inplace=False)
         return x
 
     @property