apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-19 23:16:15 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Merge branch 'develop' of https://github.com/PaddlePaddle/FastDeploy into huawei

Browse Source
This commit is contained in:
yunyaoXYY
2022-12-30 10:25:11 +00:00
parent ac07fe9e2c 6310ddc8d6
commit 42da685d7e
27 changed files with 520 additions and 89 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."

13
python/fastdeploy/vision/segmentation/ppseg/__init__.py
View File

@@ -104,10 +104,17 @@ class PaddleSegPreprocessor:
"""
return self._preprocessor.run(input_ims)
def disable_normalize_and_permute(self):
"""To disable normalize and hwc2chw in preprocessing step.
def disable_normalize(self):
"""
return self._preprocessor.disable_normalize_and_permute()
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()
@property
def is_vertical_screen(self):