9a53b7f226
* v8_seg * fix code style * fix code style * [优化代码风格] * readme中增加yolov8_seg分割的使用 * 更新readme * 更新readme * [fix python infer bug] * Update README.md * Update README.md --------- Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
11 lines
476 B
C++
11 lines
476 B
C++
#pragma once
|
|
#include "NvInfer.h"
|
|
#include <string>
|
|
#include <assert.h>
|
|
|
|
nvinfer1::IHostMemory* buildEngineYolov8Det(nvinfer1::IBuilder* builder,
|
|
nvinfer1::IBuilderConfig* config, nvinfer1::DataType dt, const std::string& wts_path, float& gd, float& gw, int& max_channels);
|
|
|
|
nvinfer1::IHostMemory* buildEngineYolov8Seg(nvinfer1::IBuilder* builder,
|
|
nvinfer1::IBuilderConfig* config, nvinfer1::DataType dt, const std::string& wts_path, float& gd, float& gw, int& max_channels);
|