Commit acb806d1 authored by suilin0432's avatar suilin0432
Browse files

update

parent 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 is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment