frames.cpp
1.9 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#include "openbr_internal.h"
namespace br
{
/*!
* \ingroup transforms
* \brief Passes along n sequential frames to the next transform.
* \author Josh Klontz \cite jklontz
*
* For a video with m frames, AggregateFrames would create a total of m-n+1 sequences ([0,n] ... [m-n+1, m]).
*/
class AggregateFrames : public TimeVaryingTransform
{
Q_OBJECT
Q_PROPERTY(int n READ get_n WRITE set_n RESET reset_n STORED false)
BR_PROPERTY(int, n, 1)
TemplateList buffer;
public:
AggregateFrames() : TimeVaryingTransform(false) {}
private:
void train(const TemplateList &data)
{
(void) data;
}
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);
dst.file = buffer.takeFirst().file;
}
void store(QDataStream &stream) const
{
(void) stream;
}
void load(QDataStream &stream)
{
(void) stream;
}
};
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 UntrainableMetaTransform
{
Q_OBJECT
Q_PROPERTY(int n READ get_n WRITE set_n RESET reset_n STORED false)
BR_PROPERTY(int, n, 1)
void project(const TemplateList &src, TemplateList &dst) const
{
if (src.first().file.get<int>("FrameNumber") % n != 0) return;
dst = src;
}
void project(const Template &src, Template &dst) const
{
(void) src; (void) dst; qFatal("shouldn't be here");
}
};
BR_REGISTER(Transform, DropFrames)
} // namespace br
#include "frames.moc"