Python通过OpenCV圆形检测找到图片上的圆形区域

173 阅读1分钟

1、我们首先需要找到两张图片,对其进行absdiff计算

2、然后对diff之后的图片进行圆形检测

import cv2
import numpy as np
import math

firstImage = cv2.imread("1.jpg")
layer2Image = cv2.imread("2.jpg")

diffImage = cv2.absdiff(layer2Image, firstImage)

cv2.imwrite("diff.jpg", diffImage)

gray = cv2.cvtColor(diffImage, cv2.COLOR_BGR2GRAY)

# 圆形检测
circles = cv2.HoughCircles(gray, cv2.HOUGH_GRADIENT, 1,
                           60, param1=170, param2=70, minRadius=1, maxRadius=100)
circles = np.uint16(np.around(circles))
circles_count = 0

for i in circles[0, :]:
    cv2.circle(diffImage, (i[0], i[1]), i[2], (30, 100, 180), 2)
    circles_count = 1+circles_count
    cv2.circle(diffImage, (i[0], i[1]), 2, (0, 255, 255), 6)
print("Count:", circles_count)

cv2.imwrite('circled.jpg', diffImage)

print(circles.shape)
print(circles)


cv2.imshow('image', diffImage)


cv2.waitKey(0)
cv2.destroyAllWindows()