Data.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/IData.hpp"
 
#include <opencv2/core/mat.hpp>
#include <opencv2/core/types.hpp>
 
namespace reusex::vision::libtorch {

struct LibTorchData : IData {
  cv::Mat image;                      
  int target_size = 640;              
  float letterbox_scale = 1.0f;       
  cv::Size original_size;             
  cv::Mat label_image;                
  float confidence_threshold = 0.25f; 
  float iou_threshold = 0.45f;        
  int max_detections = 300;           
};
 
} // namespace reusex::vision::libtorch