Validate all backends for detection models and add demo code & docs (#94)

* Validate all backends for detection models and add demo code and doc

* Delete .README.md.swp
This commit is contained in:
Jason
2022-08-11 10:03:53 +08:00
committed by GitHub
parent d8f312b0a0
commit 3e01118d01
32 changed files with 1351 additions and 25 deletions
@@ -20,10 +20,10 @@ void BindVisualize(pybind11::module& m) {
.def(pybind11::init<>())
.def_static("vis_detection",
[](pybind11::array& im_data, vision::DetectionResult& result,
int line_size, float font_size) {
float score_threshold, int line_size, float font_size) {
auto im = PyArrayToCvMat(im_data);
auto vis_im = vision::Visualize::VisDetection(
im, result, line_size, font_size);
im, result, score_threshold, line_size, font_size);
FDTensor out;
vision::Mat(vis_im).ShareWithTensor(&out);
return TensorToPyArray(out);