[Backend] cuda normalize and permute, cuda concat, optimized ppcls, ppdet & ppseg (#546)

* cuda normalize and permute, cuda concat

* add use cuda option for preprocessor

* ppyoloe use cuda normalize

* ppseg use cuda normalize

* add proclib cuda in processor base

* ppcls add use cuda preprocess api

* ppcls preprocessor set gpu id

* fix pybind

* refine ppcls preprocessing use gpu logic

* fdtensor device id is -1 by default

* refine assert message

Co-authored-by: heliqi <1101791222@qq.com>
This commit is contained in:
Wang Xinyu
2022-11-14 18:44:00 +08:00
committed by GitHub
parent 8dec2115d5
commit a36f5d3396
20 changed files with 204 additions and 26 deletions

View File

@@ -35,6 +35,13 @@ class PaddleClasPreprocessor:
"""
return self._preprocessor.run(input_ims)
def use_gpu(self, gpu_id=-1):
"""Use CUDA preprocessors
:param: gpu_id: GPU device id
"""
return self._preprocessor.use_gpu(gpu_id)
class PaddleClasPostprocessor:
def __init__(self, topk=1):