mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
Add YOLOv5Face model support (#38)
* update .gitignore * Added checking for cmake include dir * fixed missing trt_backend option bug when init from trt * remove un-need data layout and add pre-check for dtype * changed RGB2BRG to BGR2RGB in ppcls model * add model_zoo yolov6 c++/python demo * fixed CMakeLists.txt typos * update yolov6 cpp/README.md * add yolox c++/pybind and model_zoo demo * move some helpers to private * fixed CMakeLists.txt typos * add normalize with alpha and beta * add version notes for yolov5/yolov6/yolox * add copyright to yolov5.cc * revert normalize * fixed some bugs in yolox * Add YOLOv5Face Model support * fixed examples/vision typos * fixed runtime_option print func bugs
This commit is contained in:
@@ -25,6 +25,14 @@ void BindVisualize(pybind11::module& m) {
|
||||
vision::Visualize::VisDetection(&im, result, line_size,
|
||||
font_size);
|
||||
})
|
||||
.def_static(
|
||||
"vis_face_detection",
|
||||
[](pybind11::array& im_data, vision::FaceDetectionResult& result,
|
||||
int line_size, float font_size) {
|
||||
auto im = PyArrayToCvMat(im_data);
|
||||
vision::Visualize::VisFaceDetection(&im, result, line_size,
|
||||
font_size);
|
||||
})
|
||||
.def_static("vis_segmentation", [](pybind11::array& im_data,
|
||||
vision::SegmentationResult& result,
|
||||
pybind11::array& vis_im_data,
|
||||
|
||||
Reference in New Issue
Block a user