ANSCENTER/ANSCORE
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Initial setup for CLion

Browse Source
This commit is contained in:
Tuan Nghia Nguyen
2026-03-28 16:54:11 +11:00
parent 239cc02591
commit 7b4134133c
1136 changed files with 811916 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
engines/TensorRTAPI/include/interfaces/IEngine.h Normal file
View File

@@ -0,0 +1,20 @@
#pragma once
#include <vector>
#include <array>
#include <opencv2/core/cuda.hpp>
#include "NvInfer.h" // Include for nvinfer1::Dims and nvinfer1::Dims3
template <typename T>
class IEngine {
public:
virtual ~IEngine() = default;
virtual bool buildLoadNetwork(std::string onnxModelPath, const std::array<float, 3> &subVals = {0.f, 0.f, 0.f},
const std::array<float, 3> &divVals = {1.f, 1.f, 1.f}, bool normalize = true) = 0;
virtual bool loadNetwork(std::string trtModelPath, const std::array<float, 3> &subVals = {0.f, 0.f, 0.f},
const std::array<float, 3> &divVals = {1.f, 1.f, 1.f}, bool normalize = true) = 0;
virtual bool runInference(const std::vector<std::vector<cv::cuda::GpuMat>> &inputs,
std::vector<std::vector<std::vector<T>>> &featureVectors) = 0;
virtual const std::vector<nvinfer1::Dims3> &getInputDims() const = 0;
virtual const std::vector<nvinfer1::Dims> &getOutputDims() const = 0;
};