Merge branch 'master' of https://github.com/biometrics/openbr

Scott Klum
2 parents a166fe5d 29e81b41
Showing 1 changed file with 44 additions and 0 deletions
openbr/plugins/frames.cpp
@@ -29,6 +29,9 @@ private:
     void projectUpdate(const Template &src, Template &dst)
     {
+        // DropFrames will pass on empty Templates
+        // but we only want to use non-dropped frames
+        if (src.empty()) return;
         buffer.append(src);
         if (buffer.size() < n) return;
         foreach (const Template &t, buffer) dst.append(t);
@@ -48,6 +51,47 @@ private:
 BR_REGISTER(Transform, AggregateFrames)
+/*!
+ * \ingroup transforms
+ * \brief Only use one frame every n frames.
+ * \author Austin Blanton \cite imaus10
+ *
+ * For a video with m frames, DropFrames will pass on m/n frames.
+ */
+class DropFrames : public TimeVaryingTransform
+{
+    Q_OBJECT
+    Q_PROPERTY(int n READ get_n WRITE set_n RESET reset_n STORED false)
+    BR_PROPERTY(int, n, 1)
+
+public:
+    DropFrames() : TimeVaryingTransform(false) {}
+
+private:
+    void train(const TemplateList &data)
+    {
+        (void) data;
+    }
+
+    void projectUpdate(const Template &src, Template &dst)
+    {
+        if (src.file.get<int>("FrameNumber") % n != 0) return;
+        dst = src;
+    }
+
+    void store(QDataStream &stream) const
+    {
+        (void) stream;
+    }
+
+    void load(QDataStream &stream)
+    {
+        (void) stream;
+    }
+};
+
+BR_REGISTER(Transform, DropFrames)
+
 } // namespace br
 #include "frames.moc"