Commit e4264ec7 authored by suilin0432's avatar suilin0432
Browse files

update

parent 07f26afb
_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
......@@ -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"] = {
......
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
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
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