[Model] Fix nms problem (#2230)

fix nms problem
PaddlePaddle · Oct 10, 2023 · d910d61 · d910d61
1 parent aab27b3
commit d910d61
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/fastdeploy/vision/detection/ppdet/ppdet_pybind.cc b/fastdeploy/vision/detection/ppdet/ppdet_pybind.cc
@@ -87,6 +87,12 @@ void BindPPDet(pybind11::module& m) {
            [](vision::detection::PaddleDetPostprocessor& self) {
              self.ApplyNMS();
            })
+      .def("apply_nms",
+           [](vision::detection::PaddleDetPostprocessor& self,
+              vision::detection::NMSOption& option) {
+             self.ApplyNMS();
+             self.SetNMSOption(option);
+           })
       .def("run", [](vision::detection::PaddleDetPostprocessor& self,
                      std::vector<pybind11::array>& input_array) {
         std::vector<vision::DetectionResult> results;

diff --git a/python/fastdeploy/vision/detection/ppdet/__init__.py b/python/fastdeploy/vision/detection/ppdet/__init__.py
@@ -494,7 +494,6 @@ def __init__(self,
 
         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(
             model_file, params_file, config_file, self._runtime_option,
             model_format)