[cmake]C++使用yolov12结合bytetrack实现目标追踪演示

#include <iostream>
#include <opencv2/opencv.hpp>
#include "YOLO12.hpp" 
#include "cmdline.h"
#include "utils_yolo.h"

using namespace std;
using namespace cv;


int main(int argc, char* argv[])
{
    const float confThreshold = 0.3f;
    const float iouThreshold = 0.4f;

   
    bool isGPU =false;
    const std::string labelsPath = "coco.names";
    const std::string modelPath = "yolov12n.onnx";
// Initialize the YOLO detector
    YOLO12Detector detector(modelPath, labelsPath, isGPU); 

        std::cout << "Model was initialized." << std::endl;
    std::vector<Detection> result;
	cv::VideoCapture cap("car.mp4");

    // Get video properties
    int img_w = static_cast<int>(cap.get(cv::CAP_PROP_FRAME_WIDTH));
    int img_h = static_cast<int>(cap.get(cv::CAP_PROP_FRAME_HEIGHT));
    int fps = static_cast<int>(cap.get(cv::CAP_PROP_FPS));
    std::cout << "Video properties: " << img_w << "x" << img_h << ", " << fps << " FPS" << std::endl;

	long nFrame = static_cast<long>(cap.get(cv::CAP_PROP_FRAME_COUNT));
	if (!cap.isOpened())
	{
		std::cout << "open capture failured!" << std::endl;
		return -1;
	}
	Mat frame;
	BYTETracker tracker(fps, 30);
	int num_frames = 0;
	int keyvalue = 0;
    int total_ms = 1;
	while (true)
	{
		cap.read(frame);
		if (frame.empty())
		{
			std::cout << "read to end" << std::endl;
			break;
		}
		num_frames++;
		auto start = std::chrono::system_clock::now();
		result.clear();
		 // Detect objects in the frame
        result = detector.detect(frame);
            // Draw bounding boxes on the frame
       detector.drawBoundingBoxMask(frame, result); // Uncomment for mask drawing
       std::vector<STrack> output_stracks = tracker.update(result);
		auto end = chrono::system_clock::now();
		total_ms = total_ms + std::chrono::duration_cast<chrono::microseconds>(end - start).count();
		for (int i = 0; i < output_stracks.size(); i++)
		{
			vector<float> tlwh = output_stracks[i].tlwh;
			bool vertical = tlwh[2] / tlwh[3] > 1.6;
			if (tlwh[2] * tlwh[3] > 20 && !vertical)
			{
				Scalar s = tracker.get_color(output_stracks[i].track_id);
				putText(frame, format("%d", output_stracks[i].track_id), Point(tlwh[0], tlwh[1] - 5),
						0, 0.6, Scalar(0, 0, 255), 2, LINE_AA);
				cv::rectangle(frame, Rect(tlwh[0], tlwh[1], tlwh[2], tlwh[3]), s, 2);
			}
		}
		cv::putText(frame, format("frame: %d fps: %d num: %d", num_frames, num_frames * 1000000 / total_ms, (int)output_stracks.size()),
				Point(0, 30), 0, 0.6, cv::Scalar(0, 0, 255), 2, cv::LINE_AA);

		cv::imshow("demo", frame);
		keyvalue =cv:: waitKey(1);
		if (keyvalue == 113 || keyvalue == 81)
		{
			break;
		}

	}
	cap.release();


    return 0;
}