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

[Model] Fix nms problem (#2230)

Browse Source 
fix nms problem
This commit is contained in:
Jason
2023-10-10 01:24:33 -07:00
committed by GitHub
parent aab27b3db5
commit d910d6116c
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
fastdeploy/vision/detection/ppdet/ppdet_pybind.cc
View File

@@ -87,6 +87,12 @@ void BindPPDet(pybind11::module& m) {
[](vision::detection::PaddleDetPostprocessor& self) { [](vision::detection::PaddleDetPostprocessor& self) {
self.ApplyNMS(); self.ApplyNMS();
}) })
.def("apply_nms",
[](vision::detection::PaddleDetPostprocessor& self,
vision::detection::NMSOption& option) {
self.ApplyNMS();
self.SetNMSOption(option);
})
.def("run", [](vision::detection::PaddleDetPostprocessor& self, .def("run", [](vision::detection::PaddleDetPostprocessor& self,
std::vector<pybind11::array>& input_array) { std::vector<pybind11::array>& input_array) {
std::vector<vision::DetectionResult> results; std::vector<vision::DetectionResult> results;

1
python/fastdeploy/vision/detection/ppdet/__init__.py
View File

@@ -494,7 +494,6 @@ class PaddleYOLOv5(PPYOLOE):
super(PPYOLOE, self).__init__(runtime_option) super(PPYOLOE, self).__init__(runtime_option)
assert model_format == ModelFormat.PADDLE, "PaddleYOLOv5 model only support model format of ModelFormat.Paddle now."
self._model = C.vision.detection.PaddleYOLOv5( self._model = C.vision.detection.PaddleYOLOv5(
model_file, params_file, config_file, self._runtime_option, model_file, params_file, config_file, self._runtime_option,
model_format) model_format)