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
| #pragma once
| #include <map>
| #include <string>
| #include <vector>
| #include "NvInfer.h"
|
| std::map<std::string, nvinfer1::Weights> loadWeights(const std::string file);
|
| nvinfer1::IElementWiseLayer* convBnSiLU(nvinfer1::INetworkDefinition* network,
| std::map<std::string, nvinfer1::Weights> weightMap, nvinfer1::ITensor& input,
| int ch, int k, int s, int p, std::string lname);
|
| nvinfer1::IElementWiseLayer* C2F(nvinfer1::INetworkDefinition* network,
| std::map<std::string, nvinfer1::Weights> weightMap, nvinfer1::ITensor& input, int c1,
| int c2, int n, bool shortcut, float e, std::string lname);
|
| nvinfer1::IElementWiseLayer* C2(nvinfer1::INetworkDefinition* network,
| std::map<std::string, nvinfer1::Weights>& weightMap, nvinfer1::ITensor& input, int c1,
| int c2, int n, bool shortcut, float e, std::string lname);
|
| nvinfer1::IElementWiseLayer* SPPF(nvinfer1::INetworkDefinition* network,
| std::map<std::string, nvinfer1::Weights> weightMap, nvinfer1::ITensor& input, int c1,
| int c2, int k, std::string lname);
|
| nvinfer1::IShuffleLayer* DFL(nvinfer1::INetworkDefinition* network, std::map<std::string, nvinfer1::Weights> weightMap,
| nvinfer1::ITensor& input, int ch, int grid, int k, int s, int p, std::string lname);
|
| nvinfer1::IPluginV2Layer* addYoLoLayer(nvinfer1::INetworkDefinition* network,
| std::vector<nvinfer1::IConcatenationLayer*> dets, const int* px_arry,
| int px_arry_num, bool is_segmentation, bool is_pose);
|
|
