mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-04 16:22:57 +08:00
[Hackthon_4th 177] Support PP-YOLOE-R with BM1684 (#1809)
* first draft * add robx iou * add benchmark for ppyoloe_r * remove trash code * fix bugs * add pybind nms rotated option * add missing head file * fix bug * fix bug2 * fix shape bug --------- Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
This commit is contained in:
@@ -158,5 +158,14 @@ static bool UpdateModelResourceName(
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
if (config_info["backend"] == "sophgo") {
|
||||
*model_format = fastdeploy::ModelFormat::SOPHGO;
|
||||
if (!GetModelResoucesNameFromDir(FLAGS_model, model_name, "bmodel")) {
|
||||
std::cout << "Can not find sophgo model resources." << std::endl;
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
Reference in New Issue
Block a user