ObjectTrackerProperties.hpp

Go to the documentation of this file.

#pragma once
 
// std
#include <vector>
 
// project
#include "depthai-shared/common/optional.hpp"
#include "depthai-shared/properties/Properties.hpp"
 
namespace dai {
 
enum class TrackerType : std::int32_t {
    SHORT_TERM_KCF = 1,
    SHORT_TERM_IMAGELESS = 3,
    ZERO_TERM_IMAGELESS = 5,
    ZERO_TERM_COLOR_HISTOGRAM = 6
};
 
enum class TrackerIdAssignmentPolicy : std::int32_t {
    UNIQUE_ID,
    SMALLEST_ID
};
 
struct ObjectTrackerProperties : PropertiesSerializable<Properties, ObjectTrackerProperties> {
    float trackerThreshold = 0.0;
    std::int32_t maxObjectsToTrack = 60;
    std::vector<std::uint32_t> detectionLabelsToTrack;
    TrackerType trackerType = TrackerType::ZERO_TERM_IMAGELESS;
    TrackerIdAssignmentPolicy trackerIdAssignmentPolicy = TrackerIdAssignmentPolicy::UNIQUE_ID;
    bool trackingPerClass = true;
};
 
DEPTHAI_SERIALIZE_EXT(
    ObjectTrackerProperties, trackerThreshold, maxObjectsToTrack, detectionLabelsToTrack, trackerType, trackerIdAssignmentPolicy, trackingPerClass);
 
}  // namespace dai