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
    annotations: cub200_11_ORIGINAL
    n_classes: 200

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

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

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

############ 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
    # <<: *cub200
    is_uniform: true
    feature_suffix: _26parts_uniform
    rescale_size: !!int -1
    scales:
     - 0.2

  GT:             &parts_gt
    # <<: *cub200
    feature_suffix: _16parts_gt
    rescale_size: !!int -1
    scales:
     - 0.31

  GT2:            &parts_gt2
    # <<: *cub200
    annotations: cub200_11_regrouped
    feature_suffix: _5parts_gt
    rescale_size: !!int -1
    scales:
      - 0.31

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

  L1_pred:        &parts_l1p
    # <<: *cub200
    annotations: cub200_11_L1_pred
    feature_suffix: _5parts_L1_pred
    rescale_size: !!int 299
    scales:
      - 0.31

  L1_full:        &parts_l1f
    # <<: *cub200
    annotations: cub200_11_L1_full
    feature_suffix: _5parts_L1_full
    rescale_size: !!int 299
    scales:
      - 0.31

PARTS:
  #### No Parts Annotations
  INAT19_GLOBAL:
    <<: *inat19
    <<: *parts_global

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

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

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

  #### With Parts Annotations
  CUB200_UNI:
    <<: *cub200
    <<: *parts_uni

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

  CUB200_GT2:
    <<: *cub200
    <<: *parts_gt2

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

  CUB200_L1_pred:
    <<: *cub200
    <<: *parts_l1p

  CUB200_L1_full:
    <<: *cub200
    <<: *parts_l1f


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

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

  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
    feature_suffix: _5parts_L1_full.mini
    annotations: 2019_small_L1_full