1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
| #pragma once
|
| #include <opencv2/core.hpp>
| #include "detect_model.h"
| #include "../util/algorithm_util.h"
|
| namespace trt_vehicle {
| class VehicleDetector
| {
| private:
| /* data */
| DetectModel* vehicleModel = nullptr;
| std::vector<std::string> vehicle_labels = {"car", "bus", "truck"};
| public:
| VehicleDetector(const std::string& vehicleModelPath, float vehicleScoreThres = 0.5, float vehicleIouThres = 0.4);
| ~VehicleDetector();
|
| void detect(std::vector<cv::Mat>& img_datas, std::vector<std::vector<trt_vehicle::ObjBox>>& vehicles);
| };
|
| }
|
|
