Peter M. Groen / openbr

Commit 3635375bc81d1ef5e69a74d69c08b0b9894f1aae

Authored by Keyur Patel
1 parent 2cb82045

Added flag to random rects to allow for the random rects to be of the same size … 

…and ensure the patches don't contain black pixels
Inline Side-by-side
Showing 1 changed file with 26 additions and 1 deletions
openbr/plugins/metadata/randomrects.cpp
  View file @3635375
... ... @@ -4,6 +4,9 @@
4 4  
5 5 #include <openbr/plugins/openbr_internal.h>
6 6 #include <openbr/core/opencvutils.h>
  7 +#include "opencv2/imgproc/imgproc.hpp"
  8 +
  9 +using namespace cv;
7 10  
8 11 namespace br
9 12 {
... ... @@ -18,8 +21,15 @@ class RandomRectsTransform : public UntrainableMetaTransform
18 21 Q_OBJECT
19 22 Q_PROPERTY(int numRects READ get_numRects WRITE set_numRects RESET reset_numRects STORED false)
20 23 Q_PROPERTY(int minSize READ get_minSize WRITE set_minSize RESET reset_minSize STORED false)
  24 + Q_PROPERTY(bool sameSize READ get_sameSize WRITE set_sameSize RESET reset_sameSize STORED false)
  25 + Q_PROPERTY(bool filterDarkPatches READ get_filterDarkPatches WRITE set_filterDarkPatches RESET reset_filterDarkPatches STORED false)
  26 + Q_PROPERTY(float bgFgThresh READ get_bgFgThresh WRITE set_bgFgThresh RESET reset_bgFgThresh STORED false)
  27 +
21 28 BR_PROPERTY(int, numRects, 135)
22 29 BR_PROPERTY(int, minSize, 24)
  30 + BR_PROPERTY(bool, sameSize, false)
  31 + BR_PROPERTY(bool, filterDarkPatches, false)
  32 + BR_PROPERTY(float, bgFgThresh, 0.75)
23 33  
24 34 void project(const Template &, Template &) const
25 35 {
... ... @@ -28,13 +38,28 @@ class RandomRectsTransform : public UntrainableMetaTransform
28 38  
29 39 void project(const TemplateList &src, TemplateList &dst) const
30 40 {
  41 + int size = minSize;
31 42 foreach (const Template &t, src) {
32 43 int maxSize = std::min(t.m().rows, t.m().cols);
33 44 for (int i = 0; i < numRects; i++) {
34   - int size = (rand() % (maxSize - minSize)) + minSize;
  45 +
  46 + if (!sameSize){
  47 + size = (rand() % (maxSize - minSize)) + minSize;
  48 + }
  49 +
35 50 int x = rand() % (t.m().cols - size);
36 51 int y = rand() % (t.m().rows - size);
37 52  
  53 + if (filterDarkPatches){
  54 + Mat patch;
  55 + t.m()(cv::Rect(x,y,size,size)).copyTo(patch);
  56 + cv::threshold(patch,patch,5,1,cv::THRESH_BINARY);
  57 + Scalar sumForeground = sum(patch);
  58 + float bgFgRatio = sumForeground[0] / float(minSize * minSize);
  59 + if (bgFgRatio < bgFgThresh)
  60 + continue;
  61 + }
  62 +
38 63 Template out(t.file, t.m());
39 64 out.file.clearRects();
40 65 out.file.appendRect(QRect(x,y,size,size));
... ...