info.yml 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222
  1. BASE_DIR: /home/korsch1/korsch/
  2. DATA_DIR: datasets
  3. MODEL_DIR: models
  4. ############ Existing models
  5. ### weights are used for fine-tuning
  6. MODELS:
  7. inception_inat: &inception_inat
  8. # class: InceptionV3
  9. folder: inception
  10. class_key: inception
  11. weights: model.inat.ckpt.npz
  12. inception_imagenet: &inception_inet
  13. # class: InceptionV3
  14. folder: inception
  15. class_key: inception
  16. weights: model.imagenet.ckpt.npz
  17. inception: &inception
  18. <<: *inception_inat
  19. inception_tf_inat: &inception_tf_inat
  20. # class: InceptionV3_TF
  21. folder: inception_tf
  22. class_key: inception_tf
  23. weights: inception_v3_iNat_299.ckpt
  24. inception_tf_inet: &inception_tf_inet
  25. # class: InceptionV3_TF
  26. folder: inception_tf
  27. class_key: inception_tf
  28. weights: inception_v3_ILSVRC_299.ckpt
  29. inception_tf: &inception_tf
  30. <<: *inception_tf_inat
  31. resnet: &resnet50
  32. # class: ResnetLayers
  33. folder: resnet
  34. class_key: resnet
  35. weights: model.npz
  36. vgg19: &vgg19
  37. # class: VGG19Layers
  38. folder: vgg19
  39. class_key: vgg19
  40. weights: model.npz
  41. ############ Existing Datasets
  42. DATASETS:
  43. CUB200: &cub200
  44. folder: birds
  45. annotations: cub200_11_ORIGINAL
  46. n_classes: 200
  47. CARS: &cars
  48. folder: cars
  49. annotations: ORIGINAL
  50. n_classes: 196
  51. INAT19: &inat19
  52. folder: inat
  53. annotations: "2019"
  54. n_classes: 1010
  55. INAT19_MINI: &inat19_mini
  56. <<: *inat19
  57. annotations: "2019_small"
  58. INAT19_TEST: &inat19_test
  59. <<: *inat19
  60. annotations: "2019_test"
  61. ############ Existing Part Annotations and Part Features
  62. ### feature file name composition:
  63. # ${BASE_DIR}/${DATA_DIR}/${DATASETS:folder}/${PART_TYPES:annotations}/features
  64. PART_TYPES:
  65. GLOBAL: &parts_global
  66. feature_suffix: ""
  67. UNI: &parts_uni
  68. # <<: *cub200
  69. is_uniform: true
  70. feature_suffix: _26parts_uniform
  71. rescale_size: !!int -1
  72. scales:
  73. - 0.2
  74. GT: &parts_gt
  75. # <<: *cub200
  76. feature_suffix: _16parts_gt
  77. rescale_size: !!int -1
  78. scales:
  79. - 0.31
  80. GT2: &parts_gt2
  81. # <<: *cub200
  82. annotations: cub200_11_regrouped
  83. feature_suffix: _5parts_gt
  84. rescale_size: !!int -1
  85. scales:
  86. - 0.31
  87. NAC: &parts_nac
  88. # <<: *cub200
  89. annotations: NAC/2017-bilinear
  90. feature_suffix: _20parts
  91. rescale_size: !!int 224
  92. scales:
  93. - 0.31
  94. - 0.45
  95. L1_pred: &parts_l1p
  96. # <<: *cub200
  97. annotations: cub200_11_L1_pred
  98. feature_suffix: _5parts_L1_pred
  99. rescale_size: !!int 299
  100. scales:
  101. - 0.31
  102. L1_full: &parts_l1f
  103. # <<: *cub200
  104. annotations: cub200_11_L1_full
  105. feature_suffix: _5parts_L1_full
  106. rescale_size: !!int 299
  107. scales:
  108. - 0.31
  109. PARTS:
  110. #### No Parts Annotations
  111. CUB200_GLOBAL:
  112. <<: *cub200
  113. <<: *parts_global
  114. CARS_GLOBAL:
  115. <<: *cars
  116. <<: *parts_global
  117. INAT19_GLOBAL:
  118. <<: *inat19
  119. <<: *parts_global
  120. INAT19_MINI_GLOBAL:
  121. <<: *inat19_mini
  122. <<: *parts_global
  123. feature_suffix: .mini
  124. INAT19_TEST_GLOBAL:
  125. <<: *inat19_test
  126. <<: *parts_global
  127. feature_suffix: .test
  128. #### With Parts Annotations
  129. # CUB200-2011
  130. CUB200_UNI:
  131. <<: *cub200
  132. <<: *parts_uni
  133. CUB200_GT:
  134. <<: *cub200
  135. <<: *parts_gt
  136. CUB200_GT2:
  137. <<: *cub200
  138. <<: *parts_gt2
  139. CUB200_NAC:
  140. <<: *cub200
  141. <<: *parts_nac
  142. CUB200_L1_pred:
  143. <<: *cub200
  144. <<: *parts_l1p
  145. CUB200_L1_full:
  146. <<: *cub200
  147. <<: *parts_l1f
  148. # Stanford Cars
  149. CARS_L1_pred:
  150. <<: *cars
  151. <<: *parts_l1p
  152. annotations: L1_pred
  153. CARS_L1_full:
  154. <<: *cars
  155. <<: *parts_l1f
  156. annotations: L1_full
  157. # iNaturalist 2019
  158. INAT19_MINI_L1_pred:
  159. <<: *inat19_mini
  160. <<: *parts_l1p
  161. annotations: 2019_small_L1_pred
  162. feature_suffix: _5parts_L1_pred.mini
  163. INAT19_MINI_L1_full:
  164. <<: *inat19_mini
  165. <<: *parts_l1f
  166. annotations: 2019_small_L1_full
  167. feature_suffix: _5parts_L1_full.mini
  168. INAT19_TEST_L1_pred:
  169. <<: *inat19_test
  170. <<: *parts_l1p
  171. annotations: 2019_test_L1_pred
  172. feature_suffix: _5parts_L1_pred.test
  173. INAT19_TEST_L1_full:
  174. <<: *inat19_test
  175. <<: *parts_l1f
  176. annotations: 2019_test_L1_full
  177. feature_suffix: _5parts_L1_full.test