update

configs/COCO-InstanceSegmentation/voc_irn.yaml

+_BASE_: "../Base-RCNN-FPN.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  MASK_ON: True
+  RESNETS:
+    DEPTH: 50
+DATASETS:
+  TRAIN: ("voc_2012_seg_train_irn",)
+  TEST: ("voc_2012_seg_val",)
+SOLVER:
+  BASE_LR: 0.008
+  WEIGHT_DECAY: 0.0001
+  IMS_PER_BATCH: 8
+  STEPS: (16000, )
+  MAX_ITER: 20000
+  CHECKPOINT_PERIOD: 4000
\ No newline at end of file

detectron2/data/datasets/builtin.py

@@ -52,6 +52,10 @@ _PREDEFINED_SPLITS_COCO["coco"] = {
     "coco_2017_gt_random_30_1": ("coco2017/train2017", "coco2017/annotations/random30_1.json"),
     "coco_2017_gt_random_30_2": ("coco2017/train2017", "coco2017/annotations/random30_2.json"),
     "coco_2017_gt_random_30_3": ("coco2017/train2017", "coco2017/annotations/random30_3.json"),
+    "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")
 }
 
 _PREDEFINED_SPLITS_COCO["coco_person"] = {

run/WSIS/voc_irn.sh

+python tools/train_net.py \
+--config-file configs/COCO-InstanceSegmentation/voc_irn.yaml \
+--num-gpus 8 --dist-url tcp://0.0.0.0:14192 --resume \
+OUTPUT_DIR output/wsis/voc_only_2w_008 \
+SOLVER.BASE_LR 0.008

run/WSIS/voc_irn_v100.sh

+CUDA_VISIBLE_DEVICES=4,5,6,7 python tools/train_net.py \
+--config-file configs/COCO-InstanceSegmentation/voc_irn.yaml \
+--num-gpus 4 --dist-url tcp://0.0.0.0:14192 --resume \
+OUTPUT_DIR output/wsis/voc_only_2w_008 \
+SOLVER.BASE_LR 0.008