utils.hpp File Reference

#include <torch/torch.h>
#include <opencv2/core/mat.hpp>
#include <opencv2/core/types.hpp>

Include dependency graph for utils.hpp:

Go to the source code of this file.

Namespaces
namespace	ReUseX
namespace	ReUseX::vision

Functions
float	ReUseX::vision::generate_scale (cv::Mat &image, const cv::Size &target_size, bool scale_up=false)
float	ReUseX::vision::letterbox (cv::Mat &input_image, cv::Mat &output_image, const cv::Size &target_size)
float	ReUseX::vision::cropbox (cv::Mat &input_image, cv::Mat &output_image, const cv::Size &target_size)
torch::Tensor	ReUseX::vision::xyxy_to_xywh (const torch::Tensor &x)
torch::Tensor	ReUseX::vision::xywh_to_xyxy (const torch::Tensor &x)
torch::Tensor	ReUseX::vision::nms (const torch::Tensor &bboces, const torch::Tensor &scores, float iou_threshold=0.45)
torch::Tensor	ReUseX::vision::non_max_suppression (torch::Tensor &predictions, float confThreshold=0.25, float iouThreshold=0.45, int maxDetections=300)
torch::Tensor	ReUseX::vision::xyxy2xywh (const torch::Tensor &x)
torch::Tensor	ReUseX::vision::xywh2xyxy (const torch::Tensor &x)