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

update

上级 04e4446e
......@@ -15,4 +15,6 @@ SOLVER:
IMS_PER_BATCH: 8
STEPS: (16000, )
MAX_ITER: 20000
CHECKPOINT_PERIOD: 4000
\ No newline at end of file
CHECKPOINT_PERIOD: 4000
TEST:
EVAL_PERIOD: 1000
......@@ -596,9 +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))
#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
......
支持 Markdown
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册