使用OpenCV-python提取图片中的硬币

大致思路如下

  1. 对图片做降噪滤波处理
  2. 寻找硬币的边缘
  3. 使用硬币边缘打造蒙版,并切割出图像

对应的方法如下

  1. 对图片进行二值化COLOR_BGR2GRAY处理,高斯滤波GaussianBlur处理。
  2. 使用canny算子进行边缘查找,并使用findContours方法提取边缘坐标点。
  3. 使用boundingRect方法找出硬币所在的矩形进行裁切,使用minEnclosingCircle找出包络圆作为蒙版,并使用bitwise_and隐藏背景。

首先导入必要的库,测试一下是否加载成功,注意cv2.waitKey(0)一定不要忘记!

import cv2 
import numpy as np

image = cv2.imread("coins.png")
cv2.imshow("if imread successfully",image)
cv2.waitKey(0)

使用OpenCV-python提取图片中的硬币_第1张图片
接着进行灰度处理和模糊处理,在这之前需要关掉第一张图像

gray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)
blurred = cv2.GaussianBlur(gray,(11,11),0)
cv2.imshow("blurred",blurred)
cv2.waitKey(0)

使用OpenCV-python提取图片中的硬币_第2张图片
进行边缘提取

edged = cv2.Canny(blurred, 30, 150)
cv2.imshow("edged",edged)
cv2.waitKey(0)

使用OpenCV-python提取图片中的硬币_第3张图片
在原图中进行边缘提取

# 注意这里只能返回2个变量
(cnts,_) = cv2.findContours(edged.copy(),cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
print(" {} coins in the image".format(len(cnts)))
 显示:9 coins in the image

然后画在原图中

coins = image.copy()
cv2.drawContours(coins, cnts, -1, (0,255,0), 2)
cv2.imshow("coins",coins)
cv2.waitKey(0)

使用OpenCV-python提取图片中的硬币_第4张图片
最后对图片进行裁切

coins = image.copy()
cv2.drawContours(coins, cnts, -1, (0,255,0), 2)
cv2.imshow("coins",coins)
cv2.waitKey(0)

for (i,circle) in enumerate(cnts):
    # circle是每个提取出来的圆
    (x,y,w,h) = cv2.boundingRect(circle)
    print("coin #{}".format(i+1))
    coin_canvas = image[y:y+h, x:x+w]
    cv2.imshow("coin",coin_canvas)
    cv2.waitKey(0)
    
    mask = np.zeros(image.shape[:2], dtype = 'uint8')
    ((centerX,centerY),radius) = cv2.minEnclosingCircle(circle)
    # 在蒙版上画圆,这里是灰度图,所以255表示白色
    cv2.circle(mask,(int(centerX),int(centerY)), int(radius), 255, -1)
    mask = mask[y:y+h, x:x+w]
    # 自身和自身进行与运算即可
    cv2.imshow("masked coin",cv2.bitwise_and(coin_canvas, coin_canvas, mask = mask))
    cv2.waitKey(0)

裁切出来的图片
在这里插入图片描述
加上蒙版的图片
在这里插入图片描述
以上代码均在notebook中运行通过。

你可能感兴趣的:(OpenCV)