1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
| #pragma once
|
|
| #include <opencv2/core.hpp>
| #include "detect_model.h"
| #include "../util/algorithm_util.h"
|
| namespace trt_vehicle {
| class VehicleScanner
| {
| private:
| /* data */
| DetectModel* scannerModel = nullptr;
| std::vector<std::string> partLabels = {"truck",
| "bus",
| "car",
| "suv",
| "van",
| "nowindow",
| "iveco",
| "pickup",
| "wheel"};
| public:
| VehicleScanner(const std::string& scanModelPath, float scoreThres = 0.5, float iouThres = 0.4);
| ~VehicleScanner();
|
| void detect(std::vector<cv::Mat>& img_datas, std::vector<std::vector<trt_vehicle::ObjBox>>& parts);
| };
|
| }
|
|
