718c41634f
1.项目后端整体迁移至PaddleOCR-NCNN算法，已通过基本的兼容性测试 2.工程改为使用CMake组织，后续为了更好地兼容第三方库，不再提供QMake工程 3.重整权利声明文件，重整代码工程，确保最小化侵权风险 Log: 切换后端至PaddleOCR-NCNN，切换工程为CMake Change-Id: I4d5d2c5d37505a4a24b389b1a4c5d12f17bfa38c
161 lines
4.2 KiB
C++
161 lines
4.2 KiB
C++
/* Tencent is pleased to support the open source community by making ncnn available.
|
|
*
|
|
* Copyright (C) 2020 THL A29 Limited, a Tencent company. All rights reserved.
|
|
*
|
|
* Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
|
|
* in compliance with the License. You may obtain a copy of the License at
|
|
*
|
|
* https://opensource.org/licenses/BSD-3-Clause
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software distributed
|
|
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
|
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
|
* specific language governing permissions and limitations under the License.
|
|
*/
|
|
|
|
#ifndef PYBIND11_NCNN_LAYER_H
|
|
#define PYBIND11_NCNN_LAYER_H
|
|
|
|
#include <layer.h>
|
|
#include "pybind11_bind.h"
|
|
|
|
class PyLayer : public ncnn::Layer
|
|
{
|
|
public:
|
|
virtual int load_param(const ncnn::ParamDict& pd)
|
|
{
|
|
PYBIND11_OVERRIDE(
|
|
int,
|
|
ncnn::Layer,
|
|
load_param,
|
|
pd);
|
|
}
|
|
|
|
virtual int load_model(const ncnn::ModelBin& mb)
|
|
{
|
|
PYBIND11_OVERRIDE(
|
|
int,
|
|
ncnn::Layer,
|
|
load_model,
|
|
mb);
|
|
}
|
|
|
|
virtual int create_pipeline(const ncnn::Option& opt)
|
|
{
|
|
PYBIND11_OVERRIDE(
|
|
int,
|
|
ncnn::Layer,
|
|
create_pipeline,
|
|
opt);
|
|
}
|
|
|
|
virtual int destroy_pipeline(const ncnn::Option& opt)
|
|
{
|
|
PYBIND11_OVERRIDE(
|
|
int,
|
|
ncnn::Layer,
|
|
destroy_pipeline,
|
|
opt);
|
|
}
|
|
|
|
public:
|
|
virtual int forward(const std::vector<ncnn::Mat>& bottom_blobs, std::vector<ncnn::Mat>& top_blobs, const ncnn::Option& opt) const
|
|
{
|
|
PYBIND11_OVERRIDE_REFERENCE(
|
|
int,
|
|
ncnn::Layer,
|
|
forward,
|
|
bottom_blobs,
|
|
top_blobs,
|
|
opt);
|
|
}
|
|
virtual int forward(const ncnn::Mat& bottom_blob, ncnn::Mat& top_blob, const ncnn::Option& opt) const
|
|
{
|
|
PYBIND11_OVERRIDE_REFERENCE(
|
|
int,
|
|
ncnn::Layer,
|
|
forward,
|
|
bottom_blob,
|
|
top_blob,
|
|
opt);
|
|
}
|
|
|
|
virtual int forward_inplace(std::vector<ncnn::Mat>& bottom_top_blobs, const ncnn::Option& opt) const
|
|
{
|
|
PYBIND11_OVERRIDE_REFERENCE(
|
|
int,
|
|
ncnn::Layer,
|
|
forward_inplace,
|
|
bottom_top_blobs,
|
|
opt);
|
|
}
|
|
virtual int forward_inplace(ncnn::Mat& bottom_top_blob, const ncnn::Option& opt) const
|
|
{
|
|
PYBIND11_OVERRIDE_REFERENCE(
|
|
int,
|
|
ncnn::Layer,
|
|
forward_inplace,
|
|
bottom_top_blob,
|
|
opt);
|
|
}
|
|
|
|
#if NCNN_VULKAN
|
|
public:
|
|
virtual int upload_model(ncnn::VkTransfer& cmd, const ncnn::Option& opt)
|
|
{
|
|
PYBIND11_OVERRIDE_REFERENCE(
|
|
int,
|
|
ncnn::Layer,
|
|
upload_model,
|
|
cmd,
|
|
opt);
|
|
}
|
|
|
|
public:
|
|
virtual int forward(const std::vector<ncnn::VkMat>& bottom_blobs, std::vector<ncnn::VkMat>& top_blobs, ncnn::VkCompute& cmd, const ncnn::Option& opt) const
|
|
{
|
|
PYBIND11_OVERRIDE_REFERENCE(
|
|
int,
|
|
ncnn::Layer,
|
|
forward,
|
|
bottom_blobs,
|
|
top_blobs,
|
|
cmd,
|
|
opt);
|
|
}
|
|
virtual int forward(const ncnn::VkMat& bottom_blob, ncnn::VkMat& top_blob, ncnn::VkCompute& cmd, const ncnn::Option& opt) const
|
|
{
|
|
PYBIND11_OVERRIDE_REFERENCE(
|
|
int,
|
|
ncnn::Layer,
|
|
forward,
|
|
bottom_blob,
|
|
top_blob,
|
|
cmd,
|
|
opt);
|
|
}
|
|
|
|
virtual int forward_inplace(std::vector<ncnn::VkMat>& bottom_top_blobs, ncnn::VkCompute& cmd, const ncnn::Option& opt) const
|
|
{
|
|
PYBIND11_OVERRIDE_REFERENCE(
|
|
int,
|
|
ncnn::Layer,
|
|
forward_inplace,
|
|
bottom_top_blobs,
|
|
cmd,
|
|
opt);
|
|
}
|
|
virtual int forward_inplace(ncnn::VkMat& bottom_top_blob, ncnn::VkCompute& cmd, const ncnn::Option& opt) const
|
|
{
|
|
PYBIND11_OVERRIDE_REFERENCE(
|
|
int,
|
|
ncnn::Layer,
|
|
forward_inplace,
|
|
bottom_top_blob,
|
|
cmd,
|
|
opt);
|
|
}
|
|
#endif // NCNN_VULKAN
|
|
};
|
|
|
|
#endif |