[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:
thunder95
2023-04-21 10:48:05 +08:00
committed by GitHub
parent f3d44785c4
commit 51be3fea78
31 changed files with 1389 additions and 6 deletions

View File

@@ -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;
}