Commit 57c7b311ca38a1ff7be4fce649fc131517a0a3fb
1 parent
3848bb3c
remove samplefrommask
Showing
1 changed file
with
0 additions
and
57 deletions
openbr/plugins/imgproc/samplefrommask.cpp deleted
| 1 | -#include <openbr/plugins/openbr_internal.h> | |
| 2 | - | |
| 3 | -using namespace cv; | |
| 4 | - | |
| 5 | -namespace br | |
| 6 | -{ | |
| 7 | - | |
| 8 | -/*! | |
| 9 | - * \ingroup transforms | |
| 10 | - * \brief Samples pixels from a mask. | |
| 11 | - * \author Scott Klum \cite sklum | |
| 12 | - */ | |
| 13 | -class SampleFromMaskTransform : public UntrainableTransform | |
| 14 | -{ | |
| 15 | - Q_OBJECT | |
| 16 | - | |
| 17 | - Q_PROPERTY(int minIndices READ get_minIndices WRITE set_minIndices RESET reset_minIndices STORED false) | |
| 18 | - BR_PROPERTY(int, minIndices, 0) | |
| 19 | - | |
| 20 | - void project(const Template &src, Template &dst) const | |
| 21 | - { | |
| 22 | - dst = src; | |
| 23 | - | |
| 24 | - Mat mask = src.file.get<Mat>("Mask"); | |
| 25 | - Mat indices; | |
| 26 | - findNonZero(mask,indices); | |
| 27 | - | |
| 28 | - if (indices.total() > (size_t)minIndices) { | |
| 29 | - Mat masked; | |
| 30 | - src.m().copyTo(masked, mask); | |
| 31 | - if (src.m().channels() > 1) { | |
| 32 | - dst.m() = Mat(3,indices.total(),CV_32FC1); | |
| 33 | - for (size_t j=0; j<indices.total(); j++) { | |
| 34 | - Vec3b v = masked.at<Vec3b>(indices.at<Point>(j).y,indices.at<Point>(j).x); | |
| 35 | - dst.m().at<float>(0,j) = v[0]; | |
| 36 | - dst.m().at<float>(1,j) = v[1]; | |
| 37 | - dst.m().at<float>(2,j) = v[2]; | |
| 38 | - } | |
| 39 | - } else { | |
| 40 | - dst.m() = Mat(1,indices.total(),src.m().type()); | |
| 41 | - | |
| 42 | - for (size_t j=0; j<indices.total(); j++) | |
| 43 | - dst.m().at<uchar>(0,j) = masked.at<uchar>(indices.at<Point>(j).y,indices.at<Point>(j).x); | |
| 44 | - } | |
| 45 | - } else { | |
| 46 | - dst.file.fte = true; | |
| 47 | - dst.file.remove("Mask"); | |
| 48 | - qWarning("No mask content for %s.",qPrintable(src.file.baseName())); | |
| 49 | - } | |
| 50 | - } | |
| 51 | -}; | |
| 52 | - | |
| 53 | -BR_REGISTER(Transform, SampleFromMaskTransform) | |
| 54 | - | |
| 55 | -} // namespace br | |
| 56 | - | |
| 57 | -#include "imgproc/samplefrommask.moc" |