基于OpenCV的图像颜色与形状识别设计与实现实验指导书

基于OpenCV的图像颜色与形状识别设计与实现实验指导书

一、实验目的:

通过本实验,学生将了解图像颜色与形状的基本概念,并掌握使用OpenCV进行图像颜色与形状识别的方法。具体操作包括图像剪裁、颜色识别、轮廓检测。

二、实验器材:

  • 计算机
  • 安装了Python和OpenCV库的开发环境
  • 彩色图像

三、实验步骤:

1、导入必要的库:

import cv2 as cv

import numpy as np

import math

from PIL import Image, ImageDraw, ImageFont

2、设定颜色阈值: 根据需要识别的颜色,设定相应的颜色阈值。例如:

lower_red = np.array([0, 120, 100])

upper_red = np.array([10, 255, 255])

3、定义计算角度和距离的函数:

def calculating_angle(p1, p2, p0):

    # 计算角度

def calculating_distance(p0, p1):

    # 计算距离

4、定义形状识别函数:

def getShape(cnt):

    # 根据轮廓获取形状

5、定义添加文本和矩形框的函数:

def cv2ImgAddText(img, shapes, textColor=(255, 0, 0)):

    # 添加文本

def cv2ImgAddRect(img, shapes):

    # 添加矩形框

6、定义颜色分割函数:

def colorDivision(name, roi_bgr, lowerb, upperb):

    # 根据颜色阈值进行图像分割

7、加载图像:

roi_bgr = cv.imread('Shape1.png')

8、分割图像并进行形状识别:

# 每次分割一种颜色

colorDivision("红色", roi_bgr, lower_red, upper_red)

# 其他颜色的分割类似

# 添加文本和矩形框

roi_bgr = cv2ImgAddText(roi_bgr, shape_list)

roi_bgr = cv2ImgAddRect(roi_bgr, shape_list)

9、显示结果图像:

cv.imshow('roi_bgr', roi_bgr)

cv.waitKey()

cv.destroyAllWindows()

10、保存结果图像:

cv.imwrite("result.png", roi_bgr)

四、注意事项:

  • 实验中使用的图像应包含需要识别的颜色和形状。
  • 可根据需要自定义颜色阈值和形状判断条件。
  • 实验过程中可根据需要调整参数和算法以获得更好的识别效果。

五、实验扩展:

可以尝试在实验中添加其他功能,例如:添加图像预处理步骤,如高斯模糊、边缘检测等,以提高识别效果。在图像中绘制轮廓线、角点等标记,以更直观地展示识别结果。

对于无法识别的形状,尝试使用机器学习算法进行分类和识别。

通过以上实验,可以了解到图像颜色与形状识别的基本原理和方法,并掌握使用OpenCV进行相关操作的能力。同时,通过实际操作,学生可以加深对图像处理和计算机视觉的理解,为进一步的研究和应用打下基础。

六、实验报告:

撰写一个简单的实验报告,包括实验目的、步骤、结果和总结。报告中应包含实验过程中的所有数据、参数调整的观察和分析,以及对实验结果的讨论。