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
 
#include "vehicle_feature_encoder.h" 
 
 
namespace trt_vehicle { 
     
    VehicleFeatureEncoder::VehicleFeatureEncoder(const std::string& vehicleFeatureModelPath) { 
        vehicleFeatureModel = new FeatureModel(vehicleFeatureModelPath); 
    } 
     
    VehicleFeatureEncoder::~VehicleFeatureEncoder() 
    { 
        if (vehicleFeatureModel != nullptr) { 
            /* code */ 
            delete vehicleFeatureModel; 
        } 
         
    } 
     
    void VehicleFeatureEncoder::encode(std::vector<cv::Mat>& img_datas, std::vector<std::vector<float>>& features) { 
        features.clear(); 
        for (int i = 0; i < img_datas.size(); i++) { 
            /* code */ 
            auto out = vehicleFeatureModel->predictNoPadding({img_datas[i]}).front(); 
            features.push_back(out); 
        } 
    } 
}