kint8 量化模型封装代码

sunty

2022-03-21 d0a24896f95b4e060011852f80048ebfb0bf5f55

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#ifndef CLASS_YOLOV4_H_
#define CLASS_YOLOV4_H_
#include "yolo.h"
class YoloV4 :public Yolo
{
public:
    YoloV4(
        const NetworkInfo &network_info_,
        const InferParams &infer_params_);
private:
    std::vector<BBoxInfo> decodeTensor(const int imageIdx,
        const int imageH,
        const int imageW,
        const TensorInfo& tensor) override;
};
 
#endif