mirror of
https://github.com/YifanXu74/MQ-Det.git
synced 2025-06-03 15:03:07 +08:00
104 lines
7.2 KiB
YAML
104 lines
7.2 KiB
YAML
DATALOADER: {ASPECT_RATIO_GROUPING: false, SIZE_DIVISIBILITY: 32}
|
|
DATASETS:
|
|
GENERAL_COPY: 16
|
|
OVERRIDE_CATEGORY: '[{"id": 1, "name": "crab", "supercategory": "animals"}, {"id":
|
|
2, "name": "fish", "supercategory": "animals"}, {"id": 3, "name": "jellyfish",
|
|
"supercategory": "animals"}, {"id": 4, "name": "shrimp", "supercategory": "animals"},
|
|
{"id": 5, "name": "small_fish", "supercategory": "animals"}, {"id": 6, "name":
|
|
"starfish", "supercategory": "animals"}]'
|
|
PREDEFINED_TEXT: odinw/pothole/category_description.json
|
|
REGISTER:
|
|
minival: {ann_file: odinw/brackishUnderwater/960x540/mini_val/annotations_without_background.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/mini_val}
|
|
minival_10_3: {ann_file: odinw/brackishUnderwater/960x540/mini_val/fewshot_minival_shot10_seed3.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/mini_val}
|
|
minival_10_30: {ann_file: odinw/brackishUnderwater/960x540/mini_val/fewshot_minival_shot10_seed30.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/mini_val}
|
|
minival_10_300: {ann_file: odinw/brackishUnderwater/960x540/mini_val/fewshot_minival_shot10_seed300.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/mini_val}
|
|
minival_1_3: {ann_file: odinw/brackishUnderwater/960x540/mini_val/fewshot_minival_shot1_seed3.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/mini_val}
|
|
minival_1_30: {ann_file: odinw/brackishUnderwater/960x540/mini_val/fewshot_minival_shot1_seed30.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/mini_val}
|
|
minival_1_300: {ann_file: odinw/brackishUnderwater/960x540/mini_val/fewshot_minival_shot1_seed300.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/mini_val}
|
|
minival_3_3: {ann_file: odinw/brackishUnderwater/960x540/mini_val/fewshot_minival_shot3_seed3.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/mini_val}
|
|
minival_3_30: {ann_file: odinw/brackishUnderwater/960x540/mini_val/fewshot_minival_shot3_seed30.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/mini_val}
|
|
minival_3_300: {ann_file: odinw/brackishUnderwater/960x540/mini_val/fewshot_minival_shot3_seed300.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/mini_val}
|
|
minival_3shot_3seed: {ann_file: odinw/original/brackishUnderwater/960x540/mini_val/annotations_without_background_3shot_3seed.json,
|
|
img_dir: odinw/original/brackishUnderwater/960x540/mini_val}
|
|
minival_5_3: {ann_file: odinw/brackishUnderwater/960x540/mini_val/fewshot_minival_shot5_seed3.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/mini_val}
|
|
minival_5_30: {ann_file: odinw/brackishUnderwater/960x540/mini_val/fewshot_minival_shot5_seed30.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/mini_val}
|
|
minival_5_300: {ann_file: odinw/brackishUnderwater/960x540/mini_val/fewshot_minival_shot5_seed300.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/mini_val}
|
|
test: {ann_file: odinw/brackishUnderwater/960x540/test/annotations_without_background.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/test}
|
|
train: {ann_file: odinw/brackishUnderwater/960x540/train/annotations_without_background.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/train}
|
|
train_10_3: {ann_file: odinw/brackishUnderwater/960x540/train/fewshot_train_shot10_seed3.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/train}
|
|
train_10_30: {ann_file: odinw/brackishUnderwater/960x540/train/fewshot_train_shot10_seed30.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/train}
|
|
train_10_300: {ann_file: odinw/brackishUnderwater/960x540/train/fewshot_train_shot10_seed300.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/train}
|
|
train_1_3: {ann_file: odinw/brackishUnderwater/960x540/train/fewshot_train_shot1_seed3.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/train}
|
|
train_1_30: {ann_file: odinw/brackishUnderwater/960x540/train/fewshot_train_shot1_seed30.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/train}
|
|
train_1_300: {ann_file: odinw/brackishUnderwater/960x540/train/fewshot_train_shot1_seed300.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/train}
|
|
train_3_3: {ann_file: odinw/brackishUnderwater/960x540/train/fewshot_train_shot3_seed3.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/train}
|
|
train_3_30: {ann_file: odinw/brackishUnderwater/960x540/train/fewshot_train_shot3_seed30.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/train}
|
|
train_3_300: {ann_file: odinw/brackishUnderwater/960x540/train/fewshot_train_shot3_seed300.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/train}
|
|
train_5_3: {ann_file: odinw/brackishUnderwater/960x540/train/fewshot_train_shot5_seed3.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/train}
|
|
train_5_30: {ann_file: odinw/brackishUnderwater/960x540/train/fewshot_train_shot5_seed30.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/train}
|
|
train_5_300: {ann_file: odinw/brackishUnderwater/960x540/train/fewshot_train_shot5_seed300.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/train}
|
|
val: {ann_file: odinw/brackishUnderwater/960x540/valid/annotations_without_background.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/valid}
|
|
val_10_3: {ann_file: odinw/brackishUnderwater/960x540/valid/fewshot_val_shot10_seed3.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/valid}
|
|
val_10_30: {ann_file: odinw/brackishUnderwater/960x540/valid/fewshot_val_shot10_seed30.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/valid}
|
|
val_10_300: {ann_file: odinw/brackishUnderwater/960x540/valid/fewshot_val_shot10_seed300.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/valid}
|
|
val_1_3: {ann_file: odinw/brackishUnderwater/960x540/valid/fewshot_val_shot1_seed3.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/valid}
|
|
val_1_30: {ann_file: odinw/brackishUnderwater/960x540/valid/fewshot_val_shot1_seed30.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/valid}
|
|
val_1_300: {ann_file: odinw/brackishUnderwater/960x540/valid/fewshot_val_shot1_seed300.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/valid}
|
|
val_3_3: {ann_file: odinw/brackishUnderwater/960x540/valid/fewshot_val_shot3_seed3.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/valid}
|
|
val_3_30: {ann_file: odinw/brackishUnderwater/960x540/valid/fewshot_val_shot3_seed30.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/valid}
|
|
val_3_300: {ann_file: odinw/brackishUnderwater/960x540/valid/fewshot_val_shot3_seed300.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/valid}
|
|
val_5_3: {ann_file: odinw/brackishUnderwater/960x540/valid/fewshot_val_shot5_seed3.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/valid}
|
|
val_5_30: {ann_file: odinw/brackishUnderwater/960x540/valid/fewshot_val_shot5_seed30.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/valid}
|
|
val_5_300: {ann_file: odinw/brackishUnderwater/960x540/valid/fewshot_val_shot5_seed300.json,
|
|
img_dir: odinw/brackishUnderwater/960x540/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: 7}
|
|
DYHEAD: {NUM_CLASSES: 7}
|
|
FCOS: {NUM_CLASSES: 7}
|
|
ROI_BOX_HEAD: {NUM_CLASSES: 7}
|
|
SOLVER: {CHECKPOINT_PERIOD: 100, MAX_EPOCH: 12, WARMUP_ITERS: 0}
|
|
TEST: {IMS_PER_BATCH: 8}
|
|
VISION_QUERY:
|
|
DATASET_NAME: 'brackishUnderwater' |