提交 bde5a41c 编辑于 作者: suilin0432's avatar suilin0432
浏览文件

add

上级 2393e648
......@@ -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]
......
......@@ -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
......
......@@ -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
......
支持 Markdown
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册