nodes/zentrocdot-ComfyUI_Circle_Detection
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
Dr. Peter Netz
2025-02-02 12:15:22 +01:00
committed by GitHub
parent 187cd69c1b
commit 95f700ee9c
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/nodes.py
+8 -3
View File
@@ -32,6 +32,8 @@ class CircleDetection:
def draw_circles(self, img, detected_circles):
print("*** DRAW CIRCLES ***")
print(type(img))
newImg = img.copy()
print(type(img))
COLOR_TUPLE = (255, 0, 255)
THICKNESS = 5
# Declare local variables.
@@ -46,13 +48,16 @@ class CircleDetection:
a, b, r = pnt[0], pnt[1], pnt[2]
print(a,b,r)
# Draw the circumference of the circle.
cv2.circle(img, (a, b), r, COLOR_TUPLE, THICKNESS)
#cv2.circle(img, (a, b), r, COLOR_TUPLE, THICKNESS)
cv2.circle(newImg, (a, b), r, COLOR_TUPLE, THICKNESS)
# Draw a small circle of radius 1 to show the center.
cv2.circle(img, (a, b), 1, COLOR_TUPLE, 3)
#cv2.circle(img, (a, b), 1, COLOR_TUPLE, 3)
cv2.circle(newImg, (a, b), 1, COLOR_TUPLE, 3)
# Print dimensions and radius.
#print("x:", a, "y", b, "r:", r)
# Return image, co-ordinates and radius.
return img, (a, b, r)
#return img, (a, b, r)
return newImg, (a, b, r
def pre_img(self, img):
'''Preprocess image.'''