mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-26 10:00:33 +08:00
Revert "[Backend] Add KunlunXin XPU deploy support" (#893)
Revert "[Backend] Add KunlunXin XPU deploy support (#747)"
This reverts commit 5be839b322.
This commit is contained in:
@@ -72,11 +72,10 @@ class FASTDEPLOY_DECL RKYOLOPostprocessor {
|
||||
}
|
||||
|
||||
// Set Anchor
|
||||
void SetAnchor(std::vector<int> anchors, int anchor_per_branch) {
|
||||
void SetAnchor(std::vector<int> anchors,int anchor_per_branch){
|
||||
anchors_ = anchors;
|
||||
anchor_per_branch_ = anchor_per_branch;
|
||||
}
|
||||
|
||||
};
|
||||
private:
|
||||
std::vector<int> anchors_ = {10, 13, 16, 30, 33, 23, 30, 61, 62,
|
||||
45, 59, 119, 116, 90, 156, 198, 373, 326};
|
||||
|
||||
@@ -31,4 +31,4 @@ int NMS(int valid_count, std::vector<float>& output_locations,
|
||||
|
||||
} // namespace detection
|
||||
} // namespace vision
|
||||
} // namespace fastdeploy
|
||||
} // namespace fastdeploy
|
||||
@@ -27,7 +27,6 @@ YOLOv5::YOLOv5(const std::string& model_file, const std::string& params_file,
|
||||
} else {
|
||||
valid_cpu_backends = {Backend::PDINFER, Backend::ORT, Backend::LITE};
|
||||
valid_gpu_backends = {Backend::PDINFER, Backend::ORT, Backend::TRT};
|
||||
valid_xpu_backends = {Backend::LITE};
|
||||
valid_timvx_backends = {Backend::LITE};
|
||||
}
|
||||
runtime_option = custom_option;
|
||||
|
||||
@@ -185,7 +185,7 @@ class FASTDEPLOY_DECL PaddleYOLOv5 : public PPDetBase {
|
||||
const ModelFormat& model_format = ModelFormat::PADDLE)
|
||||
: PPDetBase(model_file, params_file, config_file, custom_option,
|
||||
model_format) {
|
||||
valid_cpu_backends = {Backend::ORT, Backend::PDINFER};
|
||||
valid_cpu_backends = {Backend::ORT,Backend::PDINFER};
|
||||
valid_gpu_backends = {Backend::ORT, Backend::PDINFER, Backend::TRT};
|
||||
initialized = Initialize();
|
||||
}
|
||||
@@ -201,7 +201,7 @@ class FASTDEPLOY_DECL PaddleYOLOv6 : public PPDetBase {
|
||||
const ModelFormat& model_format = ModelFormat::PADDLE)
|
||||
: PPDetBase(model_file, params_file, config_file, custom_option,
|
||||
model_format) {
|
||||
valid_cpu_backends = {Backend::OPENVINO, Backend::ORT, Backend::PDINFER};
|
||||
valid_cpu_backends = {Backend::OPENVINO, Backend::ORT,Backend::PDINFER};
|
||||
valid_gpu_backends = {Backend::ORT, Backend::PDINFER, Backend::TRT};
|
||||
initialized = Initialize();
|
||||
}
|
||||
@@ -217,7 +217,7 @@ class FASTDEPLOY_DECL PaddleYOLOv7 : public PPDetBase {
|
||||
const ModelFormat& model_format = ModelFormat::PADDLE)
|
||||
: PPDetBase(model_file, params_file, config_file, custom_option,
|
||||
model_format) {
|
||||
valid_cpu_backends = {Backend::ORT, Backend::PDINFER};
|
||||
valid_cpu_backends = {Backend::ORT,Backend::PDINFER};
|
||||
valid_gpu_backends = {Backend::ORT, Backend::PDINFER, Backend::TRT};
|
||||
initialized = Initialize();
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user