一.横平竖直”进行连线
解法1.将一些坐标点按照x相等,y相等连起来
解法1.根据 x或y总有一个相等的,用np.sum来找出和为1的点,然后在连起来,存在重复连线的问题.
import numpy as npcoord = np.array([[10, 60],[10, 20],[20, 20],[40, 40],[40, 60],[20, 40]])img = np.zeros((100, 100))h, w = coord.shapefor i in range(h):index = np.sum(np.array(coord[i]).reshape(-1, 2) == coord, axis=-1)y = np.where(index == 1)for point in coord[y]:cv2.line(img, (coord[i][0], coord[i][1]), (point[0], point[1]), color=(255,255,255), thickness=3)cv2.imshow('img', img)cv2.waitKey(0)
解法2.先连x相等,在连y相等的
def coord_sort_ind(x,y):#按照x,y来排序ind = np.lexsort((y.tolist(), x.tolist())) # Sort by x, then by yreturn ind
def simple_line_():import numpy as npcoord = np.array([[10, 60],[10, 20],[20, 20],[40, 40],[40, 60],[20, 40]])# 按照先X后Y排的序ind = coord_sort_ind(coord[:, 0], coord[:, 1])coord_X=coord[ind]print(coord_X)# # 按照先Y后X排的序ind = coord_sort_ind(coord[:, 1], coord[:, 0])print(ind)coord_Y = coord[ind]print(coord_Y)img = np.zeros((100, 100))h, w = coord.shape# x想等的连接for i in range(0, h, 2):cv2.line(img, (coord_X[i][0], coord_X[i][1]), (coord_X[i + 1][0], coord_X[i + 1][1]), color=(255, 255, 255),thickness=3)# y想等的连接cv2.line(img, (coord_Y[i][0], coord_Y[i][1]), (coord_Y[i + 1][0], coord_Y[i + 1][1]), color=(255, 255, 255),thickness=3)cv2.imshow('img', img)cv2.waitKey(0)
二.将相邻框进行合并
目前解法是将两个相邻的框合并，得到大框在继续合并，算法还需要优化成一次性合并成功
import numpy as np
import cv2
def get_cluster(points,x_dis,y1_dis):points = points[points[:, 0].argsort()]x1,y1,x2,y2=points[:,0],points[:,1],points[:,-2],points[:,-1]#只对竖值框进行合并操作points=points[np.where(x2-x1<y2-y1)[0]]# points=points[:13]# print('points:',points)# # debug to show# for i,point in enumerate(points):# font = cv2.FONT_HERSHEY_SIMPLEX# cv2.putText(img, str(i), (point[0], point[1]), font, 2.5, (255, 255, 255), 2)# cv2.rectangle(img, (point[0], point[1]), (point[-2], point[-1]), color=(255, 255, 255), thickness=3)# cv2.imwrite('img_rect.jpg', img)# print(points.shape[0])cluster = []while points.shape[0]:temp=[]# print('points:',points)#x1-x2<index_x = np.where(abs(points[0][-2]-points[:, 0]) < x_dis)[0]# print('index_x:',index_x)# if len(index_x):# #y1-y1<index_y1 = np.where(abs(points[0][1]-points[:, 1]) < y1_dis)[0]# print('index_y: ',index_y1)#y2-y2<hindex_y2 = np.where(abs(points[0][-1]-points[:, -1]) < (points[:, -1]-points[:, 1]))[0]# print('index_y2:',index_y2)# common_index=[i for i in index_x if i in index_y]common_index = list(set(index_x) & set(index_y1) & set(index_y2))# print('common_index:',common_index)# print('common_index:',common_index)if len(common_index):temp.append(points[0])temp.extend(points[common_index])points = np.delete(points, obj=common_index+[0], axis=0)else:temp.append(points[0])points=points[1:]cluster.append(temp)# # print(points)# print('cluster:',cluster)# print(len(cluster))# for i in range(len(cluster)):# every_cluster=np.array(cluster[i]).reshape(-1,4)# print('every_cluster:',every_cluster)# for point in every_cluster:# font = cv2.FONT_HERSHEY_SIMPLEX# cv2.putText(img, str(i), (point[0],point[1]), font, 2, (255, 255, 255), 2)# cv2.rectangle(img,(point[0],point[1]),(point[-2],point[-1]),color=(255,255,255),thickness=3)# cv2.imwrite('img_rect_clu.jpg',img)return cluster
def fix_bbox_point(cluster):points=[]for i,evry_cluster in enumerate(cluster):every_cluster = np.array(cluster[i]).reshape(-1, 4)x1,y1,x2,y2=np.min(every_cluster[:,0]),np.min(every_cluster[:,1])\,np.max(every_cluster[:,-2]),np.max(every_cluster[:,-1])points.append([x1,y1,x2,y2])return np.array(points)def union_bboxs(points):cluster_first = get_cluster(points, x_dis=50, y1_dis=300)# print('===cluster=====')# print(len(cluster_first))first_bboxs = fix_bbox_point(cluster_first)cluster_second = get_cluster(first_bboxs, x_dis=50, y1_dis=300)# print('===cluster=====')# print(len(cluster_second))second_bboxs = fix_bbox_point(cluster_second)cluster_third = get_cluster(second_bboxs, x_dis=50, y1_dis=400)# print('===cluster=====')# print(len(cluster_third))third_bboxs = fix_bbox_point(cluster_third)return third_bboxsif __name__ == '__main__':# points = np.array([[1264, 1480, 1346, 1790],# [2278, 1873, 2380, 2542],# [3209, 2279, 3286, 2599],# [2195, 3468, 2277, 3772],# [844, 1087, 924, 1399],# [92, 876, 181, 1327],# [951, 306, 1033, 596],# [1652, 299, 1729, 604],# [2196, 1957, 2270, 2535],# [916, 4251, 994, 4563],# [1131, 1915, 1214, 2505],# [1042, 265, 1116, 532],# [2124, 2036, 2195, 2603],# [934, 1054, 1003, 1258],# [1522, 3561, 1585, 4086],# [999, 4220, 1068, 4422],# [2404, 586, 2516, 1209],# [1452, 3618, 1528, 4151],# [1354, 1449, 1425, 1655],# [1583, 3532, 1683, 4109],# [1207, 1525, 1256, 1792],# [2144, 3511, 2191, 3779],# [3162, 2319, 3205, 2549],# [787, 1124, 837, 1395],# [896, 339, 944, 606],# [1741, 260, 1813, 467],# [219, 3230, 330, 3785],# [1424, 2675, 1506, 2818],# [2528, 538, 2714, 1709],# [2287, 664, 2394, 1596],# [231, 1703, 428, 2598],# [287, 248, 473, 1638],# [186, 280, 265, 1025],# [1670, 3480, 1756, 3852],# [1754, 2277, 1836, 2431],# [3284, 2241, 3360, 2455],# [2823, 1662, 3163, 1730],# [2897, 2852, 3285, 2946],# [2972, 2700, 3049, 2790]])points=np.array([[700,1888, 789, 2199],[1918, 1487, 2000, 1798],[1045, 683, 1129, 995],[2867, 279, 2975, 958],[1929, 3088, 2012, 3393],[1006, 272, 1087, 583],[2476, 3089, 2560, 3396],[ 820, 3094, 904, 3406],[2532, 3499, 2616, 3810],[1335, 1080, 1411, 1378],[2204, 2283, 2288, 2587],[1138, 644, 1213, 975],[1611, 1884, 1695, 2196],[2823, 2283, 2906, 2603],[ 773, 2286, 853, 2606],[1562, 262, 1642, 585],[2458, 1883, 2540, 2196],[1749, 3491, 1831, 3803],[ 634, 1266, 715, 1785],[1646, 236, 1722, 515],[ 988, 3889, 1060, 4221],[ 725, 1167, 836, 1743],[2592, 455, 2703, 1365],[2721, 375, 2853, 1289],[ 846, 1104, 938, 1481],[ 857, 2255, 935, 2583],[1419, 1043, 1494, 1316],[1707, 2310, 1814, 3001],[1327, 3046, 1405, 3387],[1244, 3081, 1327, 3404],[1814, 2264, 1903, 2656],[2543, 1844, 2617, 2053],[2845, 3894, 2918, 4215],[2147, 2330, 2197, 2559],[2402, 1922, 2452, 2198],[1058, 3860, 1122, 4065],[3308, 1779, 3396, 2183],[2293, 2248, 2370, 2454],[1857, 1531, 1908, 1760],[1697, 3535, 1745, 3803],[1189, 3133, 1236, 3367],[2419, 3135, 2468, 3360],[1873, 3124, 1923, 3400],[ 988, 719, 1038, 995],[2480, 3539, 2529, 3809],[2017, 3057, 2083, 3278],[2095, 2368, 2138, 2598],[ 766, 3135, 815, 3409],[1559, 1964, 1607, 2160],[1504, 1966, 1550, 2202],[2913, 2248, 2996, 2455],[ 766, 4318, 838, 4580],[1802, 1571, 1848, 1797],[ 374, 801, 457, 1032],[ 637, 1926, 690, 2164],[1636, 2441, 1708, 2998],[ 964, 353, 1012, 549],[3243, 290, 3389, 1431],[1869, 3963, 1952, 4206],[3131, 2120, 3215, 2800],[2858, 4314, 2934, 4593],[2802, 3939, 2844, 4189],[ 484, 716, 570, 955],[2913, 3860, 2988, 4126],[ 330, 645, 566, 717]])# print(points[30:33])# points = points[12:21]img = np.zeros((5000, 4000))cluster_first=get_cluster(points,x_dis=50,y1_dis=300)print('===first cluster=====')print(len(cluster_first))first_bboxs=fix_bbox_point(cluster_first)print('===second cluster===')cluster_second = get_cluster(first_bboxs, x_dis=50, y1_dis=400)print(len(cluster_second))second_bboxs = fix_bbox_point(cluster_second)# #debug# for i,point in enumerate(first_bboxs):# font = cv2.FONT_HERSHEY_SIMPLEX# cv2.putText(img, str(i), (point[0], point[1]), font, 2.5, (255, 255, 255), 2)# cv2.rectangle(img, (point[0], point[1]), (point[-2], point[-1]), color=(255, 255, 255), thickness=3)# cv2.imwrite('img_rect_first_bbx.jpg', img)### # debug# for i, point in enumerate(second_bboxs):# font = cv2.FONT_HERSHEY_SIMPLEX# cv2.putText(img, str(i), (point[0], point[1]), font, 2.5, (255, 255, 255), 2)# cv2.rectangle(img, (point[0], point[1]), (point[-2], point[-1]), color=(255, 255, 255), thickness=3)# cv2.imwrite('img_rect_second_bbx.jpg', img)cluster_third = get_cluster(second_bboxs, x_dis=50, y1_dis=400)print('===cluster=====')print(len(cluster_third))third_bboxs = fix_bbox_point(cluster_third)# debugfor i, point in enumerate(third_bboxs):font = cv2.FONT_HERSHEY_SIMPLEXcv2.putText(img, str(i), (point[0], point[1]), font, 2.5, (255, 255, 255), 2)cv2.rectangle(img, (point[0], point[1]), (point[-2], point[-1]), color=(255, 255, 255), thickness=3)cv2.imwrite('img_rect_third_bbx.jpg', img)
合并前：
合并后：
