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 11:24:53 +01:00
committed by GitHub
parent 4f9b4267a9
commit ccd7b2fd16
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/nodes.py
+4 -4
View File
@@ -105,7 +105,7 @@ class CircleDetection:
def post_img(self, img, detected_circles):
'''Postprocess image.'''
# Draw circles.
img, (a, b, r) = draw_circles(img, detected_circles)
img, (a, b, r) = self.draw_circles(img, detected_circles)
# Print dimensions and radius.
#print("a:", a, "b:", b, "r:", r)
# Return image and tuple.
@@ -116,11 +116,11 @@ class CircleDetection:
# Read image.
#img_input = cv2.imread(fn, cv2.IMREAD_COLOR)
# Preprocess image.
gray_blur = pre_img(img_input)
gray_blur = self.pre_img(img_input)
# Process image. Detect circles.
detected_circles = detect_circles(gray_blur)
detected_circles = self.detect_circles(gray_blur)
# Postrocess image.
img_output, _ = post_img(img_input, detected_circles)
img_output, _ = self.post_img(img_input, detected_circles)
# Write image.
#cv2.imwrite("detected_circle.jpg", img_output)
# Return None.