add polygons

This commit is contained in:
wangjingyeye 2022-07-04 08:49:34 +00:00
parent 8d9cfade49
commit 72f93cf977

View File

@ -236,7 +236,7 @@ class DBPostProcess(object):
self.dilation_kernel)
else:
mask = segmentation[batch_index]
if self.use_polygon:
if self.use_polygon is True:
boxes, scores = self.polygons_from_bitmap(pred[batch_index],
mask, src_w, src_h)
else: