mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
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_
|