提交 72f93cf9 编写于 作者: W wangjingyeye

add polygons

上级 8d9cfade
...@@ -236,7 +236,7 @@ class DBPostProcess(object): ...@@ -236,7 +236,7 @@ class DBPostProcess(object):
self.dilation_kernel) self.dilation_kernel)
else: else:
mask = segmentation[batch_index] mask = segmentation[batch_index]
if self.use_polygon: if self.use_polygon is True:
boxes, scores = self.polygons_from_bitmap(pred[batch_index], boxes, scores = self.polygons_from_bitmap(pred[batch_index],
mask, src_w, src_h) mask, src_w, src_h)
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册