mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-05-08 08:23:25 +08:00
add model
This commit is contained in:
@@ -212,7 +212,6 @@ void RuntimeOption::SetTrtCacheFile(const std::string& cache_file_path) {
|
||||
trt_serialize_file = cache_file_path;
|
||||
}
|
||||
|
||||
|
||||
bool Runtime::Init(const RuntimeOption& _option) {
|
||||
option = _option;
|
||||
if (option.model_format == Frontend::AUTOREC) {
|
||||
@@ -274,6 +273,7 @@ void Runtime::CreatePaddleBackend() {
|
||||
pd_option.mkldnn_cache_size = option.pd_mkldnn_cache_size;
|
||||
pd_option.use_gpu = (option.device == Device::GPU) ? true : false;
|
||||
pd_option.gpu_id = option.device_id;
|
||||
pd_option.cpu_thread_num = option.cpu_thread_num;
|
||||
FDASSERT(option.model_format == Frontend::PADDLE,
|
||||
"PaddleBackend only support model format of Frontend::PADDLE.");
|
||||
backend_ = new PaddleBackend();
|
||||
|
||||
@@ -60,6 +60,10 @@ class FASTDEPLOY_DECL PPYOLOE : public FastDeployModel {
|
||||
int64_t nms_top_k = 10000;
|
||||
bool normalized = true;
|
||||
bool has_nms_ = false;
|
||||
|
||||
// This function will used to check if this model contains multiclass_nms
|
||||
// and get parameters from the operator
|
||||
void GetNmsInfo();
|
||||
};
|
||||
|
||||
// Read configuration and build pipeline to process input image
|
||||
|
||||
@@ -6,6 +6,10 @@ models/*.pd*
|
||||
models/*.engine
|
||||
models/*.trt
|
||||
models/*.nb
|
||||
models/*param*
|
||||
models/*model*
|
||||
outputs/*.jpg
|
||||
outputs/*.jpeg
|
||||
outputs/*.png
|
||||
outputs/*.png
|
||||
outputs/*.txt
|
||||
outputs/*.json
|
||||
Reference in New Issue
Block a user