info.yml 3.0 KB

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