YoloDetectionNetwork node. Parses Yolo results. More...
#include <DetectionNetwork.hpp>
Public Member Functions | |
| std::map< std::string, std::vector< int > > | getAnchorMasks () const |
| Get anchor masks. More... | |
| std::vector< float > | getAnchors () const |
| Get anchors. More... | |
| int | getCoordinateSize () const |
| Get coordianate size. More... | |
| float | getIouThreshold () const |
| Get Iou threshold. More... | |
| int | getNumClasses () const |
| Get num classes. More... | |
| void | setAnchorMasks (std::map< std::string, std::vector< int >> anchorMasks) |
| Set anchor masks. More... | |
| void | setAnchors (std::vector< float > anchors) |
| Set anchors. More... | |
| void | setCoordinateSize (int coordinates) |
| Set coordianate size. More... | |
| void | setIouThreshold (float thresh) |
| Set Iou threshold. More... | |
| void | setNumClasses (int numClasses) |
| Set num classes. More... | |
| 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) | |
 Public Member Functions inherited from dai::NodeCRTP< DetectionNetwork, YoloDetectionNetwork, DetectionNetworkProperties > | |
| std::unique_ptr< Node > | clone () const override |
| const char * | getName () const override |
Additional Inherited Members | |
| Public Types inherited from dai::NodeCRTP< DetectionNetwork, YoloDetectionNetwork, DetectionNetworkProperties > | |
| using | Properties = DetectionNetworkProperties |
| Public Attributes inherited from dai::NodeCRTP< DetectionNetwork, YoloDetectionNetwork, DetectionNetworkProperties > | |
| Properties & | properties |
| Underlying properties. More... | |
Detailed Description
YoloDetectionNetwork node. Parses Yolo results.
Definition at line 65 of file DetectionNetwork.hpp.
Constructor & Destructor Documentation
◆ YoloDetectionNetwork() [1/2]
| dai::node::YoloDetectionNetwork::YoloDetectionNetwork | ( | const std::shared_ptr< PipelineImpl > & | par, |
| int64_t | nodeId | ||
| ) |
Definition at line 45 of file DetectionNetwork.cpp.
◆ YoloDetectionNetwork() [2/2]
| dai::node::YoloDetectionNetwork::YoloDetectionNetwork | ( | const std::shared_ptr< PipelineImpl > & | par, |
| int64_t | nodeId, | ||
| std::unique_ptr< Properties > | props | ||
| ) |
Definition at line 47 of file DetectionNetwork.cpp.
Member Function Documentation
◆ getAnchorMasks()
| std::map< std::string, std::vector< int > > dai::node::YoloDetectionNetwork::getAnchorMasks | ( | ) | const |
Get anchor masks.
Definition at line 89 of file DetectionNetwork.cpp.
◆ getAnchors()
| std::vector< float > dai::node::YoloDetectionNetwork::getAnchors | ( | ) | const |
Get anchors.
Definition at line 84 of file DetectionNetwork.cpp.
◆ getCoordinateSize()
| int dai::node::YoloDetectionNetwork::getCoordinateSize | ( | ) | const |
Get coordianate size.
Definition at line 79 of file DetectionNetwork.cpp.
◆ getIouThreshold()
| float dai::node::YoloDetectionNetwork::getIouThreshold | ( | ) | const |
Get Iou threshold.
Definition at line 94 of file DetectionNetwork.cpp.
◆ getNumClasses()
| int dai::node::YoloDetectionNetwork::getNumClasses | ( | ) | const |
Get num classes.
Definition at line 74 of file DetectionNetwork.cpp.
◆ setAnchorMasks()
| void dai::node::YoloDetectionNetwork::setAnchorMasks | ( | std::map< std::string, std::vector< int >> | anchorMasks | ) |
Set anchor masks.
Definition at line 65 of file DetectionNetwork.cpp.
◆ setAnchors()
| void dai::node::YoloDetectionNetwork::setAnchors | ( | std::vector< float > | anchors | ) |
Set anchors.
Definition at line 61 of file DetectionNetwork.cpp.
◆ setCoordinateSize()
| void dai::node::YoloDetectionNetwork::setCoordinateSize | ( | int | coordinates | ) |
Set coordianate size.
Definition at line 57 of file DetectionNetwork.cpp.
◆ setIouThreshold()
| void dai::node::YoloDetectionNetwork::setIouThreshold | ( | float | thresh | ) |
Set Iou threshold.
Definition at line 69 of file DetectionNetwork.cpp.
◆ setNumClasses()
| void dai::node::YoloDetectionNetwork::setNumClasses | ( | int | numClasses | ) |
Set num classes.
Definition at line 53 of file DetectionNetwork.cpp.
The documentation for this class was generated from the following files: