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-23 12:03:02 +05:30
parent 7f8c54f2cb
commit e662289ad4
2 changed files with 33 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -15,12 +15,12 @@ DATASETS:
TRAIN: ('voc_2007_trainval', 'voc_2012_trainval')
TEST: ('voc_2007_test',)
SOLVER:
STEPS: (27000, 30000)
MAX_ITER: 36000 # 17.4 epochs
STEPS: (37000, 40000)
MAX_ITER: 46000 # 17.4 epochs
# STEPS: (12000, 16000)
# MAX_ITER: 18000 # 17.4 epochs
WARMUP_ITERS: 100
OUTPUT_DIR: "./output/baseline_run"
OUTPUT_DIR: "./output/baseline_run_46000"
OWOD:
ENABLE_THRESHOLD_AUTOLABEL_UNK: False
NUM_UNK_PER_IMAGE: 1

30
detectron2/data/datasets/pascal_voc.py
View File

@ -26,6 +26,36 @@ CLASS_NAMES = (
# )
# fmt: on
VOC_CLASS_NAMES = (
"aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat",
"chair", "cow", "diningtable", "dog", "horse", "motorbike", "person",
"pottedplant", "sheep", "sofa", "train", "tvmonitor"
)
T2_CLASS_NAMES = (
"truck", "trafficlight", "firehydrant", "stopsigh", "parkingmeter",
"bench", "elephant", "bear", "zebra", "giraffe",
"backpack", "umbrella", "handbag", "tie", "suitcase",
"microwave", "oven", "toaster", "sink", "refrigerator"
)
T3_CLASS_NAMES = (
"frisbee", "skis", "snowboard", "sportsball", "kite",
"baseballbat", "baseballglove", "skateboard", "surfboard", "tennisracket",
"banana", "apple", "sandwich", "orange", "broccoli",
"carrot", "hotdog", "pizza", "donut", "cake"
)
T4_CLASS_NAMES = (
"bed", "toilet", "tv", "laptop", "mouse",
"remote", "keyboard", "cellphone", "book", "clock",
"vase", "scissors", "teddybear", "hairdrier", "toothbrush",
"bottle", "wineglass", "cup", "fork", "knife", "spoon", "bowl"
)
UNK_CLASS = ("unknown")
def load_voc_instances(dirname: str, split: str, class_names: Union[List[str], Tuple[str, ...]]):
"""