基于python使用OpenCV进行物体轮廓排序

作者:赵卓不凡? 时间:2022-06-27 23:25:01 

1 引言

在进行图像处理过程中,我们经常会遇到一些和物体轮廓相关的操作,比如求目标轮廓的周长面积等,我们直接使用OpencvfindContours函数可以很容易的得到每个目标的轮廓,但是可视化后, 这个次序是无序的,如下图左侧所示:

基于python使用OpenCV进行物体轮廓排序

本节打算实现对物体轮廓进行排序,可以实现从上到下排序或者从左倒右排序,达到上图右侧的可视化结果.

2 栗子

2.1 读取图像

首先,我们来读取图像,并得到其边缘检测图,代码如下:

image = cv2.imread(args['image'])
accumEdged = np.zeros(image.shape[:2], dtype='uint8')
for chan in cv2.split(image):
    chan = cv2.medianBlur(chan, 11)
    edged = cv2.Canny(chan, 50, 200)
    accumEdged = cv2.bitwise_or(accumEdged, edged)
cv2.imshow('edge map', accumEdged)

运行结果如下:

基于python使用OpenCV进行物体轮廓排序

左侧为原图,右侧为边缘检测图.

2.2 获取轮廓

opencv-python中查找图像轮廓的API为:findContours 函数,该函数接收二值图像作为输入,可输出物体外轮廓、内外轮廓等等.

代码如下:

cnts = cv2.findContours(accumEdged.copy(), cv2.RETR_EXTERNAL,  cv2.CHAIN_APPROX_SIMPLE)
cnts = grab_contours(cnts)
cnts = sorted(cnts, key=cv2.contourArea, reverse=True)[:5]
orig = image.copy()
# unsorted
for (i, c) in enumerate(cnts):
    orig = draw_contour(orig, c, i)
cv2.imshow('Unsorted', orig)
cv2.imwrite("./Unsorted.jpg", orig)

运行结果如下:

基于python使用OpenCV进行物体轮廓排序

需要注意的是,在OpenCV2.X版本,函数findContours返回两个值,

函数声明如下:

contours, hierarchy = cv2.findContours(binary,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)

但是在OpenCV3以上版本,该函数的声明形式如下:

image, contours, hierarchy = cv2.findContours(binary, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

所以为了适配两种模式,我们实现函数grab_contours 来根据不同的版本,选择对应的返回轮廓的下标位置,

代码如下:

def grab_contours(cnts):
    # if the length the contours tuple returned by cv2.findContours
    # is '2' then we are using either OpenCV v2.4, v4-beta, or
    # v4-official
    if len(cnts) == 2:
        cnts = cnts[0]

    # if the length of the contours tuple is '3' then we are using
    # either OpenCV v3, v4-pre, or v4-alpha
    elif len(cnts) == 3:
        cnts = cnts[1]

    return cnts

2.3 轮廓排序

通过上述步骤,我们得到了图像中的所有物体的轮廓,接下来我们定义函数sort_contours函数来实现对轮廓进行排序操作,该函数接受method参数来实现按照不同的次序对轮廓进行排序,比如从左往右,或者从右往左.

代码如下:

def sort_contours(cnts, method='left-to-right'):
    # initialize the reverse flag and sort index
    reverse = False
    i = 0
    # handle if sort in reverse
    if method == 'right-to-left' or method == 'bottom-to-top':
        reverse = True
    # handle if sort against y rather than x of the bounding box
    if method == 'bottom-to-top' or method == 'top-to-bottom':
        i = 1

    boundingBoxes = [cv2.boundingRect(c) for c in cnts]
    (cnts, boundingBoxes) = zip(*sorted(zip(cnts, boundingBoxes), key=lambda b: b[1][i], reverse=reverse))
    return (cnts, boundingBoxes)

上述代码的核心思想为先求出每个轮廓的外接矩形框,然后通过对外接框按照x或y坐标排序进而来实现对轮廓的排序.

调用代码如下:

# sorted
(cnts, boundingboxes) = sort_contours(cnts, method=args['method'])
for (i, c) in enumerate(cnts):
    image = draw_contour(image, c, i)
cv2.imshow('Sorted', image)
cv2.waitKey(0)

运行结果如下:

2.4 其他结果

利用上述代码,我们也可以实现从左往右的排序,如下所示:

基于python使用OpenCV进行物体轮廓排序

3 总结

本文利用OpenCV实现了对物体轮廓按指定顺序进行排序的功能,并给出了完整的代码示例.

来源:https://blog.csdn.net/sgzqc/article/details/121981053?spm=1001.2014.3001.5502

标签:python,OpenCV,物体轮廓,排序
0
投稿

猜你喜欢

  • Golang开发库的集合及作用说明

    2024-02-01 08:03:24
  • js 中将多个逗号替换为一个逗号的代码

    2024-04-16 10:32:48
  • matlab xlabel位置的设置方式

    2022-06-11 06:51:24
  • 如何显示数据库里的图片?

    2010-06-08 09:36:00
  • pandas is in和not in的使用说明

    2023-04-04 22:06:11
  • 条件注释使用指南[译]

    2009-03-23 17:41:00
  • SQL SERVER 中构建执行动态SQL语句的方法

    2024-01-22 10:01:46
  • 点球小游戏python脚本

    2022-07-17 23:28:03
  • Python大数据量文本文件高效解析方案代码实现全过程

    2023-01-18 04:57:01
  • Python模块学习 datetime介绍

    2023-08-15 16:01:18
  • Django中select_related和prefetch_related的用法与区别详解

    2023-10-08 12:38:08
  • Python3 ffmpeg视频转换工具使用方法解析

    2021-07-28 15:00:30
  • js中的文档模式-document.compatMode

    2009-08-14 20:30:00
  • Python中的单行、多行、中文注释方法

    2022-03-21 21:56:38
  • 一篇文章入门Python生态系统(Python新手入门指导)

    2023-11-03 01:12:00
  • 人工智能学习pyTorch的ResNet残差模块示例详解

    2022-05-04 21:45:22
  • python神经网络VGG16模型复现及其如何预测详解

    2022-10-16 06:59:15
  • 关于Python 位运算防坑指南

    2022-06-13 08:06:55
  • Python学习笔记(一)(基础入门之环境搭建)

    2021-12-26 12:51:40
  • windows环境下Mysql中文乱码问题解决方法

    2024-01-27 01:32:59
  • asp之家 网络编程 m.aspxhome.com