IDRandomSampling.cpp 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119
  1. /**
  2. * @file IDRandomSampling.cpp
  3. * @brief random interest point sampling
  4. * @author Erik Rodner
  5. * @date 02/05/2008
  6. */
  7. #include <iostream>
  8. #include "vislearning/features/localfeatures/IDRandomSampling.h"
  9. using namespace OBJREC;
  10. using namespace std;
  11. using namespace NICE;
  12. IDRandomSampling::IDRandomSampling(const Config *conf, int _numSamples) :
  13. numSamples(_numSamples)
  14. {
  15. minScale = conf->gD("IDRandomSampling", "min_scale", 1.0);
  16. srand(time(NULL));
  17. }
  18. IDRandomSampling::~IDRandomSampling()
  19. {
  20. }
  21. int IDRandomSampling::getInterests(const NICE::Image & img,
  22. std::vector<Vector> & positions) const
  23. {
  24. // FIXME: is this function broken ??
  25. double rmin = minScale * 6;
  26. double rmax = std::min(img.width(), img.height()) / 2.0;
  27. size_t len = (size_t) (rmax - rmin);
  28. NICE::Vector p(len);
  29. double sum = 0.0;
  30. for (size_t k = 0; k < len; k++)
  31. {
  32. double r = k + rmin;
  33. p[k] = img.width() * img.height() - 2 * (img.width() + img.height())
  34. * r + 4 * r * r;
  35. sum += p[k];
  36. }
  37. p[0] /= sum;
  38. for (size_t k = 1; k < len; k++)
  39. {
  40. p[k] /= sum;
  41. p[k] = p[k - 1] + p[k];
  42. }
  43. vector<double> sizes;
  44. for (size_t i = 0; i < (size_t) numSamples; i++)
  45. {
  46. double sr = drand48();
  47. int r = len - 1;
  48. while ((p[r] > sr) && (r > 0))
  49. r--;
  50. sizes.push_back(r);
  51. }
  52. sort(sizes.begin(), sizes.end());
  53. for (size_t i = 0; i < sizes.size(); i++)
  54. {
  55. double r = sizes[i];
  56. double scale = (r + rmin) / 6.0;
  57. double x = drand48() * (img.width() - 2 * r) + r;
  58. double y = drand48() * (img.height() - 2 * r) + r;
  59. Vector pos(3);
  60. pos[0] = x;
  61. pos[1] = y;
  62. pos[2] = scale; // FIXME: level = scale??
  63. positions.push_back(pos);
  64. }
  65. fprintf(stderr, "IDRandomSampling finished\n");
  66. return 0;
  67. }
  68. int IDRandomSampling::getInterests(const ImagePyramid & imp,
  69. std::vector<Vector> & positions) const
  70. {
  71. int numLevels = imp.getNumLevels();
  72. for (int j = 0; j < numSamples; j++)
  73. {
  74. int level = rand() % numLevels;
  75. if ((level < 0) || (level >= numLevels))
  76. continue;
  77. const NICE::Image & img = imp.getLevel(level);
  78. int x = rand() % (img.width() + 1);
  79. int y = rand() % (img.height() + 1);
  80. double xo, yo;
  81. imp.getOriginalCoordinates(x, y, level, xo, yo);
  82. Vector pos(3);
  83. pos[0] = xo;
  84. pos[1] = yo;
  85. pos[2] = level;
  86. positions.push_back(pos);
  87. }
  88. fprintf(stderr, "IDRandomSampling finished\n");
  89. return 0;
  90. }