Class DetectionNetwork

Inheritance Relationships

Base Type

Class Documentation

class DetectionNetwork : public dai::NodeCRTP<NeuralNetwork, DetectionNetwork, DetectionNetworkProperties>

DetectionNetwork, base for different network specializations.

Public Functions

void setConfidenceThreshold(float thresh)

Specifies confidence threshold at which to filter the rest of the detections.

Parameters:

thresh – Detection confidence must be greater than specified threshold to be added to the list

float getConfidenceThreshold() const

Retrieves threshold at which to filter the rest of the detections.

Returns:

Detection confidence

Public Members

Output out = {*this, "out", Output::Type::MSender, {{DatatypeEnum::ImgDetections, false}}}

Outputs ImgDetections message that carries parsed detection results. Overrides NeuralNetwork ‘out’ with ImgDetections output message type.

Output outNetwork = {*this, "outNetwork", Output::Type::MSender, {{DatatypeEnum::NNData, false}}}

Outputs unparsed inference results.

Public Static Attributes

static constexpr const char *NAME = "DetectionNetwork"

Protected Functions

DetectionNetwork(const std::shared_ptr<PipelineImpl> &par, int64_t nodeId)
DetectionNetwork(const std::shared_ptr<PipelineImpl> &par, int64_t nodeId, std::unique_ptr<Properties> props)