Skip to content
GitLab
菜单
项目
群组
代码片段
/
帮助
帮助
支持
社区论坛
快捷键
?
提交反馈
登录/注册
切换导航
菜单
打开侧边栏
Lin Sui
detectron2
提交
bde5a41c
提交
bde5a41c
编辑于
8月 05, 2021
作者:
suilin0432
浏览文件
add
上级
2393e648
变更
3
Hide whitespace changes
Inline
Side-by-side
detectron2/config/defaults.py
浏览文件 @
bde5a41c
...
...
@@ -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
浏览文件 @
bde5a41c
...
...
@@ -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
浏览文件 @
bde5a41c
...
...
@@ -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.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录