mirror of https://github.com/YifanXu74/MQ-Det.git
155 lines
7.1 KiB
YAML
155 lines
7.1 KiB
YAML
DATALOADER:
|
|
ASPECT_RATIO_GROUPING: false
|
|
SIZE_DIVISIBILITY: 32
|
|
DATASETS:
|
|
GENERAL_COPY: 16
|
|
OVERRIDE_CATEGORY: '[{"id": 1, "name": "Ambulance", "supercategory": "vehicles"},
|
|
{"id": 2, "name": "Bus", "supercategory": "vehicles"}, {"id": 3, "name": "Car",
|
|
"supercategory": "vehicles"}, {"id": 4, "name": "Motorcycle", "supercategory":
|
|
"vehicles"}, {"id": 5, "name": "Truck", "supercategory": "vehicles"}]'
|
|
PREDEFINED_TEXT: odinw/pothole/category_description.json
|
|
REGISTER:
|
|
minival:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/mini_val/annotations_without_background.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/mini_val
|
|
minival_10_3:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/mini_val/fewshot_minival_shot10_seed3.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/mini_val
|
|
minival_10_30:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/mini_val/fewshot_minival_shot10_seed30.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/mini_val
|
|
minival_10_300:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/mini_val/fewshot_minival_shot10_seed300.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/mini_val
|
|
minival_1_3:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/mini_val/fewshot_minival_shot1_seed3.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/mini_val
|
|
minival_1_30:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/mini_val/fewshot_minival_shot1_seed30.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/mini_val
|
|
minival_1_300:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/mini_val/fewshot_minival_shot1_seed300.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/mini_val
|
|
minival_3_3:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/mini_val/fewshot_minival_shot3_seed3.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/mini_val
|
|
minival_3_30:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/mini_val/fewshot_minival_shot3_seed30.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/mini_val
|
|
minival_3_300:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/mini_val/fewshot_minival_shot3_seed300.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/mini_val
|
|
minival_5_3:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/mini_val/fewshot_minival_shot5_seed3.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/mini_val
|
|
minival_5_30:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/mini_val/fewshot_minival_shot5_seed30.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/mini_val
|
|
minival_5_300:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/mini_val/fewshot_minival_shot5_seed300.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/mini_val
|
|
test:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/test/annotations_without_background.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/test
|
|
train:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/train/annotations_without_background.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/train
|
|
train_10_3:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/train/fewshot_train_shot10_seed3.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/train
|
|
train_10_30:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/train/fewshot_train_shot10_seed30.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/train
|
|
train_10_300:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/train/fewshot_train_shot10_seed300.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/train
|
|
train_1_3:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/train/fewshot_train_shot1_seed3.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/train
|
|
train_1_30:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/train/fewshot_train_shot1_seed30.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/train
|
|
train_1_300:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/train/fewshot_train_shot1_seed300.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/train
|
|
train_3_3:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/train/fewshot_train_shot3_seed3.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/train
|
|
train_3_30:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/train/fewshot_train_shot3_seed30.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/train
|
|
train_3_300:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/train/fewshot_train_shot3_seed300.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/train
|
|
train_5_3:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/train/fewshot_train_shot5_seed3.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/train
|
|
train_5_30:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/train/fewshot_train_shot5_seed30.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/train
|
|
train_5_300:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/train/fewshot_train_shot5_seed300.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/train
|
|
val:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/valid/annotations_without_background.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/valid
|
|
val_10_3:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/valid/fewshot_val_shot10_seed3.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/valid
|
|
val_10_30:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/valid/fewshot_val_shot10_seed30.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/valid
|
|
val_10_300:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/valid/fewshot_val_shot10_seed300.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/valid
|
|
val_1_3:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/valid/fewshot_val_shot1_seed3.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/valid
|
|
val_1_30:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/valid/fewshot_val_shot1_seed30.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/valid
|
|
val_1_300:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/valid/fewshot_val_shot1_seed300.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/valid
|
|
val_3_3:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/valid/fewshot_val_shot3_seed3.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/valid
|
|
val_3_30:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/valid/fewshot_val_shot3_seed30.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/valid
|
|
val_3_300:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/valid/fewshot_val_shot3_seed300.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/valid
|
|
val_5_3:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/valid/fewshot_val_shot5_seed3.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/valid
|
|
val_5_30:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/valid/fewshot_val_shot5_seed30.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/valid
|
|
val_5_300:
|
|
ann_file: odinw/VehiclesOpenImages/416x416/valid/fewshot_val_shot5_seed300.json
|
|
img_dir: odinw/VehiclesOpenImages/416x416/valid
|
|
TEST: ("minival",)
|
|
TRAIN: ("train",)
|
|
INPUT:
|
|
MAX_SIZE_TEST: 1333
|
|
MAX_SIZE_TRAIN: 1333
|
|
MIN_SIZE_TEST: 800
|
|
MIN_SIZE_TRAIN: 800
|
|
MODEL:
|
|
ATSS:
|
|
NUM_CLASSES: 6
|
|
DYHEAD:
|
|
NUM_CLASSES: 6
|
|
FCOS:
|
|
NUM_CLASSES: 6
|
|
ROI_BOX_HEAD:
|
|
NUM_CLASSES: 6
|
|
SOLVER:
|
|
CHECKPOINT_PERIOD: 100
|
|
MAX_EPOCH: 12
|
|
WARMUP_ITERS: 0
|
|
TEST:
|
|
IMS_PER_BATCH: 8
|
|
VISION_QUERY:
|
|
DATASET_NAME: 'Vehicles' |