| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156 |
- BASE_DIR: /home/korsch1/korsch/
- DATA_DIR: datasets
- MODEL_DIR: models
- ############ Existing models
- ### weights are used for fine-tuning
- ### ft_weights are used for feature extraction
- MODELS:
- inception_inat: &inception_inat
- # class: InceptionV3
- folder: inception
- class_key: inception
- weights: model.inat.ckpt.npz
- ft_weights: ft_inat/model.ckpt.npz
- inception_imagenet: &inception_inet
- # class: InceptionV3
- folder: inception
- class_key: inception
- weights: model.imagenet.ckpt.npz
- ft_weights: ft_imagenet/model.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
- ft_weights: ft_inat/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
- ft_weights: ft_inat/inception_v3_ILSVRC_299.ckpt
- inception_tf: &inception_tf
- <<: *inception_tf_inat
- resnet: &resnet50
- # class: ResnetLayers
- folder: resnet
- class_key: resnet
- weights: model.npz
- ft_weights: ft_cub200/model.npz
- vgg19: &vgg19
- # class: VGG19Layers
- folder: vgg19
- class_key: vgg19
- weights: model.npz
- ft_weights: ft_cub200/model.npz
- ############ Existing Datasets
- DATASETS:
- CUB200: &cub200
- folder: birds
- annotations: cub200_11_ORIGINAL
- CARS: &cars
- folder: cars
- annotations: ORIGINAL
- ############ Existing Part Annotations and Part 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
- 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
|
