cvdatasets/scripts/info_files/info.yml

BASE_DIR: /home/korsch1/korsch/

DATA_DIR: datasets
MODEL_DIR: models

############ Existing models
### weights are used for fine-tuning
MODELS:
  inception_inat:    &inception_inat
    # class: InceptionV3
    folder: inception
    class_key: inception
    weights: model.inat.ckpt.npz

  inception_imagenet:    &inception_inet
    # class: InceptionV3
    folder: inception
    class_key: inception
    weights: model.imagenet.ckpt.npz

  inception:    &inception
    <<: *inception_inat

  inception_tf_inat:  &inception_tf_inat
    # class: InceptionV3_TF
    folder: inception_tf
    class_key: inception_tf
    weights: inception_v3_iNat_299.ckpt

  inception_tf_inet:  &inception_tf_inet
    # class: InceptionV3_TF
    folder: inception_tf
    class_key: inception_tf
    weights: inception_v3_ILSVRC_299.ckpt

  inception_tf:  &inception_tf
    <<: *inception_tf_inat

  resnet:       &resnet50
    # class: ResnetLayers
    folder: resnet
    class_key: resnet
    weights: model.npz

  vgg19:       &vgg19
    # class: VGG19Layers
    folder: vgg19
    class_key: vgg19
    weights: model.npz

############ Existing Datasets
DATASETS:
  CUB200:         &cub200
    folder: birds/cub200
    annotations: "ORIGINAL"
    n_classes: 200

  NAB:         &nabirds
    folder: birds/nabirds
    annotations: "ORIGINAL"
    n_classes: 555

  CARS:         &cars
    folder: cars
    annotations: "ORIGINAL"
    n_classes: 196

  DOGS:         &dogs
    folder: dogs
    annotations: "ORIGINAL"
    n_classes: 120

  FLOWERS:         &flowers
    folder: flowers
    annotations: "flowers102"
    n_classes: 102

  INAT19:         &inat19
    folder: inat
    annotations: "2019"
    n_classes: 1010

  INAT19_MINI:    &inat19_mini
    <<: *inat19
    annotations: "2019_small"

  INAT19_TEST:    &inat19_test
    <<: *inat19
    annotations: "2019_test"


  HED:         &hed
    folder: medical/HED
    annotations: "patches224x224"
    n_classes: 2

############ Existing Part Annotations and Part Features
### feature file name composition:
# ${BASE_DIR}/${DATA_DIR}/${DATASETS:folder}/${PART_TYPES:annotations}/features

PART_TYPES:
  GLOBAL:         &parts_global
    feature_suffix: ""

  UNI:            &parts_uni
    is_uniform: true
    feature_suffix: _26parts_uniform
    rescale_size: !!int -1
    scales:
     - 0.2

  GT:             &parts_gt
    feature_suffix: _16parts_gt
    rescale_size: !!int -1
    scales:
     - 0.31

  GT2:            &parts_gt2
    feature_suffix: _5parts_gt
    rescale_size: !!int -1
    scales:
      - 0.31

  NAC:            &parts_nac
    annotations: NAC/2017-bilinear
    feature_suffix: _20parts
    rescale_size: !!int 224
    scales:
      - 0.31
      - 0.45

  L1_pred:        &parts_l1p
    feature_suffix: _5parts_L1_pred
    rescale_size: !!int 299
    scales:
      - 0.31

  L1_full:        &parts_l1f
    feature_suffix: _5parts_L1_full
    rescale_size: !!int 299
    scales:
      - 0.31

PARTS:

  #### No Parts Annotations
  CUB200_GLOBAL:
    <<: *cub200
    <<: *parts_global

  CARS_GLOBAL:
    <<: *cars
    <<: *parts_global

  DOGS_GLOBAL:
    <<: *dogs
    <<: *parts_global

  NAB_GLOBAL:
    <<: *nabirds
    <<: *parts_global

  FLOWERS_GLOBAL:
    <<: *flowers
    <<: *parts_global

  INAT19_GLOBAL:
    <<: *inat19
    <<: *parts_global

  INAT19_MINI_GLOBAL:
    <<: *inat19_mini
    <<: *parts_global
    feature_suffix: .mini

  INAT19_TEST_GLOBAL:
    <<: *inat19_test
    <<: *parts_global
    feature_suffix: .test

  HED_GLOBAL:
    <<: *hed
    <<: *parts_global

  #### With Parts Annotations


  ####################################
  # CUB200-2011
  ####################################

  CUB200_UNI:
    <<: *cub200
    <<: *parts_uni

  CUB200_GT:
    <<: *cub200
    <<: *parts_gt

  CUB200_GT2:
    <<: *cub200
    <<: *parts_gt2
    annotations: cub200_11_regrouped

  CUB200_NAC:
    <<: *cub200
    <<: *parts_nac

  CUB200_L1_pred:
    <<: *cub200
    <<: *parts_l1p
    annotations: cub200_11_L1_pred

  CUB200_L1_full:
    <<: *cub200
    <<: *parts_l1f
    annotations: cub200_11_L1_full


  ####################################
  # NA Birds
  ####################################

  NAB_GT:
    <<: *nabirds
    <<: *parts_gt

  NAB_L1_pred:
    <<: *nabirds
    <<: *parts_l1p
    annotations: L1_pred
    rescale_size: !!int 427

  NAB_L1_full:
    <<: *nabirds
    <<: *parts_l1f
    annotations: L1_full
    rescale_size: !!int 427


  ####################################
  # Stanford Cars
  ####################################

  CARS_L1_pred:
    <<: *cars
    <<: *parts_l1p
    annotations: L1_pred

  CARS_L1_full:
    <<: *cars
    <<: *parts_l1f
    annotations: L1_full


  ####################################
  # Flowers 102
  ####################################

  FLOWERS_L1_pred:
    <<: *flowers
    <<: *parts_l1p
    annotations: L1_pred

  FLOWERS_L1_full:
    <<: *flowers
    <<: *parts_l1f
    annotations: L1_full


  ####################################
  # iNaturalist 2019
  ####################################

  INAT19_MINI_L1_pred:
    <<: *inat19_mini
    <<: *parts_l1p
    annotations: 2019_small_L1_pred
    feature_suffix: _5parts_L1_pred.mini

  INAT19_MINI_L1_full:
    <<: *inat19_mini
    <<: *parts_l1f
    annotations: 2019_small_L1_full
    feature_suffix: _5parts_L1_full.mini

  INAT19_TEST_L1_pred:
    <<: *inat19_test
    <<: *parts_l1p
    annotations: 2019_test_L1_pred
    feature_suffix: _5parts_L1_pred.test

  INAT19_TEST_L1_full:
    <<: *inat19_test
    <<: *parts_l1f
    annotations: 2019_test_L1_full
    feature_suffix: _5parts_L1_full.test