mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-28 18:51:58 +08:00
[Backend] Change valid_cpu_backends order (#726)
* add onnx_ort_runtime demo * rm in requirements * support batch eval * fixed MattingResults bug * move assignment for DetectionResult * integrated x2paddle * add model convert readme * update readme * re-lint * add processor api * Add MattingResult Free * change valid_cpu_backends order Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
2
fastdeploy/vision/classification/ppcls/model.cc
Normal file → Executable file
2
fastdeploy/vision/classification/ppcls/model.cc
Normal file → Executable file
@@ -24,7 +24,7 @@ PaddleClasModel::PaddleClasModel(const std::string& model_file,
|
|||||||
const RuntimeOption& custom_option,
|
const RuntimeOption& custom_option,
|
||||||
const ModelFormat& model_format) : preprocessor_(config_file) {
|
const ModelFormat& model_format) : preprocessor_(config_file) {
|
||||||
if (model_format == ModelFormat::PADDLE) {
|
if (model_format == ModelFormat::PADDLE) {
|
||||||
valid_cpu_backends = {Backend::ORT, Backend::OPENVINO, Backend::PDINFER,
|
valid_cpu_backends = {Backend::OPENVINO, Backend::PDINFER, Backend::ORT,
|
||||||
Backend::LITE};
|
Backend::LITE};
|
||||||
valid_gpu_backends = {Backend::ORT, Backend::PDINFER, Backend::TRT};
|
valid_gpu_backends = {Backend::ORT, Backend::PDINFER, Backend::TRT};
|
||||||
valid_timvx_backends = {Backend::LITE};
|
valid_timvx_backends = {Backend::LITE};
|
||||||
|
|||||||
@@ -28,7 +28,7 @@ RobustVideoMatting::RobustVideoMatting(const std::string& model_file,
|
|||||||
const RuntimeOption& custom_option,
|
const RuntimeOption& custom_option,
|
||||||
const ModelFormat& model_format) {
|
const ModelFormat& model_format) {
|
||||||
if (model_format == ModelFormat::ONNX) {
|
if (model_format == ModelFormat::ONNX) {
|
||||||
valid_cpu_backends = {Backend::OPENVINO, Backend::ORT};
|
valid_cpu_backends = {Backend::ORT, Backend::OPENVINO};
|
||||||
valid_gpu_backends = {Backend::ORT, Backend::TRT};
|
valid_gpu_backends = {Backend::ORT, Backend::TRT};
|
||||||
} else {
|
} else {
|
||||||
valid_cpu_backends = {Backend::PDINFER, Backend::ORT};
|
valid_cpu_backends = {Backend::PDINFER, Backend::ORT};
|
||||||
|
|||||||
Reference in New Issue
Block a user