apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-11-03 11:02:01 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[Model] Add FSANet model (#448)

Browse Source 
* add yolov5cls

* fixed bugs

* fixed bugs

* fixed preprocess bug

* add yolov5cls readme

* deal with comments

* Add YOLOv5Cls Note

* add yolov5cls test

* add rvm support

* support rvm model

* add rvm demo

* fixed bugs

* add rvm readme

* add TRT support

* add trt support

* add rvm test

* add EXPORT.md

* rename export.md

* rm poros doxyen

* deal with comments

* deal with comments

* add rvm video_mode note

* add fsanet

* fixed bug

* update readme

* fixed for ci

* deal with comments

* deal with comments

* deal with comments

Co-authored-by: Jason <jiangjiajun@baidu.com>
Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
This commit is contained in:
WJJ1995
2022-11-04 11:00:35 +08:00
committed by GitHub
parent ce828ecb38
commit 7150e6405c
31 changed files with 922 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
fastdeploy/vision/visualize/visualize_pybind.cc Normal file → Executable file
View File

@@ -102,6 +102,16 @@ void BindVisualize(pybind11::module& m) {
FDTensor out;
vision::Mat(vis_im).ShareWithTensor(&out);
return TensorToPyArray(out);
})
.def("vis_headpose",
[](pybind11::array& im_data, vision::HeadPoseResult& result,
int size, int line_size) {
auto im = PyArrayToCvMat(im_data);
auto vis_im =
vision::VisHeadPose(im, result, size, line_size);
FDTensor out;
vision::Mat(vis_im).ShareWithTensor(&out);
return TensorToPyArray(out);
});
pybind11::class_<vision::Visualize>(m, "Visualize")