#include <SpatialDetectionNetwork.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 (const int coordinates) |
| Set coordianate size. More... | |
| void | setIouThreshold (float thresh) |
| Set Iou threshold. More... | |
| void | setNumClasses (const int numClasses) |
| Set num classes. More... | |
| YoloSpatialDetectionNetwork (const std::shared_ptr< PipelineImpl > &par, int64_t nodeId) | |
 Public Member Functions inherited from dai::NodeCRTP< SpatialDetectionNetwork, YoloSpatialDetectionNetwork, SpatialDetectionNetworkProperties > | |
| std::unique_ptr< Node > | clone () const override |
| const char * | getName () const override |
Additional Inherited Members | |
| Public Types inherited from dai::NodeCRTP< SpatialDetectionNetwork, YoloSpatialDetectionNetwork, SpatialDetectionNetworkProperties > | |
| using | Properties = SpatialDetectionNetworkProperties |
| Public Attributes inherited from dai::NodeCRTP< SpatialDetectionNetwork, YoloSpatialDetectionNetwork, SpatialDetectionNetworkProperties > | |
| Properties & | properties |
| Underlying properties. More... | |
Detailed Description
YoloSpatialDetectionNetwork node. Yolo-based network with spatial location data.
Definition at line 117 of file SpatialDetectionNetwork.hpp.
Constructor & Destructor Documentation
◆ YoloSpatialDetectionNetwork()
| dai::node::YoloSpatialDetectionNetwork::YoloSpatialDetectionNetwork | ( | const std::shared_ptr< PipelineImpl > & | par, |
| int64_t | nodeId | ||
| ) |
Definition at line 53 of file SpatialDetectionNetwork.cpp.
Member Function Documentation
◆ getAnchorMasks()
| std::map< std::string, std::vector< int > > dai::node::YoloSpatialDetectionNetwork::getAnchorMasks | ( | ) | const |
Get anchor masks.
Definition at line 94 of file SpatialDetectionNetwork.cpp.
◆ getAnchors()
| std::vector< float > dai::node::YoloSpatialDetectionNetwork::getAnchors | ( | ) | const |
Get anchors.
Definition at line 89 of file SpatialDetectionNetwork.cpp.
◆ getCoordinateSize()
| int dai::node::YoloSpatialDetectionNetwork::getCoordinateSize | ( | ) | const |
Get coordianate size.
Definition at line 84 of file SpatialDetectionNetwork.cpp.
◆ getIouThreshold()
| float dai::node::YoloSpatialDetectionNetwork::getIouThreshold | ( | ) | const |
Get Iou threshold.
Definition at line 99 of file SpatialDetectionNetwork.cpp.
◆ getNumClasses()
| int dai::node::YoloSpatialDetectionNetwork::getNumClasses | ( | ) | const |
Get num classes.
Definition at line 79 of file SpatialDetectionNetwork.cpp.
◆ setAnchorMasks()
| void dai::node::YoloSpatialDetectionNetwork::setAnchorMasks | ( | std::map< std::string, std::vector< int >> | anchorMasks | ) |
Set anchor masks.
Definition at line 70 of file SpatialDetectionNetwork.cpp.
◆ setAnchors()
| void dai::node::YoloSpatialDetectionNetwork::setAnchors | ( | std::vector< float > | anchors | ) |
Set anchors.
Definition at line 66 of file SpatialDetectionNetwork.cpp.
◆ setCoordinateSize()
| void dai::node::YoloSpatialDetectionNetwork::setCoordinateSize | ( | const int | coordinates | ) |
Set coordianate size.
Definition at line 62 of file SpatialDetectionNetwork.cpp.
◆ setIouThreshold()
| void dai::node::YoloSpatialDetectionNetwork::setIouThreshold | ( | float | thresh | ) |
Set Iou threshold.
Definition at line 74 of file SpatialDetectionNetwork.cpp.
◆ setNumClasses()
| void dai::node::YoloSpatialDetectionNetwork::setNumClasses | ( | const int | numClasses | ) |
Set num classes.
Definition at line 58 of file SpatialDetectionNetwork.cpp.
The documentation for this class was generated from the following files: