导读
本文给大家分享一个用OpenCV传统方法实现形状检测的小案例。
背景介绍
实例来源:https://github.com/akshaybhatia10/ComputerVision-Projects/tree/master/FindShapes
其中典型的测试图片如下:
上图中包含了矩形、正方形、三角形、圆形和五角形共5种形状,我们的目的是将其定位并标注对应的形状,效果如下:
实现步骤
【1】 图片转为灰度图,做二值化。原图比较简单,可以用固定阈值或OTSU阈值方法;
# 固定阈值
ret,thresh = cv2.threshold(gray, 70, 255, cv2.THRESH_BINARY_INV)
# OTSU阈值
ret,thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)