mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
bb655af629
* add python API for detector * integrate detection * add python segmentor * add segmentation support * add classifier, text-detector, text-recognizer and restorer * integrate classifier * integrate textdet, textrecog and restorer * simplify * add inst-seg * fix inst-seg * integrate inst-seg * Moidfy _build_wrapper * better pipeline substitution * use registry for backend model creation * build Python module according to C API targets * minor fix * move sdk data pipeline to backend_config * remove debugging lines * add docstring for SDKEnd2EndModel * fix type hint * fix lint * fix lint * insert build/lib to sys.path Co-authored-by: SingleZombie <singlezombie@163.com>
42 lines
1.0 KiB
C++
42 lines
1.0 KiB
C++
// Copyright (c) OpenMMLab. All rights reserved.
|
|
|
|
#include "apis/python/common.h"
|
|
|
|
namespace mmdeploy {
|
|
|
|
std::map<std::string, void (*)(py::module &)> &gPythonBindings() {
|
|
static std::map<std::string, void (*)(py::module &)> v;
|
|
return v;
|
|
}
|
|
|
|
mm_mat_t GetMat(const PyImage &img) {
|
|
auto info = img.request();
|
|
if (info.ndim != 3) {
|
|
fprintf(stderr, "info.ndim = %d\n", (int)info.ndim);
|
|
throw std::runtime_error("continuous uint8 HWC array expected");
|
|
}
|
|
auto channels = (int)info.shape[2];
|
|
mm_mat_t mat{};
|
|
if (channels == 1) {
|
|
mat.format = MM_GRAYSCALE;
|
|
} else if (channels == 3) {
|
|
mat.format = MM_BGR;
|
|
} else {
|
|
throw std::runtime_error("images of 1 or 3 channels are supported");
|
|
}
|
|
mat.height = (int)info.shape[0];
|
|
mat.width = (int)info.shape[1];
|
|
mat.channel = channels;
|
|
mat.type = MM_INT8;
|
|
mat.data = (uint8_t *)info.ptr;
|
|
return mat;
|
|
}
|
|
|
|
} // namespace mmdeploy
|
|
|
|
PYBIND11_MODULE(mmdeploy_python, m) {
|
|
for (const auto &[_, f] : mmdeploy::gPythonBindings()) {
|
|
f(m);
|
|
}
|
|
}
|