DetectionParserOptions.hpp

Go to the documentation of this file.

#pragma once
 
#include "depthai-shared/common/DetectionNetworkType.hpp"
#include "depthai-shared/utility/Serialization.hpp"
 
namespace dai {
 
struct DetectionParserOptions {
    DetectionNetworkType nnFamily;
    float confidenceThreshold;
 
    int classes;
    int coordinates;
    std::vector<float> anchors;
    std::map<std::string, std::vector<int>> anchorMasks;
    float iouThreshold;
};
 
DEPTHAI_SERIALIZE_EXT(DetectionParserOptions, nnFamily, confidenceThreshold, classes, coordinates, anchors, anchorMasks, iouThreshold);
 
}  // namespace dai