Skip to content

Commit 2bc8c15

Browse files
author
SzabolcsGergely
committed
Merge remote-tracking branch 'origin/develop' into HEAD
2 parents 2687c4c + f6c6f00 commit 2bc8c15

File tree

3 files changed

+9
-1
lines changed

3 files changed

+9
-1
lines changed

include/depthai/pipeline/node/ObjectTracker.hpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,11 @@ class ObjectTracker : public NodeCRTP<Node, ObjectTracker, ObjectTrackerProperti
9393
* @param type Tracker ID assignment policy.
9494
*/
9595
void setTrackerIdAssignmentPolicy(TrackerIdAssignmentPolicy type);
96+
97+
/**
98+
* Whether tracker should take into consideration class label for tracking.
99+
*/
100+
void setTrackingPerClass(bool trackingPerClass);
96101
};
97102

98103
} // namespace node

src/pipeline/node/ObjectTracker.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,9 @@ void ObjectTracker::setTrackerType(TrackerType type) {
3232
void ObjectTracker::setTrackerIdAssignmentPolicy(TrackerIdAssignmentPolicy type) {
3333
properties.trackerIdAssignmentPolicy = type;
3434
}
35+
void ObjectTracker::setTrackingPerClass(bool trackingPerClass) {
36+
properties.trackingPerClass = trackingPerClass;
37+
}
3538

3639
} // namespace node
3740
} // namespace dai

0 commit comments

Comments
 (0)