apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 00:57:33 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add uie python example and doc (#221)

Browse Source 
* add fastdeploy.text.UIEModel

* Add uie python example

* Add one schema for cpp demo

* Add ConvertUIEResultToDict for pretty the uie result in python

* remove default args for SchemaNode

* Add uie example args

* Add uie python api desc

* Add infer.py usage

* truncate some example output

* Add uie schema usage

* Add uie result md

* Add uie c++ api doc
This commit is contained in:
Jack Zhou
2022-09-15 06:06:40 +08:00
committed by GitHub
parent fb0a428c3c
commit 14ba9ce6c2
11 changed files with 760 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
fastdeploy/text/text_pybind.cc
View File

@@ -14,18 +14,47 @@
#include "fastdeploy/pybind/main.h"
namespace py = pybind11;
using namespace py::literals;
namespace fastdeploy {
void BindUIE(pybind11::module& m);
void BindUIE(py::module& m);
void BindText(pybind11::module& m) {
pybind11::class_<text::UIEResult>(m, "UIEResult")
.def(pybind11::init())
py::dict ConvertUIEResultToDict(const text::UIEResult& self) {
py::dict d;
d["start"] = self.start_;
d["end"] = self.end_;
d["probability"] = self.probability_;
d["text"] = self.text_;
if (!self.relation_.empty()) {
d["relation"] = py::dict();
for (auto iter = self.relation_.begin(); iter != self.relation_.end();
++iter) {
py::list l;
for (auto result_iter = iter->second.begin();
result_iter != iter->second.end(); ++result_iter) {
l.append(ConvertUIEResultToDict(*result_iter));
}
d["relation"][iter->first.c_str()] = l;
}
}
return d;
}
void BindText(py::module& m) {
py::class_<text::UIEResult>(m, "UIEResult", py::dynamic_attr())
.def(py::init())
.def_readwrite("start", &text::UIEResult::start_)
.def_readwrite("end", &text::UIEResult::end_)
.def_readwrite("probability_", &text::UIEResult::probability_)
.def_readwrite("probability", &text::UIEResult::probability_)
.def_readwrite("text", &text::UIEResult::text_)
.def_readwrite("relation", &text::UIEResult::relation_)
.def("get_dict",
[](const text::UIEResult& self) {
return ConvertUIEResultToDict(self);
})
.def("__repr__", &text::UIEResult::Str)
.def("__str__", &text::UIEResult::Str);
BindUIE(m);