Skip to content

Commit 891b4d0

Browse files
committed
fix(paint_polygon_evaluation): Set painted_output shape properly.
Was only working for square shapes, as the order of dimensions was wrong.
1 parent ea0b954 commit 891b4d0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/osm_ai_helper/utils/polygons.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ def paint_polygon_evaluation(
119119
false_alamars: List[Polygon],
120120
missed: List[Polygon],
121121
):
122-
painted_output = np.zeros((*image.size, 3))
122+
painted_output = np.zeros((image.size[1], image.size[1], 3))
123123
for p in found:
124124
paint_polygon(p, painted_output, (0, 255, 0))
125125
for p in false_alamars:

0 commit comments

Comments
 (0)