mmdeploy/csrc/apis/python/common.h

// Copyright (c) OpenMMLab. All rights reserved.

#ifndef MMDEPLOY_CSRC_APIS_PYTHON_COMMON_H_
#define MMDEPLOY_CSRC_APIS_PYTHON_COMMON_H_

#include <stdexcept>

#include "apis/c/common.h"
#include "pybind11/numpy.h"
#include "pybind11/pybind11.h"
#include "pybind11/stl.h"

namespace py = pybind11;

using PyImage = py::array_t<uint8_t, py::array::c_style | py::array::forcecast>;

namespace mmdeploy {

std::map<std::string, void (*)(py::module &)> &gPythonBindings();

mm_mat_t GetMat(const PyImage &img);

class Value;

py::object ConvertToPyObject(const Value &value);

Value ConvertToValue(const py::object &obj);

}  // namespace mmdeploy

#endif  // MMDEPLOY_CSRC_APIS_PYTHON_COMMON_H_