Skip to content

Commit ae4f401

Browse files
committed
sagemaker changed
1 parent 591845b commit ae4f401

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

superannotate/input_converters/converters/sagemaker_converters/sagemaker_to_sa_pixel.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ def sagemaker_instance_segmentation_to_sa_pixel(data_path, main_key):
9898
'attributes': [],
9999
'locked': False,
100100
'visible': True,
101-
'groupId': 0
101+
'parts': []
102102
}
103103
for contour in contours:
104104
bitmask = np.zeros((H, W))
@@ -113,7 +113,7 @@ def sagemaker_instance_segmentation_to_sa_pixel(data_path, main_key):
113113
cv2.fillPoly(bitmask, [pts], 1)
114114
sa_mask[bitmask == 1
115115
] = list(hex_to_rgb(blue_colors[idx]))[::-1] + [255]
116-
sa_obj['parts'] = {'color': blue_colors[idx]}
116+
sa_obj['parts'].append({'color': blue_colors[idx]})
117117
idx += 1
118118
sa_loader.append(sa_obj.copy())
119119
sa_jsons[sa_name] = sa_loader

0 commit comments

Comments
 (0)