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

[Model] update PPDet pybind and docs (#1550)

Browse Source 
update pybind and docs
This commit is contained in:
Zheng-Bicheng
2023-03-09 16:22:31 +08:00
committed by GitHub
parent 091e039991
commit 32f0d2e350
7 changed files with 15 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -74,14 +74,14 @@ class PaddleDetPostprocessor:
return self._postprocessor.run(runtime_results)
def apply_nms(self):
self.apply_nms()
self._postprocessor.apply_nms()
def set_nms_option(self, nms_option=None):
"""This function will enable decode and nms in postprocess step.
"""
if nms_option is None:
nms_option = NMSOption()
self._postprocessor.ApplyDecodeAndNMS(self, nms_option.nms_option)
self._postprocessor.set_nms_option(self, nms_option.nms_option)
class PPYOLOE(FastDeployModel):