Program Listing for File DetectionParserOptions.hpp

↰ Return to documentation for file (include/depthai/common/DetectionParserOptions.hpp)

#pragma once

#include "depthai/common/DetectionNetworkType.hpp"
#include "depthai/common/optional.hpp"
#include "depthai/utility/Serialization.hpp"

namespace dai {

struct DetectionParserOptions {
    DetectionNetworkType nnFamily;
    std::string subtype;
    float confidenceThreshold;

    int classes;
    std::optional<std::vector<std::string>> classNames;
    int coordinates;
    std::vector<float> anchors;
    std::map<std::string, std::vector<int>> anchorMasks;
    std::vector<std::vector<std::vector<float>>> anchorsV2;
    float iouThreshold;
};

DEPTHAI_SERIALIZE_EXT(
    DetectionParserOptions, nnFamily, subtype, confidenceThreshold, classes, classNames, coordinates, anchors, anchorMasks, anchorsV2, iouThreshold);

}  // namespace dai