Skip to content

Commit c043c8b

Browse files
authored
Update edge_drawing.py
1 parent 0da7255 commit c043c8b

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

modules/ximgproc/samples/edge_drawing.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -67,12 +67,14 @@ def main():
6767

6868
#Draw detected circles and ellipses
6969
if ellipses is not None: # Check if circles and ellipses have been found and only then iterate over these and add them to the image
70-
ellipses = np.uint16(np.around(ellipses))
7170
for i in range(len(ellipses)):
71+
center = (int(ellipses[i][0][0]), int(ellipses[i][0][1]))
72+
axes = (int(ellipses[i][0][2])+int(ellipses[i][0][3]),int(ellipses[i][0][2])+int(ellipses[i][0][4]))
73+
angle = ellipses[i][0][5]
7274
color = (0, 0, 255)
7375
if ellipses[i][0][2] == 0:
7476
color = (0, 255, 0)
75-
cv.ellipse(esrc, (ellipses[i][0][0], ellipses[i][0][1]), (ellipses[i][0][2]+ellipses[i][0][3],ellipses[i][0][2]+ellipses[i][0][4]),ellipses[i][0][5],0, 360, color, 2, cv.LINE_AA)
77+
cv.ellipse(esrc, center, axes, angle,0, 360, color, 2, cv.LINE_AA)
7678

7779
cv.imshow("detected ellipses", esrc)
7880
cv.waitKey(0)

0 commit comments

Comments
 (0)