info.yml 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  1. BASE_DIR: /home/korsch1/korsch/
  2. DATA_DIR: datasets
  3. MODEL_DIR: models
  4. ############ Existing models
  5. MODELS:
  6. inception: &inception
  7. class: InceptionV3
  8. folder: inception
  9. weights: ft_inat/model.ckpt.npz
  10. ft_weights: ft_inat/model.ckpt.npz
  11. inception_tf: &inception_tf
  12. class: InceptionV3_TF
  13. folder: inception_tf
  14. weights: ft_inat/inception_v3_iNat_299.ckpt
  15. ft_weights: ft_inat/inception_v3_iNat_299.ckpt
  16. resnet: &resnet50
  17. class: ResnetLayers
  18. folder: resnet
  19. weights: model.npz
  20. ft_weights: ft_cub200/model.npz
  21. vgg19: &vgg19
  22. class: VGG19Layers
  23. folder: vgg19
  24. weights: model.npz
  25. ft_weights: ft_cub200/model.npz
  26. ############ Existing Datasets
  27. DATASETS:
  28. CUB200: &cub200
  29. folder: birds
  30. annotations: cub200_11_ORIGINAL
  31. CARS: &cars
  32. folder: cars
  33. annotations: ORIGINAL
  34. ############ Existing Part Annotations and Part Features
  35. PART_TYPES:
  36. GLOBAL: &parts_global
  37. feature_suffix: ""
  38. UNI: &parts_uni
  39. # <<: *cub200
  40. is_uniform: true
  41. feature_suffix: _26parts_uniform
  42. rescale_size: !!int -1
  43. scales:
  44. - 0.2
  45. GT: &parts_gt
  46. # <<: *cub200
  47. feature_suffix: _16parts_gt
  48. rescale_size: !!int -1
  49. scales:
  50. - 0.31
  51. GT2: &parts_gt2
  52. # <<: *cub200
  53. annotations: cub200_11_regrouped
  54. feature_suffix: _5parts_gt
  55. rescale_size: !!int -1
  56. scales:
  57. - 0.31
  58. NAC: &parts_nac
  59. # <<: *cub200
  60. annotations: NAC/2017-bilinear
  61. feature_suffix: _20parts
  62. rescale_size: !!int 224
  63. scales:
  64. - 0.31
  65. - 0.45
  66. L1_pred: &parts_l1p
  67. # <<: *cub200
  68. annotations: cub200_11_L1_pred
  69. feature_suffix: _5parts_L1_pred
  70. rescale_size: !!int 299
  71. scales:
  72. - 0.31
  73. L1_full: &parts_l1f
  74. # <<: *cub200
  75. annotations: cub200_11_L1_full
  76. feature_suffix: _5parts_L1_full
  77. rescale_size: !!int 299
  78. scales:
  79. - 0.31
  80. PARTS:
  81. #### No Parts Annotations
  82. CUB200_GLOBAL:
  83. <<: *cub200
  84. <<: *parts_global
  85. CARS_GLOBAL:
  86. <<: *cars
  87. <<: *parts_global
  88. #### With Parts Annotations
  89. CUB200_UNI:
  90. <<: *cub200
  91. <<: *parts_uni
  92. CUB200_GT:
  93. <<: *cub200
  94. <<: *parts_gt
  95. CUB200_GT2:
  96. <<: *cub200
  97. <<: *parts_gt2
  98. CUB200_NAC:
  99. <<: *cub200
  100. <<: *parts_nac
  101. CUB200_L1_pred:
  102. <<: *cub200
  103. <<: *parts_l1p
  104. CUB200_L1_full:
  105. <<: *cub200
  106. <<: *parts_l1f