Skip to content

Commit f6c6f00

Browse files
author
SzabolcsGergely
committed
Add API to set trackingPerClass in ObjectTracker node
1 parent c8cb7ca commit f6c6f00

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
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)