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

[Model] Support PP-StructureV2-Layout model (#1867)

Browse Source 
* [Model] init pp-structurev2-layout code

* [Model] init pp-structurev2-layout code

* [Model] init pp-structurev2-layout code

* [Model] add structurev2_layout_preprocessor

* [PP-StructureV2] add postprocessor and layout detector class

* [PP-StructureV2] add postprocessor and layout detector class

* [PP-StructureV2] add postprocessor and layout detector class

* [PP-StructureV2] add postprocessor and layout detector class

* [PP-StructureV2] add postprocessor and layout detector class

* [pybind] add pp-structurev2-layout model pybind

* [pybind] add pp-structurev2-layout model pybind

* [Bug Fix] fixed code style

* [examples] add pp-structurev2-layout c++ examples

* [PP-StructureV2] add python example and docs

* [benchmark] add pp-structurev2-layout benchmark support
This commit is contained in:
DefTruth
2023-05-05 13:05:58 +08:00
committed by GitHub
parent 2c5fd91a7f
commit 6d0261e9e4
26 changed files with 1255 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
python/fastdeploy/vision/visualize/__init__.py
View File

@@ -23,7 +23,9 @@ def vis_detection(im_data,
labels=[],
score_threshold=0.0,
line_size=1,
font_size=0.5):
font_size=0.5,
font_color=[255, 255, 255],
font_thickness=1):
"""Show the visualized results for detection models
:param im_data: (numpy.ndarray)The input image data, 3-D array with layout HWC, BGR format
@@ -32,10 +34,13 @@ def vis_detection(im_data,
:param score_threshold: (float) score_threshold threshold for result scores, the bounding box will not be shown if the score is less than score_threshold
:param line_size: (float) line_size line size for bounding boxes
:param font_size: (float) font_size font size for text
:param font_color: (list of int) font_color for text
:param font_thickness: (int) font_thickness for text
:return: (numpy.ndarray) image with visualized results
"""
return C.vision.vis_detection(im_data, det_result, labels, score_threshold,
line_size, font_size)
line_size, font_size, font_color,
font_thickness)
def vis_perception(im_data,