info.yml 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195
  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. ############ Existing Part Annotations and Part Features
  59. ### feature file name composition:
  60. # ${BASE_DIR}/${DATA_DIR}/${DATASETS:folder}/${PART_TYPES:annotations}/features
  61. PART_TYPES:
  62. GLOBAL: &parts_global
  63. feature_suffix: ""
  64. UNI: &parts_uni
  65. # <<: *cub200
  66. is_uniform: true
  67. feature_suffix: _26parts_uniform
  68. rescale_size: !!int -1
  69. scales:
  70. - 0.2
  71. GT: &parts_gt
  72. # <<: *cub200
  73. feature_suffix: _16parts_gt
  74. rescale_size: !!int -1
  75. scales:
  76. - 0.31
  77. GT2: &parts_gt2
  78. # <<: *cub200
  79. annotations: cub200_11_regrouped
  80. feature_suffix: _5parts_gt
  81. rescale_size: !!int -1
  82. scales:
  83. - 0.31
  84. NAC: &parts_nac
  85. # <<: *cub200
  86. annotations: NAC/2017-bilinear
  87. feature_suffix: _20parts
  88. rescale_size: !!int 224
  89. scales:
  90. - 0.31
  91. - 0.45
  92. L1_pred: &parts_l1p
  93. # <<: *cub200
  94. annotations: cub200_11_L1_pred
  95. feature_suffix: _5parts_L1_pred
  96. rescale_size: !!int 299
  97. scales:
  98. - 0.31
  99. L1_full: &parts_l1f
  100. # <<: *cub200
  101. annotations: cub200_11_L1_full
  102. feature_suffix: _5parts_L1_full
  103. rescale_size: !!int 299
  104. scales:
  105. - 0.31
  106. PARTS:
  107. #### No Parts Annotations
  108. INAT19_GLOBAL:
  109. <<: *inat19
  110. <<: *parts_global
  111. INAT19_MINI_GLOBAL:
  112. <<: *inat19_mini
  113. <<: *parts_global
  114. feature_suffix: .mini
  115. #### No Parts Annotations
  116. CUB200_GLOBAL:
  117. <<: *cub200
  118. <<: *parts_global
  119. CARS_GLOBAL:
  120. <<: *cars
  121. <<: *parts_global
  122. #### With Parts Annotations
  123. CUB200_UNI:
  124. <<: *cub200
  125. <<: *parts_uni
  126. CUB200_GT:
  127. <<: *cub200
  128. <<: *parts_gt
  129. CUB200_GT2:
  130. <<: *cub200
  131. <<: *parts_gt2
  132. CUB200_NAC:
  133. <<: *cub200
  134. <<: *parts_nac
  135. CUB200_L1_pred:
  136. <<: *cub200
  137. <<: *parts_l1p
  138. CUB200_L1_full:
  139. <<: *cub200
  140. <<: *parts_l1f
  141. CARS_L1_pred:
  142. <<: *cars
  143. <<: *parts_l1p
  144. annotations: L1_pred
  145. CARS_L1_full:
  146. <<: *cars
  147. <<: *parts_l1f
  148. annotations: L1_full
  149. INAT19_MINI_L1_pred:
  150. <<: *inat19_mini
  151. <<: *parts_l1p
  152. annotations: 2019_small_L1_pred
  153. feature_suffix: _5parts_L1_pred.mini
  154. INAT19_MINI_L1_full:
  155. <<: *inat19_mini
  156. <<: *parts_l1f
  157. feature_suffix: _5parts_L1_full.mini
  158. annotations: 2019_small_L1_full