info.yml 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305
  1. BASE_DIR: /home/korsch_data/
  2. DATA_DIR: datasets
  3. MODEL_DIR: models
  4. ############ Existing models
  5. ### weights are used for fine-tuning
  6. MODELS:
  7. cvmodelz.InceptionV3: &inception
  8. folder: inception
  9. weights:
  10. inat: model.inat.ckpt.npz
  11. imagenet: model.imagenet.ckpt.npz
  12. cvmodelz.ResNet50: &resnet50
  13. folder: resnet
  14. weights:
  15. imagenet: model.npz
  16. cvmodelz.VGG19: &vgg19
  17. folder: vgg19
  18. weights:
  19. imagenet: model.npz
  20. ############ Existing Datasets
  21. DATASETS:
  22. IMAGENET: &inet
  23. folder: ImageNet
  24. annotations: "BJOERN"
  25. annotation_type: FOLDER
  26. train_images: ILSVRC2012_img_train
  27. val_images: ILSVRC2012_img_val
  28. n_classes: 1000
  29. IMAGENET_TOP_INAT20: &inet_top_inat20
  30. <<: *inet
  31. annotations: "TOP_INAT20"
  32. n_classes: 44
  33. CUB200: &cub200
  34. folder: birds/cub200
  35. annotations: "ORIGINAL"
  36. annotation_type: FILE_LIST
  37. n_classes: 200
  38. CUB10: &cub10
  39. <<: *cub200
  40. annotations: "cub10"
  41. n_classes: 10
  42. CUB200_2FOLD: &cub200_2fold
  43. <<: *cub200
  44. folder: birds/cub200_2fold
  45. CUB200_GOOGLE: &cub200_google
  46. <<: *cub200
  47. folder: birds/cub200_google_images
  48. CUB200_GOOGLE_SEM: &cub200_google_sem
  49. <<: *cub200
  50. folder: birds/cub200_google_images
  51. annotations: "sem_selected"
  52. NAB: &nabirds
  53. folder: birds/nabirds
  54. annotations: "ORIGINAL"
  55. annotation_type: FILE_LIST
  56. n_classes: 555
  57. BIRDSNAP: &birdsnap
  58. folder: birds/birdsnap
  59. annotations: "RESIZED"
  60. annotation_type: FILE_LIST
  61. n_classes: 500
  62. CARS: &cars
  63. folder: cars
  64. annotations: "ORIGINAL"
  65. annotation_type: FILE_LIST
  66. n_classes: 196
  67. DOGS: &dogs
  68. folder: dogs
  69. annotations: "ORIGINAL"
  70. annotation_type: FILE_LIST
  71. n_classes: 120
  72. FLOWERS: &flowers
  73. folder: flowers
  74. annotations: "flowers102"
  75. annotation_type: FILE_LIST
  76. n_classes: 102
  77. HERBA19: &herba19
  78. folder: herbarium
  79. annotations: "2019"
  80. annotation_type: FOLDER
  81. train_images: small-train
  82. val_images: small-validation
  83. test_images: small-test
  84. n_classes: 683
  85. INAT20: &inat20
  86. folder: inat
  87. annotations: "2020/PLAIN"
  88. annotation_type: JSON
  89. n_classes: 200
  90. INAT20_IN_CLASS: &inat20_in_class
  91. <<: *inat20
  92. annotations: "2020/IN_CLASS"
  93. INAT20_NOISY_IN_CLASS: &inat20_noisy_in_class
  94. <<: *inat20
  95. annotations: "2020/NOISY_IN_CLASS"
  96. INAT20_OUT_CLASS: &inat20_out_class
  97. <<: *inat20
  98. annotations: "2020/OUT_CLASS"
  99. INAT20_NOISY_OUT_CLASS: &inat20_noisy_out_class
  100. <<: *inat20
  101. annotations: "2020/NOISY_OUT_CLASS"
  102. INAT20_U_OUT_CLASS: &inat20_u_out_class
  103. <<: *inat20
  104. annotations: "2020/U_OUT_CLASS"
  105. INAT20_TEST: &inat20_test
  106. <<: *inat20
  107. annotations: "2020/TEST"
  108. INAT19: &inat19
  109. folder: inat
  110. annotations: "2019"
  111. annotation_type: JSON
  112. n_classes: 1010
  113. INAT19_MINI: &inat19_mini
  114. <<: *inat19
  115. annotations: "2019_small"
  116. feature_suffix: .mini
  117. INAT19_TEST: &inat19_test
  118. <<: *inat19
  119. annotations: "2019_test"
  120. feature_suffix: .test
  121. INAT18: &inat18
  122. folder: inat
  123. annotations: "2018"
  124. annotation_type: JSON
  125. n_classes: 8142
  126. HED: &hed
  127. folder: medical/HED
  128. annotations: "patches224x224"
  129. annotation_type: FILE_LIST
  130. n_classes: 2
  131. TIGERS: &tigers
  132. folder: tigers
  133. annotations: "reid/train"
  134. annotation_type: FILE_LIST
  135. n_classes: 107
  136. TIGERS_TEST: &tigers_test
  137. <<: *tigers
  138. annotations: "reid/test"
  139. ############ Existing Part Annotations and Part Features
  140. ### feature file name composition:
  141. # ${BASE_DIR}/${DATA_DIR}/${DATASETS:folder}/${PART_TYPES:annotations}/features
  142. PART_TYPES:
  143. GLOBAL: &parts_global
  144. feature_suffix: ""
  145. UNI: &parts_uni
  146. is_uniform: true
  147. feature_suffix: _26parts_uniform
  148. rescale_size: !!int -1
  149. scales:
  150. - 0.2
  151. GT: &parts_gt
  152. feature_suffix: _16parts_gt
  153. rescale_size: !!int -1
  154. scales:
  155. - 0.31
  156. GT2: &parts_gt2
  157. # TODO: should be changed, since GT2 parts can also be present for other datasets
  158. annotations: cub200_11_regrouped
  159. feature_suffix: _5parts_gt
  160. rescale_size: !!int -1
  161. scales:
  162. - 0.31
  163. NAC: &parts_nac
  164. annotations: NAC/2017-bilinear
  165. feature_suffix: _20parts
  166. rescale_size: !!int 224
  167. scales:
  168. - 0.31
  169. - 0.45
  170. L1_pred: &parts_l1p
  171. annotations: L1_pred
  172. feature_suffix: _5parts_L1_pred
  173. rescale_size: !!int 427
  174. scales: [-1]
  175. L1_full: &parts_l1f
  176. annotations: L1_full
  177. feature_suffix: _5parts_L1_full
  178. rescale_size: !!int 427
  179. scales: [-1]
  180. NTS: &parts_nts
  181. annotations: NTS
  182. feature_suffix: _7parts_nts
  183. rescale_size: !!int 448
  184. scales: [-1]
  185. NTS2: &parts_nts2
  186. annotations: NTS2
  187. feature_suffix: _5parts_nts
  188. rescale_size: !!int 448
  189. scales: [-1]
  190. PARTS:
  191. # all <DATASET>_<PART_TYPES> combinations are created implicitely.
  192. # if you want some changes in the configs, then crete here an entry
  193. # and update the config values
  194. ####################################
  195. # CUB200-2011
  196. ####################################
  197. CUB200_L1_pred:
  198. <<: *cub200
  199. <<: *parts_l1p
  200. annotations: cub200_11_L1_pred_old
  201. rescale_size: !!int 299
  202. CUB200_L1_full:
  203. <<: *cub200
  204. <<: *parts_l1f
  205. annotations: cub200_11_L1_full_old
  206. rescale_size: !!int 299
  207. CUB200_L1_pred_15:
  208. <<: *cub200
  209. <<: *parts_l1p
  210. annotations: L1_pred_15
  211. feature_suffix: _15parts_L1_pred
  212. CUB200_L1_full_15:
  213. <<: *cub200
  214. <<: *parts_l1f
  215. annotations: L1_full_15
  216. feature_suffix: _15parts_L1_full
  217. CUB200_L1_pred_2:
  218. <<: *cub200
  219. <<: *parts_l1p
  220. annotations: L1_pred_2
  221. feature_suffix: _2parts_L1_pred
  222. CUB200_L1_full_2:
  223. <<: *cub200
  224. <<: *parts_l1f
  225. annotations: L1_full_2
  226. feature_suffix: _2parts_L1_full
  227. ####################################
  228. # iNaturalist 2019
  229. ####################################
  230. INAT19_MINI_L1_pred:
  231. <<: *inat19_mini
  232. <<: *parts_l1p
  233. annotations: 2019_small_L1_pred
  234. feature_suffix: _5parts_L1_pred.mini
  235. INAT19_MINI_L1_full:
  236. <<: *inat19_mini
  237. <<: *parts_l1f
  238. annotations: 2019_small_L1_full
  239. feature_suffix: _5parts_L1_full.mini
  240. INAT19_TEST_L1_pred:
  241. <<: *inat19_test
  242. <<: *parts_l1p
  243. annotations: 2019_test_L1_pred
  244. feature_suffix: _5parts_L1_pred.test
  245. INAT19_TEST_L1_full:
  246. <<: *inat19_test
  247. <<: *parts_l1f
  248. annotations: 2019_test_L1_full
  249. feature_suffix: _5parts_L1_full.test