mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
[CVCUDA] Utilize CV-CUDA batch processing function (#1223)
* norm and permute batch processing * move cache to mat, batch processors * get batched tensor logic, resize on cpu logic * fix cpu compile error * remove vector mat api * nits * add comments * nits * fix batch size * move initial resize on cpu option to use_cuda api * fix pybind * processor manager pybind * rename mat and matbatch * move initial resize on cpu to ppcls preprocessor --------- Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
@@ -16,6 +16,7 @@
|
||||
|
||||
namespace fastdeploy {
|
||||
|
||||
void BindProcessorManager(pybind11::module& m);
|
||||
void BindDetection(pybind11::module& m);
|
||||
void BindClassification(pybind11::module& m);
|
||||
void BindSegmentation(pybind11::module& m);
|
||||
@@ -204,6 +205,7 @@ void BindVision(pybind11::module& m) {
|
||||
m.def("disable_flycv", &vision::DisableFlyCV,
|
||||
"Disable image preprocessing by FlyCV, change to use OpenCV.");
|
||||
|
||||
BindProcessorManager(m);
|
||||
BindDetection(m);
|
||||
BindClassification(m);
|
||||
BindSegmentation(m);
|
||||
|
||||
Reference in New Issue
Block a user