vision
Loading...
Searching...
No Matches
Public Member Functions | Private Member Functions | Private Attributes | List of all members
DnnDetector Class Reference
Inheritance diagram for DnnDetector:
roboticslab::IDetector DnnDetector_ParamsParser

Public Member Functions

bool open (yarp::os::Searchable &config) override
 
bool detect (const yarp::sig::Image &inYarpImg, yarp::os::Bottle &detectedObjects) override
 
- Public Member Functions inherited from DnnDetector_ParamsParser
bool parseParams (const yarp::os::Searchable &config) override
 
std::string getDeviceClassName () const override
 
std::string getDeviceName () const override
 
std::string getDocumentationOfDeviceParams () const override
 
std::vector< std::string > getListOfParams () const override
 

Private Member Functions

void preprocess (const cv::Mat &frame)
 
void postprocess (const cv::Size &size, const std::vector< cv::Mat > &outs, yarp::os::Bottle &detectedObjects)
 

Private Attributes

cv::dnn::Net net
 
std::vector< std::string > classes
 
std::vector< std::string > outNames
 
cv::Scalar mean
 

Additional Inherited Members

- Public Attributes inherited from DnnDetector_ParamsParser
const std::string m_device_classname = {"DnnDetector"}
 
const std::string m_device_name = {"DnnDetector"}
 
bool m_parser_is_strict = false
 
const parser_version_type m_parser_version = {}
 
const std::string m_trainedModel_defaultValue = {"yolov3-tiny/yolov3-tiny.weights"}
 
const std::string m_configDNNModel_defaultValue = {"yolov3-tiny/yolov3-tiny.cfg"}
 
const std::string m_framework_defaultValue = {"darknet"}
 
const std::string m_classesTrainedModel_defaultValue = {"coco-object-categories.txt"}
 
const std::string m_backend_defaultValue = {"cuda"}
 
const std::string m_target_defaultValue = {"cpu"}
 
const std::string m_scale_defaultValue = {"0.00392"}
 
const std::string m_mean_defaultValue = {"0.0"}
 
const std::string m_confThreshold_defaultValue = {"0.1"}
 
const std::string m_nmsThreshold_defaultValue = {"0.4"}
 
std::string m_trainedModel = {"yolov3-tiny/yolov3-tiny.weights"}
 
std::string m_configDNNModel = {"yolov3-tiny/yolov3-tiny.cfg"}
 
std::string m_framework = {"darknet"}
 
std::string m_classesTrainedModel = {"coco-object-categories.txt"}
 
std::string m_backend = {"cuda"}
 
std::string m_target = {"cpu"}
 
float m_scale = {0.00392}
 
double m_mean = {0.0}
 
float m_confThreshold = {0.1}
 
float m_nmsThreshold = {0.4}
 

Member Function Documentation

◆ detect()

bool DnnDetector::detect ( const yarp::sig::Image &  inYarpImg,
yarp::os::Bottle &  detectedObjects 
)
overridevirtual

The documentation for this class was generated from the following files: