ReUseX  0.0.5
3D Point Cloud Processing for Building Reuse
Loading...
Searching...
No Matches
reusex::vision::libtorch::LibTorchData Struct Reference

Data carrier for LibTorch YOLO inference pipeline. More...

#include <Data.hpp>

Inheritance diagram for reusex::vision::libtorch::LibTorchData:
Collaboration diagram for reusex::vision::libtorch::LibTorchData:

Public Attributes

cv::Mat image
 Input: raw image (BGR, HWC).
int target_size = 640
 Letterbox target dimension.
float letterbox_scale = 1.0f
 Scale applied during letterbox.
cv::Size original_size
 Pre-letterbox dimensions.
cv::Mat label_image
 Output: label image (CV_16U, class+1).
float confidence_threshold = 0.25f
 Detection confidence threshold.
float iou_threshold = 0.45f
 NMS IoU threshold.
int max_detections = 300
 Maximum detections per image.

Additional Inherited Members

Public Member Functions inherited from reusex::vision::IData
virtual ~IData ()=default

Detailed Description

Data carrier for LibTorch YOLO inference pipeline.

Holds the input image, preprocessing parameters, and output label image for a single sample processed through the LibTorch backend.

Definition at line 18 of file Data.hpp.

Member Data Documentation

◆ confidence_threshold

float reusex::vision::libtorch::LibTorchData::confidence_threshold = 0.25f

Detection confidence threshold.

Definition at line 24 of file Data.hpp.

◆ image

cv::Mat reusex::vision::libtorch::LibTorchData::image

Input: raw image (BGR, HWC).

Definition at line 19 of file Data.hpp.

◆ iou_threshold

float reusex::vision::libtorch::LibTorchData::iou_threshold = 0.45f

NMS IoU threshold.

Definition at line 25 of file Data.hpp.

◆ label_image

cv::Mat reusex::vision::libtorch::LibTorchData::label_image

Output: label image (CV_16U, class+1).

Definition at line 23 of file Data.hpp.

◆ letterbox_scale

float reusex::vision::libtorch::LibTorchData::letterbox_scale = 1.0f

Scale applied during letterbox.

Definition at line 21 of file Data.hpp.

◆ max_detections

int reusex::vision::libtorch::LibTorchData::max_detections = 300

Maximum detections per image.

Definition at line 26 of file Data.hpp.

◆ original_size

cv::Size reusex::vision::libtorch::LibTorchData::original_size

Pre-letterbox dimensions.

Definition at line 22 of file Data.hpp.

◆ target_size

int reusex::vision::libtorch::LibTorchData::target_size = 640

Letterbox target dimension.

Definition at line 20 of file Data.hpp.


The documentation for this struct was generated from the following file:
  • libs/reusex/include/vision/libtorch/Data.hpp