Sfoglia il codice sorgente

updated info file for new cvfinetune API (using cvmodelz module)

Dimitri Korsch 4 anni fa
parent
commit
c1e8c2d833
1 ha cambiato i file con 51 aggiunte e 35 eliminazioni
  1. 51 35
      scripts/info_files/info.yml

+ 51 - 35
scripts/info_files/info.yml

@@ -6,46 +6,62 @@ MODEL_DIR: models
 ############ Existing models
 ############ Existing models
 ### weights are used for fine-tuning
 ### weights are used for fine-tuning
 MODELS:
 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
     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
     folder: resnet
-    class_key: resnet
-    weights: model.npz
+    weights:
+      imagenet: model.npz
 
 
-  vgg19:       &vgg19
+  cvmodelz.VGG19:     &vgg19
     folder: 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
 ############ Existing Datasets
 DATASETS:
 DATASETS: