import cv2 as cv

import cv2

import numpy as np

import time

from matplotlib import pyplot as plt

value = 60

value1 = 0

canny_min = 100

canny_max = 150

minlineLength = 200

maxlineGap = 10

area_threshold = 90

#用于去除杂点

def removeLittlePoint(img,threshold):

image, contours, hierarch = cv2.findContours(img, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)

for i in range(len(contours)):

area = cv2.contourArea(contours[i])

if area < threshold:

cv2.drawContours(image, [contours[i]], 0, 0, cv2.FILLED)

return image

def callback(object):

global value,value1,src2,src2_copy,src,minlineLength,maxlineGap,canny_max,canny_min,area_threshold

src2[:,:] = src2_copy[:,:]

#得到滑动条对应的值

value = cv2.getTrackbarPos('value', 'adapative')

value1 = cv2.getTrackbarPos('value1', 'adapative')

canny_min = cv2.getTrackbarPos('canny_min', 'adapative')

canny_max = cv2.getTrackbarPos('canny_max', 'adapative')

minlineLength = cv2.getTrackbarPos('minlineLength', 'adapative')

maxlineGap = cv2.getTrackbarPos('maxlineGap', 'adapative')

area_threshold = cv2.getTrackbarPos('area_threshold', 'adapative')

src = cv2.bilateralFilter(src2, 11, value, value1)

cv2.imshow("src", src)

src = cv2.cvtColor(src, cv2.COLOR_BGR2GRAY)

roi = cv2.adaptiveThreshold(src, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, \

cv2.THRESH_BINARY, 11, 3)

mask = np.ones((3, 1), np.uint8)

roi = cv2.morphologyEx(roi, cv2.MORPH_OPEN, mask, iterations=2)

roi = cv2.bitwise_not(roi)

roi = cv2.medianBlur(roi, 3,)

cv2.imshow('roi', roi)

roi_all = cv2.resize(roi, (round(roi.shape[1]/6), round(roi.shape[0]/6)))

cv2.imshow('roi_all', roi_all)

roi_canny = cv2.Canny(roi, canny_min, canny_max)

roi_canny = cv2.resize(roi_canny, (round(roi_canny.shape[1] ), round(roi_canny.shape[0] )))

cv2.imshow('roi_gray', roi_canny)

oshow = roi_canny.copy()

oshow = cv2.cvtColor(roi_canny, cv2.COLOR_GRAY2BGR)

draw = np.zeros((roi_canny.shape[0],roi_canny.shape[1]), dtype=np.uint8)

#霍夫直线检测，并绘出

lines = cv2.HoughLines(roi_canny, 1, np.pi / 180, 1000)

print(len(lines))

for line in lines:

rho, theta = line[0]

a = np.cos(theta)

b = np.sin(theta)

x0 = a * rho

y0 = b * rho

x1 = int(x0 + 10000 * (-b))

y1 = int(y0 + 10000 * (a))

x2 = int(x0 - 10000 * (-b))

y2 = int(y0 - 10000 * (a))

cv2.line(oshow, (x1, y1), (x2, y2), (0, 0, 255), 6,cv2.LINE_AA)

cv2.line(draw, (x1, y1), (x2, y2), 255, 6, cv2.LINE_AA)

oshow1 = cv2.resize(oshow, (round(oshow.shape[1] / 6), round(oshow.shape[0] / 6)))

cv2.imshow('oshow', oshow1)

kernel = np.ones((3, 3), np.uint8)

dilation = cv2.dilate(oshow, kernel, iterations=6)

draw = cv2.dilate(draw, kernel, iterations=6)

dilation_all = cv2.resize(dilation, (round(dilation.shape[1] / 6), round(dilation.shape[0] / 6)))

cv2.imshow('dilation_all', dilation_all)

cv2.imshow('dilation', dilation)

cv2.imshow('draw', draw)

output = cv2.bitwise_and(draw, roi)

result = roi[:, :] - output[:, :]

output1 = cv2.resize(output, (round(output.shape[1]/6), round(output.shape[0]/6)))

cv2.imshow('output', output1)

result1 = cv2.resize(result, (round(result.shape[1]/6), round(result.shape[0]/6)))

cv2.imshow('result', result1)

new_result = removeLittlePoint(result, area_threshold)

new_result1 = cv2.resize(new_result, (round(new_result.shape[1] / 6), round(new_result.shape[0] / 6)))

cv2.imshow('new_result1', new_result1)

#这个没用的

# lines = cv2.HoughLinesP(roi, 1, np.pi/180, 1, minLineLength=minlineLength, maxLineGap=maxlineGap)

# print(len(lines))

#

# for line in lines:

# x1,y1,x2,y2 = line[0]

# cv2.line(oshow,(x1,y1),(x2,y2),(255,0,0),5)

# cv2.imshow('oshow', oshow)

#Sobely = cv2.Sobel(roi, cv2.CV_64F, 0, 1)

#Sobely = cv2.convertScaleAbs(Sobely)

#cv2.imshow('sobely',Sobely)

dianpianwrong = cv.imread("E:/python_opencv_demo/dianPianTest/ELtestpicture/ELtestPicture/ok/P402202008310005.jpg")

src2 = dianpianwrong.copy()

dianpianwrong = cv2.resize(dianpianwrong, (round(dianpianwrong.shape[1]/6 ), round(dianpianwrong.shape[0]/6 )))

print("src.shape", src2.shape)

cv2.imshow("origon-picture", dianpianwrong)

src2_copy = np.zeros((src2.shape[0], src2.shape[1],3), np.uint8)

src2_copy[:, :] = src2[:, :]

cv2.namedWindow("adapative", cv2.WINDOW_AUTOSIZE)

#用于创建滑动条便于调试

cv2.createTrackbar('value', 'adapative', value, 255, callback)

cv2.createTrackbar('value1', 'adapative', value1, 255, callback)

cv2.createTrackbar('canny_min', 'adapative', canny_min, 255, callback)

cv2.createTrackbar('canny_max', 'adapative', canny_max, 255, callback)

cv2.createTrackbar('minlineLength', 'adapative', minlineLength, 255, callback)

cv2.createTrackbar('maxlineGap', 'adapative', maxlineGap, 255, callback)

cv2.createTrackbar('area_threshold', 'adapative', area_threshold, 255, callback)

cv2.waitKey(0)

原文链接:https://blog.csdn.net/weixin_44289316/article/details/109178506