|
|
@@ -6,14 +6,17 @@ MODEL_DIR: models
|
|
|
############ Existing models
|
|
|
### weights are used for fine-tuning
|
|
|
MODELS:
|
|
|
+ efficientnet: &efficientnet
|
|
|
+ folder: efficientnet
|
|
|
+ class_key: efficientnet
|
|
|
+ weights: model.imagenet.npz
|
|
|
+
|
|
|
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
|
|
|
@@ -22,13 +25,11 @@ MODELS:
|
|
|
<<: *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
|
|
|
@@ -37,13 +38,11 @@ MODELS:
|
|
|
<<: *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
|
|
|
@@ -176,6 +175,12 @@ PART_TYPES:
|
|
|
scales:
|
|
|
- 0.31
|
|
|
|
|
|
+ NTS: &parts_nts
|
|
|
+ feature_suffix: _7parts_NTS
|
|
|
+ rescale_size: !!int 448
|
|
|
+ scales:
|
|
|
+ - 0.31
|
|
|
+
|
|
|
PARTS:
|
|
|
|
|
|
#### No Parts Annotations
|
|
|
@@ -290,6 +295,11 @@ PARTS:
|
|
|
<<: *parts_nac
|
|
|
annotations: NAC/2017-bilinear
|
|
|
|
|
|
+ CUB200_NTS:
|
|
|
+ <<: *cub200
|
|
|
+ <<: *parts_nts
|
|
|
+ annotations: cub200_11_nts
|
|
|
+
|
|
|
CUB200_L1_pred:
|
|
|
<<: *cub200
|
|
|
<<: *parts_l1p
|
