core_detect.hpp

// This file is part of OpenCV project.




// It is subject to the license terms in the LICENSE file found in the top-level directory




// of this distribution and at http://opencv.org/license.html.




#ifndef _OPENCV_DNN_OBJDETECT_CORE_DETECT_HPP_




#define _OPENCV_DNN_OBJDETECT_CORE_DETECT_HPP_








#include <vector>




#include <memory>








#include <opencv2/imgproc.hpp>








namespace

cv



{



namespace
dnn_objdetect


{











typedef
struct




{



int
xmin, xmax;



int
ymin, ymax;



size_t
class_idx;



std::string label_name;



double
class_prob;



}
object;











class
CV_EXPORTS
InferBbox




{



public:



InferBbox(Mat
_delta_bbox,
Mat
_class_scores,
Mat
_conf_scores);







void
filter(double
thresh =  0.8);







std::vector<object>
detections;







protected:



void
transform_bboxes(std::vector<std::vector<double> > *bboxes);







void
final_probability_dist(std::vector<std::vector<double> > *final_probs);







void
transform_bboxes_inv(std::vector<std::vector<double> > *pre,



std::vector<std::vector<double> > *post);







void
assert_predictions(std::vector<std::vector<double> > *min_max_boxes);







void
filter_top_n(std::vector<std::vector<double> > *probs,



std::vector<std::vector<double> > *boxes,



std::vector<std::vector<double> > &top_n_boxes,



std::vector<size_t> &top_n_idxs,



std::vector<double> &top_n_probs);







void
nms_wrapper(std::vector<std::vector<double> > &top_n_boxes,



std::vector<size_t> &top_n_idxs,



std::vector<double> &top_n_probs);







std::vector<bool>
non_maximal_suppression(std::vector<std::vector<double> >



*boxes, std::vector<double> *probs);







void
intersection_over_union(std::vector<std::vector<double> > *boxes,



std::vector<double> *base_box, std::vector<double> *iou);







static
inline
bool
comparator (std::pair<double, size_t> l1,



std::pair<double, size_t> l2)



{



return
l1.first > l2.first;



}







private:



Mat
delta_bbox;



Mat
class_scores;



Mat
conf_scores;







unsigned
int
image_width;



unsigned
int
image_height;







unsigned
int
W, H;



std::vector<std::vector<double> > anchors_values;



std::vector<std::pair<double, double> > anchor_center;



std::vector<std::pair<double, double> > anchor_shapes;







std::vector<std::string> label_map;







unsigned
int
num_classes;



unsigned
int
anchors_per_grid;



size_t
anchors;



double
intersection_thresh;



double
nms_intersection_thresh;



size_t
n_top_detections;



double
epsilon;



};






}
// namespace dnn_objdetect



}
// namespace cv




#endif