mirrors
/
OWOD
mirror of https://github.com/JosephKJ/OWOD.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Updates for MBZUAI server

Nov3
Joseph 2020-10-21 17:54:55 +05:30
parent f8efa40972
commit d622322332
2 changed files with 4 additions and 3 deletions
configs/PascalVOC-Detection
detectron2/data/datasets

4
configs/PascalVOC-Detection/faster_rcnn_R_50_C4.yaml
View File

@ -5,8 +5,8 @@ MODEL:
RESNETS:
DEPTH: 50
ROI_HEADS:
NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
# NUM_CLASSES: 20
# NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
NUM_CLASSES: 20
INPUT:
MIN_SIZE_TRAIN: (480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800)
MIN_SIZE_TEST: 800

3
detectron2/data/datasets/builtin.py
View File

@ -247,5 +247,6 @@ if __name__.endswith(".builtin"):
register_all_cityscapes(_root)
register_all_cityscapes_panoptic(_root)
# register_all_pascal_voc(_root)
register_all_pascal_voc('/home/joseph/workspace/OWOD/datasets')
# register_all_pascal_voc('/home/joseph/workspace/OWOD/datasets')
register_all_pascal_voc('/home/fk1/workspace/OWOD/datasets')
register_all_ade20k(_root)