24 #ifndef __TRT_UTILS_H__
25 #define __TRT_UTILS_H__
37 #define UNUSED(expr) (void)(expr)
38 #define DIVUP(n, d) ((n) + (d)-1) / (d)
40 std::string
trim(std::string s);
41 float clamp(
const float val,
const float minVal,
const float maxVal);
42 bool fileExists(
const std::string fileName,
bool verbose =
true);
43 std::vector<float>
loadWeights(
const std::string weightsFilePath,
const std::string& networkType);
49 nvinfer1::ILayer*
netAddMaxpool(
int layerIdx, std::map<std::string, std::string>& block,
50 nvinfer1::ITensor* input, nvinfer1::INetworkDefinition* network);
51 nvinfer1::ILayer*
netAddConvLinear(
int layerIdx, std::map<std::string, std::string>& block,
52 std::vector<float>& weights,
53 std::vector<nvinfer1::Weights>& trtWeights,
int& weightPtr,
54 int& inputChannels, nvinfer1::ITensor* input,
55 nvinfer1::INetworkDefinition* network);
56 nvinfer1::ILayer*
netAddConvBNLeaky(
int layerIdx, std::map<std::string, std::string>& block,
57 std::vector<float>& weights,
58 std::vector<nvinfer1::Weights>& trtWeights,
int& weightPtr,
59 int& inputChannels, nvinfer1::ITensor* input,
60 nvinfer1::INetworkDefinition* network);
61 nvinfer1::ILayer*
netAddUpsample(
int layerIdx, std::map<std::string, std::string>& block,
62 std::vector<float>& weights,
63 std::vector<nvinfer1::Weights>& trtWeights,
int& inputChannels,
64 nvinfer1::ITensor* input, nvinfer1::INetworkDefinition* network);
65 void printLayerInfo(std::string layerIndex, std::string layerName, std::string layerInput,
66 std::string layerOutput, std::string weightPtr);