createObject.hpp

Go to the documentation of this file.

#pragma once
#include "reusex/vision/common/object.hpp"
 
#include <string>
#include <vector>
 
namespace cv {
class Mat;
}
 
namespace ReUseX::vision::common::object {

DetectionBox createBox(float left, float top, float right, float bottom,
                       float score, int class_id,
                       const std::string &class_name);

DetectionBox createPositionBox(float left, float top, float right, float bottom,
                               float score, int class_id,
                               const std::string &class_name);

DetectionBox createTrackBox(float left, float top, float right, float bottom,
                            float score, int track_id,
                            const std::string &class_name);

DetectionBox createTrackBox(float left, float top, float right, float bottom,
                            float score, int track_id,
                            const std::string &class_name,
                            const object::Pose &pose);

DetectionBox createTrackBox(float left, float top, float right, float bottom,
                            float score, int track_id,
                            const std::string &class_name,
                            const object::Obb &obb);

DetectionBox createTrackBox(float left, float top, float right, float bottom,
                            float score, int track_id,
                            const std::string &class_name,
                            const object::Segmentation &seg);

DetectionBox createObbBox(float cx, float cy, float w, float h, float angle,
                          float score, int class_id,
                          const std::string &class_name);

DetectionBox createPoseBox(float left, float top, float right, float bottom,
                           const std::vector<PosePoint> &pose_points,
                           float score, int class_id,
                           const std::string &class_name);

DetectionBox createSegmentationBox(float left, float top, float right,
                                   float bottom, const cv::Mat &mask,
                                   float score, int class_id,
                                   const std::string &class_name);

DetectionBox createDepthProBox(const cv::Mat &depth, float fog_data);

DetectionBox createDepthAnythingBox(const cv::Mat &depth);
 
} // namespace ReUseX::vision::common::object