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

Fix bug for part of detection model (#243)

Browse Source 
* fix error for part of detection model

* fix error for part of detection model

* add patch paddle inference
This commit is contained in:
Jason
2022-09-16 16:31:34 +08:00
committed by GitHub
parent 4d2fbcb030
commit 5046574b95
10 changed files with 61 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/backends/ort/utils.cc
View File

@@ -26,6 +26,10 @@ ONNXTensorElementDataType GetOrtDtype(const FDDataType& fd_dtype) {
return ONNX_TENSOR_ELEMENT_DATA_TYPE_INT32;
} else if (fd_dtype == FDDataType::INT64) {
return ONNX_TENSOR_ELEMENT_DATA_TYPE_INT64;
} else if (fd_dtype == FDDataType::UINT8) {
return ONNX_TENSOR_ELEMENT_DATA_TYPE_UINT8;
} else if (fd_dtype == FDDataType::INT8) {
return ONNX_TENSOR_ELEMENT_DATA_TYPE_INT8;
}
FDERROR << "Unrecognized fastdeply data type:" << Str(fd_dtype) << "."
<< std::endl;