lzhangzz 5e51739485
[Feature] Support DBNet, PANet and PSENet for SDK, with GPU aided post-processing (#526)
* add SDK support for PANet

* fix panet

* fix panet

* simplify panet

* add PSENet support

* fix-psenet

* add CUDA post-processing for DBNet

* fix dbnet

* fix dbnet

* add cpu support for PANet

* fix panet

* add CUDA support for PANet

* fix panet

* format

* add cpu impls for PSENet

* fix psenet

* add cuda impl for PSENet

* fix psenet

* add param parsing

* simplify impls

* simplify impls

* clean-up

* fix lint

* fix cuda-10 build

* fix cuda-10 build
2022-05-31 21:24:09 +08:00

42 lines
996 B
C++

// Copyright (c) OpenMMLab. All rights reserved.
#ifndef MMDEPLOY_CSRC_CODEBASE_MMOCR_CUDA_UTILS_H_
#define MMDEPLOY_CSRC_CODEBASE_MMOCR_CUDA_UTILS_H_
#include <cstdint>
#include "cuda_runtime.h"
namespace mmdeploy {
namespace mmocr {
namespace panet {
void ProcessMasks(const float* d_text_pred, const float* d_kernel_pred, float text_thr,
float kernel_thr, int n, uint8_t* d_text_mask, uint8_t* d_kernel_mask,
float* d_text_score, cudaStream_t stream);
void Transpose(const float* d_input, int h, int w, float* d_output, cudaStream_t stream);
} // namespace panet
namespace dbnet {
void Threshold(const float* d_score, int n, float thr, uint8_t* d_mask, cudaStream_t stream);
}
namespace psenet {
void ProcessMasks(const float* d_preds, int c, int n, float thr, uint8_t* d_masks, float* d_score,
cudaStream_t stream);
}
} // namespace mmocr
} // namespace mmdeploy
#endif // MMDEPLOY_CSRC_CODEBASE_MMOCR_CUDA_UTILS_H_