create_object.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2025 Povl Filip Sonne-Frederiksen
//
// SPDX-License-Identifier: GPL-3.0-or-later
 
#pragma once
#include "reusex/vision/common/object.hpp"
 
#include <string>
#include <vector>
 
namespace cv {
class Mat;
}
 
namespace reusex::vision::common::object {

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

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

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

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

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

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

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

DetectionBox create_pose_box(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 create_segmentation_box(float left, float top, float right,
                                     float bottom, const cv::Mat &mask,
                                     float score, int class_id,
                                     const std::string &class_name);

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

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