forked from Smorodov/Multitarget-tracker
-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathBaseTracker.h
More file actions
41 lines (34 loc) · 1.17 KB
/
BaseTracker.h
File metadata and controls
41 lines (34 loc) · 1.17 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
#pragma once
#include <vector>
#include <memory>
#include "defines.h"
#include "trajectory.h"
#include "TrackerSettings.h"
///
/// \brief The CTracker class
///
class BaseTracker
{
public:
BaseTracker() = default;
BaseTracker(const BaseTracker&) = delete;
BaseTracker(BaseTracker&&) = delete;
BaseTracker& operator=(const BaseTracker&) = delete;
BaseTracker& operator=(BaseTracker&&) = delete;
virtual ~BaseTracker(void) = default;
virtual void Update(const regions_t& regions, cv::UMat currFrame, time_point_t frameTime) = 0;
virtual void UpdateMat(const regions_t& regions, cv::Mat currFrame, time_point_t frameTime)
{
cv::UMat frame = currFrame.getUMat(cv::ACCESS_READ);
Update(regions, frame, frameTime);
}
virtual void GetTracks(std::vector<TrackingObject>& tracks) const = 0;
virtual std::vector<TrackingObject> GetTracksCopy() const
{
std::vector<TrackingObject> tracks;
GetTracks(tracks);
return tracks;
}
virtual void GetRemovedTracks(std::vector<track_id_t>& trackIDs) const = 0;
static std::unique_ptr<BaseTracker> CreateTracker(const TrackerSettings& settings, float fps);
};