add model

This commit is contained in:
jiangjiajun
2022-08-03 08:25:24 +00:00
parent 56bdbe5b0d
commit aeaad8ef83
3 changed files with 10 additions and 2 deletions
+1 -1
View File
@@ -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();
+4
View File
@@ -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
+5 -1
View File
@@ -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