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

[CVCUDA] PP-OCR and PPDet preprocessor support use_cuda python API(#1734)

Browse Source 
add cvcuda support in ppocr and ppdet
This commit is contained in:
guxukai
2023-03-29 20:57:36 +08:00
committed by GitHub
parent 3ef48f7746
commit 38f056d1c1
2 changed files with 34 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -17,36 +17,28 @@ from typing import Union, List
import logging
from .... import FastDeployModel, ModelFormat
from .... import c_lib_wrap as C
from ...common import ProcessorManager
class PaddleDetPreprocessor:
class PaddleDetPreprocessor(ProcessorManager):
def __init__(self, config_file):
"""Create a preprocessor for PaddleDetection Model from configuration file
:param config_file: (str)Path of configuration file, e.g ppyoloe/infer_cfg.yml
"""
self._preprocessor = C.vision.detection.PaddleDetPreprocessor(
config_file)
def run(self, input_ims):
"""Preprocess input images for PaddleDetection Model
:param: input_ims: (list of numpy.ndarray)The input image
:return: list of FDTensor, include image, scale_factor, im_shape
"""
return self._preprocessor.run(input_ims)
self._manager = C.vision.detection.PaddleDetPreprocessor(config_file)
def disable_normalize(self):
"""
This function will disable normalize in preprocessing step.
"""
self._preprocessor.disable_normalize()
self._manager.disable_normalize()
def disable_permute(self):
"""
This function will disable hwc2chw in preprocessing step.
"""
self._preprocessor.disable_permute()
self._manager.disable_permute()
class NMSOption: