标签:shape ros eva 检测 fill chain numpy date trace
import cv2
import numpy as np
#载入图片
#img_original=cv2.imread(‘1110_3_1_X_537_Y_1700_qly_100_fail_13d17h46m24s.bmp‘,0)
img_original=cv2.imread(‘1141_1_1_X_551_Y_1701_qly_100_pass_subtpl_25_update_1153677968_13d17h53m10s.bmp‘,0)
h,w=img_original.shape
img_original=cv2.resize(img_original,(3*w,3*h))
rangecolorimg=img_original.copy()
#设置窗口
cv2.namedWindow(‘Canny‘)
cv2.namedWindow(‘morphed‘)
cv2.namedWindow(‘colorRange‘)
#定义回调函数
def nothing(x):
pass
#创建两个滑动条,分别控制threshold1,threshold2
cv2.createTrackbar(‘threshold1‘,‘Canny‘,50,100,nothing)
cv2.createTrackbar(‘threshold2‘,‘Canny‘,100,100,nothing)
cv2.createTrackbar(‘morph‘,‘morphed‘,5,20,nothing)
cv2.createTrackbar(‘range1‘,‘colorRange‘,0,255,nothing)
cv2.createTrackbar(‘range2‘,‘colorRange‘,100,255,nothing)
cv2.createTrackbar(‘replacevalue‘,‘colorRange‘,120,255,nothing)
while(1):
#返回滑动条所在位置的值
rangecolorimg=img_original.copy()
blackcolorimg=img_original.copy()
threshold1=cv2.getTrackbarPos(‘threshold1‘,‘Canny‘)
threshold2=cv2.getTrackbarPos(‘threshold2‘,‘Canny‘)
range1=cv2.getTrackbarPos(‘range1‘,‘colorRange‘)
range2=cv2.getTrackbarPos(‘range2‘,‘colorRange‘)
replacevalue=cv2.getTrackbarPos(‘replacevalue‘,‘colorRange‘)
maskrange=np.zeros_like(img_original)
blackrange=np.ones_like(img_original)*255
x1=img_original[img_original>range1]
x2=img_original[img_original<range2]
#x3=(img_original>range1)&(img_original<range2)
x3 = cv2.inRange(img_original, range1, range2)
import pdb
#pdb.set_trace()
#x3=cv2.bitwise_and(x1, x2)
#bitwise_and
#rangecolorimg[x3]=255
rangecolorimg[x3!=0]=replacevalue
blackmask = 255 * np.ones(rangecolorimg.shape, rangecolorimg.dtype)
#blackmask=src_mask
width, height = img_original.shape
center = (int(height/2), int(width/2))
seamlessCloneimg=rangecolorimg
#seamlessCloneimg = cv2.inpaint(seamlessCloneimg, x3, 10, cv2.INPAINT_NS)#cv2.INPAINT_NScv2.INPAINT_TELEA
seamlessCloneimg = cv2.seamlessClone(cv2.cvtColor(img_original, cv2.COLOR_GRAY2BGR) ,cv2.cvtColor(rangecolorimg, cv2.COLOR_GRAY2BGR) , blackmask, center, cv2.MIXED_CLONE)
blackrange[x3!=0]=blackcolorimg[x3!=0]
normalizeimg = np.zeros(blackrange.shape, dtype=np.float32)
cv2.normalize(np.float32(rangecolorimg), dst=normalizeimg, alpha=1.0, beta=0, norm_type=cv2.NORM_MINMAX)
#Canny边缘检测
img_edges=cv2.Canny(img_original,threshold1,threshold2)
#显示图片
img=img_edges
morphradius=cv2.getTrackbarPos(‘morph‘,‘morphed‘)
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (morphradius,morphradius))
morphed = cv2.morphologyEx(img, cv2.MORPH_CLOSE, kernel)
contours,_ = cv2.findContours(morphed, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
import pdb
#pdb.set_trace()
rgbmorphed=cv2.cvtColor(morphed, cv2.COLOR_GRAY2BGR)
# for i in range(len(contours)):
# #cv2.drawContours(immask,[contours[i]],-1,(x+np.random.randint(0, 30),x+np.random.randint(0, 30),x+np.random.randint(0, 30)),cv2.FILLED)
# cv2.drawContours(rgbmorphed,[contours[i]],-1,(0,255,0),cv2.FILLED)
cv2.drawContours(rgbmorphed,contours,-1,(255,255,255),cv2.FILLED)
import pdb
#pdb.set_trace()
if(0):
cv2.imshow("morphed2",morphed)
cv2.imshow("morphed",morphed)
cv2.imshow(‘Canny‘,img_edges)
cv2.imshow(‘rgbmorphed‘,rgbmorphed)
#cv2.waitKey(0)
cv2.imshow("colorRange",rangecolorimg)
cv2.imshow("x3",x3)
cv2.imshow("blackrange",blackrange)
cv2.imshow(‘normalizeimg‘,np.uint8(normalizeimg*255))
#cv2.waitKey(0)
cv2.imshow(‘original‘,img_original)
cv2.imshow(‘seamlessCloneimg‘,seamlessCloneimg)
if cv2.waitKey(1)==ord(‘q‘):
break
cv2.destroyAllWindows()
createTrackbar
标签:shape ros eva 检测 fill chain numpy date trace
原文地址:https://www.cnblogs.com/skydaddy/p/11698362.html