-
Notifications
You must be signed in to change notification settings - Fork 261
/
CVTracker.h
107 lines (81 loc) · 2.81 KB
/
CVTracker.h
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
#ifndef OPENSHOT_CVTRACKER_H
#define OPENSHOT_CVTRACKER_H
#include <google/protobuf/util/time_util.h>
#define int64 opencv_broken_int
#define uint64 opencv_broken_uint
#include <opencv2/opencv.hpp>
#include <opencv2/tracking.hpp>
#include <opencv2/core.hpp>
#undef uint64
#undef int64
#include <fstream>
#include "Clip.h"
#include "KeyFrame.h"
#include "Frame.h"
#include "Json.h"
#include "ProcessingController.h"
#include "trackerdata.pb.h"
using namespace std;
using google::protobuf::util::TimeUtil;
// Tracking info struct
struct FrameData{
size_t frame_id = -1;
float rotation = 0;
int x1 = -1;
int y1 = -1;
int x2 = -1;
int y2 = -1;
// Constructors
FrameData()
{}
FrameData( size_t _frame_id)
{frame_id = _frame_id;}
FrameData( size_t _frame_id , float _rotation, int _x1, int _y1, int _x2, int _y2)
{
frame_id = _frame_id;
rotation = _rotation;
x1 = _x1;
y1 = _y1;
x2 = _x2;
y2 = _y2;
}
};
class CVTracker {
private:
std::map<size_t, FrameData> trackedDataById; // Save tracked data
std::string trackerType; // Name of the chosen tracker
cv::Ptr<cv::Tracker> tracker; // Pointer of the selected tracker
cv::Rect2d bbox; // Bounding box coords
std::string protobuf_data_path; // Path to protobuf data file
uint progress; // Pre-processing effect progress
/// Will handle a Thread safely comutication between ClipProcessingJobs and the processing effect classes
ProcessingController *processingController;
bool json_interval;
size_t start;
size_t end;
// Initialize the tracker
bool initTracker(cv::Mat &frame, size_t frameId);
// Update the object tracker according to frame
bool trackFrame(cv::Mat &frame, size_t frameId);
public:
// Constructor
CVTracker(std::string processInfoJson, ProcessingController &processingController);
// Set desirable tracker method
cv::Ptr<cv::Tracker> selectTracker(std::string trackerType);
// Track object in the hole clip or in a given interval
// If start, end and process_interval are passed as argument, clip will be processed in [start,end)
void trackClip(openshot::Clip& video, size_t _start=0, size_t _end=0, bool process_interval=false);
// Get tracked data for a given frame
FrameData GetTrackedData(size_t frameId);
/// Protobuf Save and Load methods
// Save protobuf file
bool SaveTrackedData();
// Add frame tracked data into protobuf message.
void AddFrameDataToProto(libopenshottracker::Frame* pbFrameData, FrameData& fData);
// Load protobuf file
bool LoadTrackedData();
/// Get and Set JSON methods
void SetJson(const std::string value); ///< Load JSON string into this object
void SetJsonValue(const Json::Value root); ///< Load Json::Value into this object
};
#endif