Class YoloDetectionNetwork

Defined in File DetectionNetwork.hpp

Inheritance Relationships

Base Type

public dai::NodeCRTP< DetectionNetwork, YoloDetectionNetwork, DetectionNetworkProperties > (Template Class NodeCRTP)

Class Documentation

class YoloDetectionNetwork : public dai::NodeCRTP<DetectionNetwork, YoloDetectionNetwork, DetectionNetworkProperties>

YoloDetectionNetwork node. Parses Yolo results.

Public Functions

YoloDetectionNetwork(const std::shared_ptr<PipelineImpl> &par, int64_t nodeId)

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

void setNumClasses(int numClasses): Set num classes.

void setCoordinateSize(int coordinates): Set coordianate size.

void setAnchors(std::vector<float> anchors): Set anchors.

void setAnchorMasks(std::map<std::string, std::vector<int>> anchorMasks): Set anchor masks.

void setIouThreshold(float thresh): Set Iou threshold.

int getNumClasses() const: Get num classes.

int getCoordinateSize() const: Get coordianate size.

std::vector<float> getAnchors() const: Get anchors.

std::map<std::string, std::vector<int>> getAnchorMasks() const: Get anchor masks.

float getIouThreshold() const: Get Iou threshold.