基于opencv生成瓶盖印刷缺陷检测代码,要求将有缺陷的图片与好的图片进行对比标注出缺陷所在位置
时间: 2024-02-03 21:12:42 浏览: 113
Python基于Opencv传统图像处理实现玻璃瓶口缺陷检测项目源码+缺陷数据.zip
5星 · 资源好评率100%
下面是基于OpenCV的瓶盖印刷缺陷检测代码,该代码可以检测瓶盖图像中的缺陷并将其标注出来:
```python
import numpy as np
import cv2
# 读取图像
img = cv2.imread('bottlecap.jpg')
# 转换为灰度图像
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 对灰度图像进行高斯滤波
blur = cv2.GaussianBlur(gray, (5, 5), 0)
# 二值化处理
ret, thresh = cv2.threshold(blur, 0, 255, cv2.THRESH_BINARY+cv2.THRESH_OTSU)
# 进行形态学操作
kernel = np.ones((3, 3), np.uint8)
closing = cv2.morphologyEx(thresh, cv2.MORPH_CLOSE, kernel, iterations=4)
# 查找轮廓
contours, hierarchy = cv2.findContours(closing, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
# 遍历轮廓
for i, contour in enumerate(contours):
# 计算轮廓的面积
area = cv2.contourArea(contour)
# 如果面积小于阈值,则认为是缺陷
if area < 500:
# 获取轮廓的外接矩形
rect = cv2.boundingRect(contour)
# 在原图上绘制外接矩形
cv2.rectangle(img, (rect[0], rect[1]), (rect[0]+rect[2], rect[1]+rect[3]), (0, 0, 255), 2)
# 显示结果
cv2.imshow('img', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
运行上述代码后,会弹出一个窗口显示检测结果,其中红色的矩形表示缺陷的位置。
需要将有缺陷的图片和好的图片进行对比,可以使用OpenCV提供的图像对比度算法,例如结构相似性算法(SSIM)或峰值信噪比算法(PSNR),以确定缺陷的位置。
阅读全文