mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
5e51739485
* 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
39 lines
1011 B
C++
39 lines
1011 B
C++
// Copyright (c) OpenMMLab. All rights reserved.
|
|
|
|
#ifndef MMDEPLOY_CSRC_CODEBASE_MMOCR_PSENET_H_
|
|
#define MMDEPLOY_CSRC_CODEBASE_MMOCR_PSENET_H_
|
|
|
|
#include "codebase/mmocr/mmocr.h"
|
|
#include "core/device.h"
|
|
#include "core/registry.h"
|
|
#include "core/tensor.h"
|
|
#include "opencv2/core.hpp"
|
|
|
|
namespace mmdeploy {
|
|
namespace mmocr {
|
|
|
|
class PseHeadImpl {
|
|
public:
|
|
virtual ~PseHeadImpl() = default;
|
|
|
|
virtual void Init(const Stream& stream) { stream_ = stream; }
|
|
|
|
virtual Result<void> Process(Tensor preds, //
|
|
float min_kernel_confidence, //
|
|
cv::Mat_<float>& score, //
|
|
cv::Mat_<uint8_t>& masks, //
|
|
cv::Mat_<int>& label, //
|
|
int& region_num) = 0;
|
|
|
|
protected:
|
|
Stream stream_;
|
|
};
|
|
|
|
} // namespace mmocr
|
|
|
|
MMDEPLOY_DECLARE_REGISTRY(mmocr::PseHeadImpl);
|
|
|
|
} // namespace mmdeploy
|
|
|
|
#endif // MMDEPLOY_CSRC_CODEBASE_MMOCR_PSENET_H_
|