|
|
@@ -5,21 +5,18 @@ 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
|
|
|
@@ -29,14 +26,12 @@ MODELS:
|
|
|
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
|
|
|
@@ -46,26 +41,29 @@ MODELS:
|
|
|
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
|
|
|
+ n_classes: 200
|
|
|
|
|
|
CARS: &cars
|
|
|
folder: cars
|
|
|
annotations: ORIGINAL
|
|
|
+ n_classes: 196
|
|
|
|
|
|
############ 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: ""
|
