Commit 0ebeedc8 authored by suilin0432's avatar suilin0432
Browse files

update

parent 4b75c01c
_BASE_: "../Base-RCNN-FPN.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
MASK_ON: True
RESNETS:
DEPTH: 50
ROI_HEADS:
NUM_CLASSES: 20
DATASETS:
TRAIN: ("voc_2012_seg_train_irn_plus_sbd",)
TEST: ("voc_2012_seg_val",)
SOLVER:
BASE_LR: 0.008
WEIGHT_DECAY: 0.0001
IMS_PER_BATCH: 8
STEPS: (8000, 11000)
MAX_ITER: 14000
CHECKPOINT_PERIOD: 4000
TEST:
EVAL_PERIOD: 1000
......@@ -36,7 +36,8 @@ _PREDEFINED_SPLITS_COCO["voc"] = {
"voc_2012_seg_train": ("VOC2012/JPEGImages", "VOC2012/voc_2012_train_instance.json"),
"voc_2012_seg_val": ("VOC2012/JPEGImages", "VOC2012/voc_2012_val_instance.json"),
"voc_2012_seg_train_irn": ("VOC2012/JPEGImages", "VOC2012/voc_2012_train_instance_irn.json"),
"sbd_seg": ("VOC_SBD/images", "VOC2012/sbd_9118_instance.json")
"sbd_seg": ("VOC_SBD/images", "VOC2012/sbd_9118_instance.json"),
"voc_2012_seg_train_irn_plus_sbd": ("VOC2012/voc_2012_train_instance_irn_plus_sbd.json")
}
_PREDEFINED_SPLITS_COCO["coco"] = {
"coco_2014_train": ("coco/train2014", "coco/annotations/instances_train2014.json"),
......
CUDA_VISIBLE_DEVICES=4,5,6,7 python tools/train_net.py \
--config-file configs/COCO-InstanceSegmentation/voc_irn_plus.yaml \
--num-gpus 4 --dist-url tcp://0.0.0.0:14192 --resume \
OUTPUT_DIR output/wsis/voc_plus_1_4w_008 \
SOLVER.BASE_LR 0.008
\ No newline at end of file
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