Skip to content

Commit 193ebaa

Browse files
committed
Fix fuse gen
1 parent 2069672 commit 193ebaa

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

.pylintrc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
[BASIC]
2-
good-names=df,f,e,i,j,k
2+
good-names=df,f,e,i,j,k,pt
33

44
bad-functions=
55
apply,

superannotate/db/images.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -980,7 +980,7 @@ def create_fuse_image(
980980
if output_overlay:
981981
fi_pil_ovl = Image.fromarray(fi_ovl)
982982
draw_ovl = ImageDraw.Draw(fi_pil_ovl)
983-
for annotation in annotation_json:
983+
for annotation in annotation_json["instances"]:
984984
if "className" not in annotation:
985985
continue
986986
color = class_color_dict[annotation["className"]]
@@ -1080,7 +1080,7 @@ def create_fuse_image(
10801080
else:
10811081
annotation_mask = np.array(Image.open(annotation_path[1]))
10821082
# print(annotation_mask.shape, annotation_mask.dtype)
1083-
for annotation in annotation_json:
1083+
for annotation in annotation_json["instances"]:
10841084
if "className" not in annotation or "parts" not in annotation:
10851085
continue
10861086
color = class_color_dict[annotation["className"]]

0 commit comments

Comments
 (0)