apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-26 18:10:32 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

[Other] Upgrade runtime module (#1068)

Browse Source 
* Upgrade runtime module

* Update option.h

* Fix build error

* Move enumerates

* little modification

* little modification

* little modification:

* Remove some useless flags
This commit is contained in:
Jason
2023-01-06 13:44:05 +08:00
committed by GitHub
parent 1135d33dd7
commit d7a65e5c70
31 changed files with 1838 additions and 1778 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
fastdeploy/vision/detection/contrib/yolov7end2end_trt.cc
View File

@@ -75,14 +75,14 @@ YOLOv7End2EndTRT::YOLOv7End2EndTRT(const std::string& model_file,
runtime_option.model_format = model_format;
runtime_option.model_file = model_file;
if (runtime_option.device != Device::GPU) {
FDWARNING << Str(runtime_option.device)
FDWARNING << runtime_option.device
<< " is not support for YOLOv7End2EndTRT,"
<< "will fallback to Device::GPU." << std::endl;
runtime_option.device = Device::GPU;
}
if (runtime_option.backend != Backend::UNKNOWN) {
if (runtime_option.backend != Backend::TRT) {
FDWARNING << Str(runtime_option.backend)
FDWARNING << runtime_option.backend
<< " is not support for YOLOv7End2EndTRT,"
<< "will fallback to Backend::TRT." << std::endl;
runtime_option.backend = Backend::TRT;
@@ -347,4 +347,4 @@ bool YOLOv7End2EndTRT::Predict(cv::Mat* im, DetectionResult* result,
} // namespace detection
} // namespace vision
} // namespace fastdeploy
} // namespace fastdeploy