|
|
@@ -6,46 +6,62 @@ 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
|
|
|
- folder: inception
|
|
|
- class_key: inception
|
|
|
- weights: model.inat.ckpt.npz
|
|
|
-
|
|
|
- inception_imagenet: &inception_inet
|
|
|
+ cvmodelz.InceptionV3: &inception
|
|
|
folder: inception
|
|
|
- class_key: inception
|
|
|
- weights: model.imagenet.ckpt.npz
|
|
|
-
|
|
|
- inception: &inception
|
|
|
- <<: *inception_inat
|
|
|
-
|
|
|
- inception_tf_inat: &inception_tf_inat
|
|
|
- folder: inception_tf
|
|
|
- class_key: inception_tf
|
|
|
- weights: inception_v3_iNat_299.ckpt
|
|
|
-
|
|
|
- inception_tf_inet: &inception_tf_inet
|
|
|
- folder: inception_tf
|
|
|
- class_key: inception_tf
|
|
|
- weights: inception_v3_ILSVRC_299.ckpt
|
|
|
-
|
|
|
- inception_tf: &inception_tf
|
|
|
- <<: *inception_tf_inat
|
|
|
+ weights:
|
|
|
+ inat: model.inat.ckpt.npz
|
|
|
+ imagenet: model.imagenet.ckpt.npz
|
|
|
|
|
|
- resnet: &resnet50
|
|
|
+ cvmodelz.ResNet50: &resnet50
|
|
|
folder: resnet
|
|
|
- class_key: resnet
|
|
|
- weights: model.npz
|
|
|
+ weights:
|
|
|
+ imagenet: model.npz
|
|
|
|
|
|
- vgg19: &vgg19
|
|
|
+ cvmodelz.VGG19: &vgg19
|
|
|
folder: vgg19
|
|
|
- class_key: vgg19
|
|
|
- weights: model.npz
|
|
|
+ weights:
|
|
|
+ imagenet: model.npz
|
|
|
+
|
|
|
+ # efficientnet: &efficientnet
|
|
|
+ # folder: efficientnet
|
|
|
+ # class_key: efficientnet
|
|
|
+ # weights: model.imagenet.npz
|
|
|
+
|
|
|
+ # inception_inat: &inception_inat
|
|
|
+ # folder: inception
|
|
|
+ # class_key: inception
|
|
|
+ # weights: model.inat.ckpt.npz
|
|
|
+
|
|
|
+ # inception_imagenet: &inception_inet
|
|
|
+ # folder: inception
|
|
|
+ # class_key: inception
|
|
|
+ # weights: model.imagenet.ckpt.npz
|
|
|
+
|
|
|
+ # inception:
|
|
|
+ # <<: *inception_inat
|
|
|
+
|
|
|
+ # inception_tf_inat: &inception_tf_inat
|
|
|
+ # folder: inception_tf
|
|
|
+ # class_key: inception_tf
|
|
|
+ # weights: inception_v3_iNat_299.ckpt
|
|
|
+
|
|
|
+ # inception_tf_inet: &inception_tf_inet
|
|
|
+ # folder: inception_tf
|
|
|
+ # class_key: inception_tf
|
|
|
+ # weights: inception_v3_ILSVRC_299.ckpt
|
|
|
+
|
|
|
+ # inception_tf: &inception_tf
|
|
|
+ # <<: *inception_tf_inat
|
|
|
+
|
|
|
+ # resnet: &resnet50
|
|
|
+ # folder: resnet
|
|
|
+ # class_key: resnet
|
|
|
+ # weights: model.npz
|
|
|
+
|
|
|
+ # vgg19: &vgg19
|
|
|
+ # folder: vgg19
|
|
|
+ # class_key: vgg19
|
|
|
+ # weights: model.npz
|
|
|
|
|
|
############ Existing Datasets
|
|
|
DATASETS:
|
