info.yml 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  1. BASE_DIR: /home/korsch1/korsch/
  2. DATA_DIR: datasets
  3. MODEL_DIR: models
  4. ############ Existing models
  5. MODELS:
  6. InceptionV3: &inception
  7. folder: inception
  8. weights: ft_inat/model.ckpt.npz
  9. ft_weights: ft_inat/model.ckpt.npz
  10. TFInceptionV3: &inception_tf
  11. folder: inception_tf
  12. weights: ft_inat/inception_v3_iNat_299.ckpt
  13. ft_weights: ft_inat/inception_v3_iNat_299.ckpt
  14. ResNet50: &resnet50
  15. folder: resnet
  16. weights: model.npz
  17. ft_weights: ft_cub200/model.npz
  18. ############ Existing Datasets
  19. DATASETS:
  20. CUB200: &cub200
  21. folder: birds
  22. annotations: cub200_11_ORIGINAL
  23. CARS: &cars
  24. folder: cars
  25. annotations: ORIGINAL
  26. ############ Existing Part Annotations and Part Features
  27. PARTS:
  28. CARS_UNI: &cars_uni
  29. <<: *cars
  30. is_uniform: true
  31. feature_suffix: 26parts_uniform
  32. rescale_size: !!int -1
  33. scales:
  34. - 0.2
  35. UNI: &parts_uni
  36. <<: *cub200
  37. is_uniform: true
  38. feature_suffix: 26parts_uniform
  39. rescale_size: !!int -1
  40. scales:
  41. - 0.2
  42. GT: &parts_gt
  43. <<: *cub200
  44. feature_suffix: 16parts_gt
  45. rescale_size: !!int -1
  46. scales:
  47. - 0.31
  48. GT2: &parts_gt2
  49. <<: *cub200
  50. annotations: cub200_11_regrouped
  51. feature_suffix: 5parts_gt
  52. rescale_size: !!int -1
  53. scales:
  54. - 0.31
  55. NAC: &parts_nac
  56. <<: *cub200
  57. annotations: NAC/2017-bilinear
  58. feature_suffix: 20parts
  59. rescale_size: !!int 224
  60. scales:
  61. - 0.31
  62. - 0.45
  63. L1_pred: &parts_l1p
  64. <<: *cub200
  65. annotations: cub200_11_L1_pred
  66. feature_suffix: 5parts_L1_pred
  67. rescale_size: !!int 299
  68. scales:
  69. - 0.31
  70. L1_full: &parts_l1f
  71. <<: *cub200
  72. annotations: cub200_11_L1_full
  73. feature_suffix: 5parts_L1_full
  74. rescale_size: !!int 299
  75. scales:
  76. - 0.31