nms.hpp File Reference

#include <torch/torch.h>

Include dependency graph for nms.hpp:

Go to the source code of this file.

Namespaces
namespace	reusex
namespace	reusex::vision

Functions
torch::Tensor	reusex::vision::xyxy_to_xywh (const torch::Tensor &x)
	Convert bounding boxes from (x1,y1,x2,y2) to (cx,cy,w,h) format.
torch::Tensor	reusex::vision::xywh_to_xyxy (const torch::Tensor &x)
	Convert bounding boxes from (cx,cy,w,h) to (x1,y1,x2,y2) format.
torch::Tensor	reusex::vision::nms (const torch::Tensor &bboxes, const torch::Tensor &scores, float iou_threshold=0.45)
	Non-maximum suppression on bounding boxes.
torch::Tensor	reusex::vision::non_max_suppression (torch::Tensor predictions, float confThreshold=0.25, float iouThreshold=0.45, int maxDetections=300)
	YOLO-style non-maximum suppression with class-aware filtering.