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

[Model] Update PPDet Preprocess (#1006)

Browse Source 
* 更新导航文档

* 更新导航文档

* 更新导航文档

* 更新导航文档

* 更新PPDet PreProcess

* 更新PPDet PreProcess

* 更新PPDet pybind and python

* 更新

* 更新ppdet
This commit is contained in:
Zheng-Bicheng
2022-12-29 21:15:23 +08:00
committed by GitHub
parent dd5759bd99
commit d275c3ba02
5 changed files with 74 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -36,6 +36,18 @@ class PaddleDetPreprocessor:
"""
return self._preprocessor.run(input_ims)
def disable_normalize(self):
"""
This function will disable normalize in preprocessing step.
"""
self._preprocessor.disable_normalize()
def disable_permute(self):
"""
This function will disable hwc2chw in preprocessing step.
"""
self._preprocessor.disable_permute()
class PaddleDetPostprocessor:
def __init__(self):
@@ -500,4 +512,4 @@ class RTMDet(PPYOLOE):
self._model = C.vision.detection.RTMDet(
model_file, params_file, config_file, self._runtime_option,
model_format)
assert self.initialized, "RTMDet model initialize failed."
assert self.initialized, "RTMDet model initialize failed."