用python和opencv实现物体框选并保存坐标信息及面积信息（附代码）

原图：

框选后：

整体思路：将图片进行 高斯滤波 + 灰度处理 + 二值化 等操作，获取处理后图片的HSV值；根据HSV值的范围使用cv2.inRange函数去除背景，查找轮廓线，框选最小矩形，获取坐标信息及面积信息；设置面积范围，在范围内再进行框选，（也可在图片中显示坐标信息并将矩形按面积大小排序），最后保存框选后的图片。

高斯滤波后：

灰度处理后：

二值化：

  处理后显示坐标等信息的结果：

保存的坐标信息文件：

面积信息文件：

 添加面积限制条件前的结果：

源代码：

import cv2
import numpy as np
import os

# 高斯滤波
def GausBlur(src):
    dst = cv2.GaussianBlur(src, (5, 5), 1.5)
    cv2.imshow('GausBlur', dst)
    return dst

# 灰度处理
def Gray_img(src):
    gray = cv2.cvtColor(src, cv2.COLOR_BGR2GRAY)
    cv2.imshow('gray', gray)
    return gray

# 二值化
def threshold_img(src):
    ret, binary = cv2.threshold(src, 0, 255, cv2.THRESH_BINARY | cv2.THRESH_TRIANGLE)
    # print("threshold value %s" % ret)
    cv2.imshow('threshold', binary)
    return binary


# 轮廓绘制
def process(image):
    hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
    hsv = cv2.medianBlur(hsv, 5)

    mask = cv2.inRange(hsv, (0, 0, 1), (0, 0, 255))  # inrange

    line = cv2.getStructuringElement(cv2.MORPH_RECT, (5, 5), (-1, -1))
    mask = cv2.morphologyEx(mask, cv2.MORPH_OPEN, line)
    # cv2.imshow("mask", mask)

    contours, hierarchy = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
    contours_num = len(contours)
    # print('hierarchy', hierarchy)
    # print(contours_num)

    i = 0
    while i < contours_num:
        c = sorted(contours, key=cv2.contourArea, reverse=True)[i]  # 排序，key为排序比较元素，true为降序
        rect = cv2.minAreaRect(c)
        box = np.int0(cv2.boxPoints(rect))
        # print(box)

        # file_path = os.path.join(os.getcwd(), "Coordinate information", time.strftime("%Y_%m_%d_%H_%M_%S", time.localtime()) + '.txt')
        file_path = os.path.join(os.getcwd(), r".informationcoordinate_information.txt")
        file = open(file_path, "a+")
        file.write(str(box) + ',' + 'n')

        area = cv2.contourArea(box)
        file_path = os.path.join(os.getcwd(), r".informationrect_area.txt")
        file = open(file_path, "a+")
        file.write("rect " + str(i+1) + " area:" + str(area) + 'n')

        if area >= 1300 and area <= 2000:
            cv2.drawContours(image, [box], -1, (0, 255, 0), 3)
            # cv2.putText(image, str(i+1), (box[0][0], box[0][1]), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 111, 111), 3)
            # cv2.putText(image, "1." + str(box[0]), (box[0][0], box[0][1]), cv2.FONT_HERSHEY_SIMPLEX, 0.3, (0, 0, 255), 1)
            # cv2.putText(image, "2." + str(box[1]), (box[1][0], box[1][1]), cv2.FONT_HERSHEY_SIMPLEX, 0.3, (0, 0, 255), 1)
            # cv2.putText(image, "3." + str(box[2]), (box[2][0], box[2][1]), cv2.FONT_HERSHEY_SIMPLEX, 0.3, (0, 0, 255), 1)
            # cv2.putText(image, "4." + str(box[3]), (box[3][0], box[3][1]), cv2.FONT_HERSHEY_SIMPLEX, 0.3, (0, 0, 255), 1)
            save_path = r".imagesscreenshottarget_rect.jpg"
            cv2.imencode('.jpg', image)[1].tofile(save_path)

        i += 1

    return image