info.yml 7.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403
  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. 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/cub200
  45. annotations: "ORIGINAL"
  46. n_classes: 200
  47. CUB200_2FOLD: &cub200_2fold
  48. folder: birds/cub200_2fold
  49. annotations: "ORIGINAL"
  50. n_classes: 200
  51. CUB200_GOOGLE: &cub200_google
  52. folder: birds/cub200_google_images
  53. annotations: "ORIGINAL"
  54. n_classes: 200
  55. CUB200_GOOGLE_SEM: &cub200_google_sem
  56. folder: birds/cub200_google_images
  57. annotations: "sem_selected"
  58. n_classes: 200
  59. NAB: &nabirds
  60. folder: birds/nabirds
  61. annotations: "ORIGINAL"
  62. n_classes: 555
  63. CARS: &cars
  64. folder: cars
  65. annotations: "ORIGINAL"
  66. n_classes: 196
  67. DOGS: &dogs
  68. folder: dogs
  69. annotations: "ORIGINAL"
  70. n_classes: 120
  71. FLOWERS: &flowers
  72. folder: flowers
  73. annotations: "flowers102"
  74. n_classes: 102
  75. INAT19: &inat19
  76. folder: inat
  77. annotations: "2019"
  78. n_classes: 1010
  79. INAT18: &inat18
  80. folder: inat
  81. annotations: "2018"
  82. n_classes: 8142
  83. INAT19_MINI: &inat19_mini
  84. <<: *inat19
  85. annotations: "2019_small"
  86. INAT19_TEST: &inat19_test
  87. <<: *inat19
  88. annotations: "2019_test"
  89. HED: &hed
  90. folder: medical/HED
  91. annotations: "patches224x224"
  92. n_classes: 2
  93. TIGERS: &tigers
  94. folder: tigers
  95. annotations: "reid/train"
  96. n_classes: 107
  97. TIGERS_TEST: &tigers_test
  98. folder: tigers
  99. annotations: "reid/test"
  100. n_classes: 107
  101. ############ Existing Part Annotations and Part Features
  102. ### feature file name composition:
  103. # ${BASE_DIR}/${DATA_DIR}/${DATASETS:folder}/${PART_TYPES:annotations}/features
  104. PART_TYPES:
  105. GLOBAL: &parts_global
  106. feature_suffix: ""
  107. UNI: &parts_uni
  108. is_uniform: true
  109. feature_suffix: _26parts_uniform
  110. rescale_size: !!int -1
  111. scales:
  112. - 0.2
  113. GT: &parts_gt
  114. feature_suffix: _16parts_gt
  115. rescale_size: !!int -1
  116. scales:
  117. - 0.31
  118. GT2: &parts_gt2
  119. feature_suffix: _5parts_gt
  120. rescale_size: !!int -1
  121. scales:
  122. - 0.31
  123. NAC: &parts_nac
  124. feature_suffix: _20parts
  125. rescale_size: !!int 224
  126. scales:
  127. - 0.31
  128. - 0.45
  129. L1_pred: &parts_l1p
  130. feature_suffix: _5parts_L1_pred
  131. rescale_size: !!int 299
  132. scales:
  133. - 0.31
  134. L1_full: &parts_l1f
  135. feature_suffix: _5parts_L1_full
  136. rescale_size: !!int 299
  137. scales:
  138. - 0.31
  139. PARTS:
  140. #### No Parts Annotations
  141. CUB200_2FOLD_GLOBAL:
  142. <<: *cub200_2fold
  143. <<: *parts_global
  144. CUB200_GLOBAL:
  145. <<: *cub200
  146. <<: *parts_global
  147. CUB200_GOOGLE_GLOBAL:
  148. <<: *cub200_google
  149. <<: *parts_global
  150. CUB200_GOOGLE_SEM_GLOBAL:
  151. <<: *cub200_google_sem
  152. <<: *parts_global
  153. CARS_GLOBAL:
  154. <<: *cars
  155. <<: *parts_global
  156. DOGS_GLOBAL:
  157. <<: *dogs
  158. <<: *parts_global
  159. NAB_GLOBAL:
  160. <<: *nabirds
  161. <<: *parts_global
  162. FLOWERS_GLOBAL:
  163. <<: *flowers
  164. <<: *parts_global
  165. INAT18_GLOBAL:
  166. <<: *inat18
  167. <<: *parts_global
  168. INAT19_GLOBAL:
  169. <<: *inat19
  170. <<: *parts_global
  171. INAT19_MINI_GLOBAL:
  172. <<: *inat19_mini
  173. <<: *parts_global
  174. feature_suffix: .mini
  175. INAT19_TEST_GLOBAL:
  176. <<: *inat19_test
  177. <<: *parts_global
  178. feature_suffix: .test
  179. HED_GLOBAL:
  180. <<: *hed
  181. <<: *parts_global
  182. TIGERS_GLOBAL:
  183. <<: *tigers
  184. <<: *parts_global
  185. rescale_size: !!int -1
  186. scales:
  187. - .31
  188. TIGERS_TEST_GLOBAL:
  189. <<: *tigers_test
  190. <<: *parts_global
  191. #### With Parts Annotations
  192. ####################################
  193. # CUB200-2011 2-Fold training set
  194. ####################################
  195. CUB200_2FOLD_L1_pred:
  196. <<: *cub200_2fold
  197. <<: *parts_l1p
  198. annotations: L1_pred
  199. rescale_size: !!int 427
  200. CUB200_2FOLD_L1_full:
  201. <<: *cub200_2fold
  202. <<: *parts_l1f
  203. annotations: L1_full
  204. rescale_size: !!int 427
  205. ####################################
  206. # CUB200-2011
  207. ####################################
  208. CUB200_UNI:
  209. <<: *cub200
  210. <<: *parts_uni
  211. CUB200_GT:
  212. <<: *cub200
  213. <<: *parts_gt
  214. CUB200_GT2:
  215. <<: *cub200
  216. <<: *parts_gt2
  217. annotations: cub200_11_regrouped
  218. CUB200_NAC:
  219. <<: *cub200
  220. <<: *parts_nac
  221. annotations: NAC/2017-bilinear
  222. CUB200_L1_pred:
  223. <<: *cub200
  224. <<: *parts_l1p
  225. annotations: cub200_11_L1_pred_old
  226. # rescale_size: !!int 427
  227. CUB200_L1_full:
  228. <<: *cub200
  229. <<: *parts_l1f
  230. annotations: cub200_11_L1_full_old
  231. # rescale_size: !!int 427
  232. CUB200_L1_pred_15:
  233. <<: *cub200
  234. <<: *parts_l1p
  235. annotations: L1_pred_15
  236. rescale_size: !!int 427
  237. feature_suffix: _15parts_L1_pred
  238. CUB200_L1_full_15:
  239. <<: *cub200
  240. <<: *parts_l1f
  241. annotations: L1_full_15
  242. rescale_size: !!int 427
  243. feature_suffix: _15parts_L1_full
  244. CUB200_L1_pred_2:
  245. <<: *cub200
  246. <<: *parts_l1p
  247. annotations: L1_pred_2
  248. rescale_size: !!int 427
  249. feature_suffix: _2parts_L1_pred
  250. CUB200_L1_full_2:
  251. <<: *cub200
  252. <<: *parts_l1f
  253. annotations: L1_full_2
  254. rescale_size: !!int 427
  255. feature_suffix: _2parts_L1_full
  256. ####################################
  257. # NA Birds
  258. ####################################
  259. NAB_GT:
  260. <<: *nabirds
  261. <<: *parts_gt
  262. NAB_L1_pred:
  263. <<: *nabirds
  264. <<: *parts_l1p
  265. annotations: L1_pred
  266. rescale_size: !!int 427
  267. NAB_L1_full:
  268. <<: *nabirds
  269. <<: *parts_l1f
  270. annotations: L1_full
  271. rescale_size: !!int 427
  272. ####################################
  273. # Stanford Cars
  274. ####################################
  275. CARS_L1_pred:
  276. <<: *cars
  277. <<: *parts_l1p
  278. annotations: L1_pred
  279. CARS_L1_full:
  280. <<: *cars
  281. <<: *parts_l1f
  282. annotations: L1_full
  283. ####################################
  284. # Flowers 102
  285. ####################################
  286. FLOWERS_L1_pred:
  287. <<: *flowers
  288. <<: *parts_l1p
  289. annotations: L1_pred
  290. FLOWERS_L1_full:
  291. <<: *flowers
  292. <<: *parts_l1f
  293. annotations: L1_full
  294. ####################################
  295. # iNaturalist 2019
  296. ####################################
  297. INAT19_MINI_L1_pred:
  298. <<: *inat19_mini
  299. <<: *parts_l1p
  300. annotations: 2019_small_L1_pred
  301. feature_suffix: _5parts_L1_pred.mini
  302. INAT19_MINI_L1_full:
  303. <<: *inat19_mini
  304. <<: *parts_l1f
  305. annotations: 2019_small_L1_full
  306. feature_suffix: _5parts_L1_full.mini
  307. INAT19_TEST_L1_pred:
  308. <<: *inat19_test
  309. <<: *parts_l1p
  310. annotations: 2019_test_L1_pred
  311. feature_suffix: _5parts_L1_pred.test
  312. INAT19_TEST_L1_full:
  313. <<: *inat19_test
  314. <<: *parts_l1f
  315. annotations: 2019_test_L1_full
  316. feature_suffix: _5parts_L1_full.test