Class YoloSpatialDetectionNetwork

Defined in File SpatialDetectionNetwork.hpp

Inheritance Relationships

Base Type

public dai::NodeCRTP< SpatialDetectionNetwork, YoloSpatialDetectionNetwork, SpatialDetectionNetworkProperties > (Template Class NodeCRTP)

Class Documentation

class YoloSpatialDetectionNetwork : public dai::NodeCRTP<SpatialDetectionNetwork, YoloSpatialDetectionNetwork, SpatialDetectionNetworkProperties>

YoloSpatialDetectionNetwork node. Yolo-based network with spatial location data.

Public Functions

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

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

void setCoordinateSize(const 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.