completed circle detection with certain errors

shreyakupadhyay · May 13, 2018 · feee738 · feee738
1 parent 2b1beeb
commit feee738
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/DFA/shapes/circle_detection.py b/DFA/shapes/circle_detection.py
@@ -5,18 +5,23 @@
 import numpy as np
 
 def DetectCircles(image):
-    houghcircles = cv2.HoughCircles(image, cv2.HOUGH_GRADIENT,1,20,param1=50,param2=30,minRadius=35,maxRadius=60)
+    minpercent = 0.13
+    maxpercent = 0.33
+    height,width = image.shape[:2]
+    minValue = min(height,width)
+    maxValue = max(height,width)
+    min_radius = int(minpercent * minValue/2)
+    max_radius = int(maxpercent * maxValue/2)
+    houghcircles = cv2.HoughCircles(image, cv2.HOUGH_GRADIENT,1,20,param1=50,param2=30,minRadius=min_radius,maxRadius=max_radius)
     houghcircles = np.uint16(np.around(houghcircles))
     return houghcircles
 
 def test():
     img = cv2.imread(sys.argv[1],0)
-    img = cv2.medianBlur(img,5)
-    editimg = cv2.cvtColor(img,cv2.COLOR_GRAY2BGR)
     for i in DetectCircles(img)[0,:]:
-        cv2.circle(editimg,(i[0],i[1]),i[2],(0,255,0),2)
-        cv2.circle(editimg,(i[0],i[1]),2,(0,0,255),3)
-    cv2.imshow('detected circles',editimg)
+        cv2.circle(img,(i[0],i[1]),i[2],(0,255,0),2)
+        cv2.circle(img,(i[0],i[1]),2,(0,0,255),3)
+    cv2.imshow('detected circles',img)
     cv2.waitKey(0)
     cv2.destroyAllWindows()