info.yml 3.1 KB

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