| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252 |
- 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
- 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"
- ############ 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
- 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
- #### 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
- ####################################
- # Stanford Cars
- ####################################
- CARS_L1_pred:
- <<: *cars
- <<: *parts_l1p
- annotations: L1_pred
- CARS_L1_full:
- <<: *cars
- <<: *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
|
