22 lines
648 B
C++
22 lines
648 B
C++
#pragma once
|
|
|
|
#include<string>
|
|
|
|
class Yolo
|
|
{
|
|
private:
|
|
cv::Mat image;
|
|
cv::Mat boxes;
|
|
public:
|
|
Yolo(std::string modelPath); // initialize detector here, may need more args
|
|
// void gpuInit(/* args */); // initialize the GPU, may called by the constructor
|
|
void detect(cv::Mat image); // detect objects in the image, return the 2d mat
|
|
// which every row represents for an object
|
|
cv::Mat getBoxes(); // return infomation about the detected objects
|
|
// may called by the detect function
|
|
// void preProcess(/* args */);
|
|
// void postProcess(/* args */);
|
|
};
|
|
|
|
|