提交 ecf6362a 编写于 作者: S Shannon 提交者: Kentaro Wada

Fix segmentation generation for rectangles

上级 1d5f125d
......@@ -126,6 +126,11 @@ def main():
masks[instance] = mask
points = np.asarray(points).flatten().tolist()
if shape['shape_type'] == 'rectangle':
x_1, y_1, x_2, y_2 = points
points = [x_1, y_1, x_2, y_1, x_2, y_2, x_1, y_2]
segmentations[instance].append(points)
segmentations = dict(segmentations)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册