initial commit

wangrong

2025-01-03 2c4b8838f89bebca97615c5f0bac2c58e0349771

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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#pragma once 
 
#include "base_model.h" 
#include <opencv2/core/core.hpp> 
#include <opencv2/opencv.hpp> 
#include <string> 
#include <vector> 
 
 
namespace trt_vehicle{ 
    class DetectModel: public BaseModel{ 
    public: 
        DetectModel(const std::string& modelPath,float scoreThres,float iouThres); 
        ~DetectModel(); 
 
        //inferential prediction of the picture after pad 
        int predictPadding(std::vector<cv::Mat> imgs,std::vector<std::vector<ObjBox>>& outBoxes, int paddingValue); 
 
    private: 
        int predict(std::vector<cv::Mat> imgs,std::vector<std::vector<ObjBox>>& outBoxes); 
        int setSize(); 
        int extractFeature(std::vector<cv::Mat> imgs); 
         
    private: 
        int m_outputNum; 
        int m_classNum; 
        int m_boxNum; 
        float m_scoreThres; 
        float m_iouThres; 
        float m_ratioH; 
        float m_ratioW; 
 
        std::vector<int> m_hStarts; 
        std::vector<int> m_wStarts; 
 
        float* m_scoreCuda; 
        float* m_boxCuda; 
        float* m_score; 
        float* m_box; 
        float* m_imageDataCuda; 
 
        CudaPredictor m_predictor; 
    }; 
}