Commit bf18172d authored by 赵小蒙's avatar 赵小蒙
Browse files

update: model parse support paddle output

parent e92de758
...@@ -37,7 +37,14 @@ class MagicModel: ...@@ -37,7 +37,14 @@ class MagicModel:
) )
layout_dets = model_page_info["layout_dets"] layout_dets = model_page_info["layout_dets"]
for layout_det in layout_dets: for layout_det in layout_dets:
x0, y0, _, _, x1, y1, _, _ = layout_det["poly"]
if layout_det.get("bbox") is not None:
# 兼容直接输出bbox的模型数据,如paddle
x0, y0, x1, y1 = layout_det["bbox"]
else:
# 兼容直接输出poly的模型数据,如xxx
x0, y0, _, _, x1, y1, _, _ = layout_det["poly"]
bbox = [ bbox = [
int(x0 / horizontal_scale_ratio), int(x0 / horizontal_scale_ratio),
int(y0 / vertical_scale_ratio), int(y0 / vertical_scale_ratio),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment