Dimitri Korsch 6 роки тому
батько
коміт
88acb43517
1 змінених файлів з 74 додано та 0 видалено
  1. 74 0
      scripts/info_files/info.yml

+ 74 - 0
scripts/info_files/info.yml

@@ -0,0 +1,74 @@
+BASE_DIR: /home/korsch1/korsch/
+
+DATA_DIR: datasets
+MODEL_DIR: models
+
+############ Existing models
+MODELS:
+  InceptionV3:    &inception
+    folder: inception
+    weights: ft_inat/model.ckpt.npz
+
+  TFInceptionV3:  &inception_tf
+    folder: inception_tf
+    weights: ft_inat/inception_v3_iNat_299.ckpt
+
+  ResNet50:       &resnet50
+    folder: resnet
+    weights: ft_cub200/model.npz
+
+############ Existing Datasets
+DATASETS:
+  CUB200:         &cub200
+    folder: birds
+    annotations: cub200_11_ORIGINAL
+
+############ Existing Part Annotations and Part Features
+PARTS:
+  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