import cv2 as cv
import numpy as np
image = cv.imread('/home/chen/00/yolov5_attention-main/runs/detect/exp18/crops/solid/51.jpg')
image2 = cv.imread('/home/chen/00/yolov5_attention-main/runs/detect/exp18/51.png')
gray = cv.cvtColor(image, cv.COLOR_BGR2GRAY)
ret, binary_img = cv.threshold(gray, 160, 255, cv.THRESH_BINARY)
xgrd = cv.Sobel(binary_img,cv.CV_16SC1,1,0)
ygrd = cv.Sobel(binary_img,cv.CV_16SC1,0,1)
edges = cv.Canny(xgrd,ygrd,50,150)
h,w=edges.shape
print(h,w)
list_x=[]
list_y=[]
for i in range(1,h,1):
for j in range(1,w,1):
if edges[i][j] == 255:
list_x.append(j)
list_y.append(i)
# print(list_x)
# print(list_y)
z = list(zip(list_x,list_y))
# print(z)
point_size = 1
point_color = (0, 255, 255) # BGR
point_color=255#单通道
thickness = 4 # 可以为 0 、4、8
img=edges
for point in z:
cv.circle(img,(point[0],point[1]), point_size, point_color, thickness)
print(z)
cv.namedWindow("images")
cv.imshow('images', img)
cv.waitKey (0) #
cv.destroyAllWindows()
自己用的