Various helper functions

Brendan Klare
1 parent 54a98a97
Showing 5 changed files with 75 additions and 0 deletions
openbr/core/eigenutils.cpp
openbr/core/eigenutils.h
openbr/core/opencvutils.cpp
openbr/core/opencvutils.h
openbr/plugins/template.cpp
@@ -36,6 +36,15 @@ void writeEigen(VectorXd X, QString filename) {
     format->write(br::Template(m));
 }
+void writeEigen(VectorXf X, QString filename) {
+    Mat m(X.size(),1,CV_32FC1);
+    for (int i = 0; i < X.rows(); i++) {
+        m.at<float>(i,0) = X(i);
+    }
+    QScopedPointer<br::Format> format(br::Factory<br::Format>::make(filename));
+    format->write(br::Template(m));
+}
+
 void printEigen(Eigen::MatrixXd X) {
     for (int i = 0; i < X.rows(); i++) {
         QString str;
@@ -24,6 +24,7 @@
 void writeEigen(Eigen::MatrixXf X, QString filename);
 void writeEigen(Eigen::MatrixXd X, QString filename);
 void writeEigen(Eigen::VectorXd X, QString filename);
+void writeEigen(Eigen::VectorXf X, QString filename);
 void printEigen(Eigen::MatrixXd X);
 template<typename _Scalar, int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>
@@ -300,6 +300,30 @@ QList&lt;QRectF&gt; OpenCVUtils::fromRects(const QList&lt;Rect&gt; &amp;cvRects)
     return qRects;
 }
+float OpenCVUtils::overlap(const Rect &rect1, const Rect &rect2) {
+    float left = max(rect1.x, rect2.x);
+    float top = max(rect1.y, rect2.y);
+    float right = min(rect1.x + rect1.width, rect2.x + rect2.width);
+    float bottom = min(rect1.y + rect1.height, rect2.y + rect2.height);
+
+    float overlap = (right - left + 1) * (top - bottom + 1) / max(rect1.width * rect1.height, rect2.width * rect2.height);
+    if (overlap < 0)
+        return 0;
+    return overlap;
+}
+
+float OpenCVUtils::overlap(const QRectF &rect1, const QRectF &rect2) {
+    float left = max(rect1.x(), rect2.x());
+    float top = max(rect1.y(), rect2.y());
+    float right = min(rect1.x() + rect1.width(), rect2.x() + rect2.width());
+    float bottom = min(rect1.y() + rect1.height(), rect2.y() + rect2.height());
+
+    float overlap = (right - left + 1) * (top - bottom + 1) / max(rect1.width() * rect1.height(), rect2.width() * rect2.height());
+    if (overlap < 0)
+        return 0;
+    return overlap;
+}
+
 bool OpenCVUtils::overlaps(const QList<Rect> &posRects, const Rect &negRect, double overlap)
 {
     foreach (const Rect &posRect, posRects) {
@@ -88,6 +88,8 @@ namespace OpenCVUtils
     QList<cv::Rect> toRects(const QList<QRectF> &qRects);
     QList<QRectF> fromRects(const QList<cv::Rect> &cvRects);
     bool overlaps(const QList<cv::Rect> &posRects, const cv::Rect &negRect, double overlap);
+    float overlap(const cv::Rect &rect1, const cv::Rect &rect2);
+    float overlap(const QRectF &rect1, const QRectF &rect2);
     int getFourcc();
 }
@@ -70,6 +70,45 @@ class RemoveMetadataTransform : public UntrainableTransform
 };
 BR_REGISTER(Transform, RemoveMetadataTransform)
+/*!
+ * \ingroup transforms
+ * \brief Retains only landmarks/points at the provided indices
+ * \author Brendan Klare \cite bklare
+ */
+class SelectPointsTransform : public UntrainableTransform
+{
+    Q_OBJECT
+    Q_PROPERTY(QList<int> indices READ get_indices WRITE set_indices RESET reset_indices STORED false)
+    BR_PROPERTY(QList<int>, indices, QList<int>())
+
+    void project(const Template &src, Template &dst) const
+    {
+        dst = src;
+        QList<QPointF> origPoints = src.file.points();
+        dst.file.clearPoints();
+        for (int i = 0; i < indices.size(); i++)
+            dst.file.appendPoint(origPoints[indices[i]]);
+    }
+};
+
+BR_REGISTER(Transform, SelectPointsTransform)
+
+/*!
+ * \ingroup transforms
+ * \brief Does nothing.
+ */
+class NoneTransform : public UntrainableMetaTransform
+{
+    Q_OBJECT
+
+    void project(const Template &src, Template &dst) const
+    {
+        dst = src;
+    }
+};
+
+BR_REGISTER(Transform, NoneTransform)
+
 } // namespace br
 #include "template.moc"