apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-07 09:31:35 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[feature][vision] Add YOLOv7 End2End model with TRT NMS (#157)

Browse Source 
* [feature][vision] Add YOLOv7 End2End model with TRT NMS

* [docs] update yolov7end2end_trt examples docs

Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
DefTruth
2022-08-30 15:02:48 +08:00
committed by GitHub
parent 30bb233db8
commit 3c1330e896
14 changed files with 902 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
csrc/fastdeploy/fastdeploy_model.cc
View File

@@ -27,7 +27,9 @@ bool FastDeployModel::InitRuntime() {
}
if (runtime_option.backend != Backend::UNKNOWN) {
if (!IsBackendAvailable(runtime_option.backend)) {
FDERROR << Str(runtime_option.backend) << " is not compiled with current FastDeploy library." << std::endl;
FDERROR << Str(runtime_option.backend)
<< " is not compiled with current FastDeploy library."
<< std::endl;
return false;
}
@@ -70,7 +72,7 @@ bool FastDeployModel::InitRuntime() {
FDWARNING << "FastDeploy will choose " << Str(valid_gpu_backends[0])
<< " for model inference." << std::endl;
} else {
FDASSERT(valid_gpu_backends.size() > 0,
FDASSERT(valid_cpu_backends.size() > 0,
"There's no valid cpu backend for %s.", ModelName().c_str());
FDWARNING << "FastDeploy will choose " << Str(valid_cpu_backends[0])
<< " for model inference." << std::endl;