Add YOLOv8 support
This commit is contained in:
@@ -23,7 +23,6 @@
|
||||
* https://www.github.com/marcoslucianops
|
||||
*/
|
||||
|
||||
|
||||
#ifndef __UTILS_H__
|
||||
#define __UTILS_H__
|
||||
|
||||
@@ -36,11 +35,17 @@
|
||||
#include "NvInfer.h"
|
||||
|
||||
std::string trim(std::string s);
|
||||
|
||||
float clamp(const float val, const float minVal, const float maxVal);
|
||||
|
||||
bool fileExists(const std::string fileName, bool verbose = true);
|
||||
|
||||
std::vector<float> loadWeights(const std::string weightsFilePath, const std::string& networkType);
|
||||
|
||||
std::string dimsToString(const nvinfer1::Dims d);
|
||||
|
||||
int getNumChannels(nvinfer1::ITensor* t);
|
||||
|
||||
void printLayerInfo(
|
||||
std::string layerIndex, std::string layerName, std::string layerInput, std::string layerOutput, std::string weightPtr);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user