yolov8-opencv-onnx/yolov8_pose.h

#pragma once
#include<iostream>
#include<opencv2/opencv.hpp>
#include "yolov8_utils.h"

class Yolov8Pose {
public:
	Yolov8Pose() {}
	~Yolov8Pose() {}

	bool ReadModel(cv::dnn::Net& net, std::string& netPath, bool isCuda);
	bool Detect(cv::Mat& srcImg, cv::dnn::Net& net, std::vector<OutputParams>& output);

	int _netWidth = 640;   //ONNX图片输入宽度
	int _netHeight = 640;  //ONNX图片输入高度


	//类别名，自己的模型需要修改此项
	std::vector<std::string> _className ={ "person" };

private:
	float _classThreshold = 0.25;
	float _nmsThreshold = 0.45;
	//float _keyPointThreshold = 0.5;
};