This file contains the implementation of the TorchObjectDetection class, which is used to detect and store information about objects using a PyTorch model. More...

#include "../../util/vision/ObjectDetectionUtility.hpp"
#include "../../util/vision/YOLOModel.hpp"

Include dependency graph for TorchObjectDetection.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
namespace	torchobject
	Namespace containing functions related to torch object detections operations on images using PyTorch.

Functions
std::vector< objectdetectutils::Object >	torchobject::Detect (const cv::Mat &cvFrame, yolomodel::pytorch::PyTorchInterpreter &trPyTorchDetector, const float fMinObjectConfidence=0.40f, const float fNMSThreshold=0.60f)
	Detects objects in the given image using a PyTorch model.

void	torchobject::DrawDetections (cv::Mat &cvDetectionsFrame, const std::vector< objectdetectutils::Object > &vDetectedTags)
	Given a vector of objectdetectutils::Object structs draw each tag corner and confidence onto the given image.

Detailed Description

This file contains the implementation of the TorchObjectDetection class, which is used to detect and store information about objects using a PyTorch model.

Author: clayjay3 (clayt.nosp@m.onra.nosp@m.ycowe.nosp@m.n@gm.nosp@m.ail.c.nosp@m.om)

Date: 2025-05-05

Namespaces

Functions

Detailed Description