#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Thu Mar 21 12:42:15 2019
@author: lg
"""
import cv2
import numpy as np
# 形态学处理
def Process(img):
# 高斯平滑
gaussian = cv2.GaussianBlur(img, (3, 3), 0, 0, cv2.BORDER_DEFAULT)
# 中值滤波
median = cv2.medianBlur(gaussian, 5)
# Sobel算子
# 梯度方向: x
sobel = cv2.Sobel(median, cv2.CV_8U, 1, 0, ksize=3)
# 二值化
ret, binary = cv2.threshold(sobel, 170, 255, cv2.THRESH_BINARY)
# 核函数
element1 = cv2.getStructuringElement(cv2.MORPH_RECT, (9, 1))
element2 = cv2.getStructuringElement(cv2.MORPH_RECT, (9, 7))
# 膨胀
dilation = cv2.dilate(binary, element2, iterations=1)
# 腐蚀
erosion = cv2.erode(dilation, element1, iterations=1)
# 膨胀
dilation2 = cv2.dilate(erosion, element2, iterations=3)
return dilation2
def GetRegion(img):
regions = []
# 查找轮廓
contours, hierarchy = cv2.findContours(img, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
for contour in contours:
area = cv2.contourArea(contour)
if (area < 5000):
continue
eps = 1e-3 * cv2.arcLength(contour, True)
approx = cv2.approxPolyDP(contour, eps, True)
rect = cv2.minAreaRect(contour)
box = cv2.boxPoints(rect)
box = np.int0(box)
height = abs(box[0][1] - box[2][1])
width = abs(box[0][0] - box[2][0])
ratio =float(width) / float(height)
if (ratio < 5 and ratio > 1.8):
regions.append(box)
return regions
def detect(img):
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
prc = Process(gray)
regions = GetRegion(prc)
return regions
#输入的参数为图片的路径
img = cv2.imread('red.jpg')
b=detect(img)
b1=b[0]
hm=b1[:,1].max()
hn=b1[:,1].min()
cm=b1[:,0].max()
cn=b1[:,0].min()
# -*- coding=GBK -*-
import cv2 as cv
import numpy as np
#截取图片中的指定区域或在指定区域添加某一图片
def jie_image(src1):
src2 = src1[hn:hm,cn:cm]#截取第5行到89行的第500列到630列的区域
cv.imshow("截取", src2)
cv2.imwrite('jiequ.jpg', src2)
# src1[105:189, 300:430] = src2#指定位置填充,大小要一样才能填充
# cv.imshow("合成", src1)
src = cv.imread("cc.jpeg")
#cv.imshow("原来", src)
jie_image(src)
cv.waitKey(0)
cv.destroyAllWindows()