解释如下代码: for pic_id1 in range(1,N_pic+1): print('matching ' + set_name +': ' +str(pic_id1).zfill(5)) N_CHANGE = 0 for T_id in range(1,16,3): for H_id in range(2,5): FAIL_CORNER = 0 data_mat1 = read_data(input_file,pic_id1,T_id,H_id) search_list = range( max((pic_id1-10),1),pic_id1)+ range(pic_id1+1, min((pic_id1 + 16),N_pic + 1 ) ) for cor_ind in range(0,N_cor): row_cent1 = cor_row_center[cor_ind] col_cent1 = cor_col_center[cor_ind] img_corner = data_mat1[(row_cent1-N_pad): (row_cent1+N_pad+1), (col_cent1-N_pad): (col_cent1+N_pad+1) ] if ((len(np.unique(img_corner))) >2)&(np.sum(img_corner ==1)< 0.8*(N_pad2+1)**2) : for pic_id2 in search_list: data_mat2 = read_data(input_file,pic_id2,T_id,H_id) match_result = cv2_based(data_mat2,img_corner) if len(match_result[0]) ==1: row_cent2 = match_result[0][0]+ N_pad col_cent2 = match_result[1][0]+ N_pad N_LEF = min( row_cent1 , row_cent2) N_TOP = min( col_cent1, col_cent2 ) N_RIG = min( L_img-1-row_cent1 , L_img-1-row_cent2) N_BOT = min( L_img-1-col_cent1 , L_img-1-col_cent2) IMG_CHECK1 = data_mat1[(row_cent1-N_LEF): (row_cent1+N_RIG+1), (col_cent1-N_TOP): (col_cent1+N_BOT+1) ] IMG_CHECK2 = data_mat2[(row_cent2-N_LEF): (row_cent2+N_RIG+1), (col_cent2-N_TOP): (col_cent2+N_BOT+1) ] if np.array_equal(IMG_CHECK1,IMG_CHECK2) : check_row_N = IMG_CHECK1.shape[0] check_col_N = IMG_CHECK1.shape[1] if (check_col_Ncheck_row_N>=25): match_all.append( (pic_id1, row_cent1, col_cent1, pic_id2 , row_cent2, col_cent2) ) search_list.remove(pic_id2) else: FAIL_CORNER = FAIL_CORNER +1 N_CHANGE = N_CHANGE + 1 #%% break if less than 1 useless corners, or have detected more than 10 images from 60 if(FAIL_CORNER <= 1): break match_all_pd = pd.DataFrame(match_all,columns = ['pic_id1','row_id1','col_id1','pic_id2','row_id2','col_id2']) pd_add = pd.DataFrame(np.arange(1,N_pic+1), columns = ['pic_id1']) pd_add['pic_id2'] = pd_add['pic_id1'] pd_add['row_id1'] = 0 pd_add['row_id2'] = 0 pd_add['col_id1'] = 0 pd_add['col_id2'] = 0 match_all_pd = pd.concat([match_all_pd,pd_add]) match_all_pd.index = np.arange(len(match_all_pd))
时间: 2024-03-29 13:35:24 浏览: 43
这段代码是一个图像处理算法,通过对一组图片进行比对,找出其中相似的部分。首先通过一个循环对每一张图片进行处理,然后在其中嵌套两个循环,对每个图片的不同区域进行比对。在比对过程中,首先读取每个区域的像素值,然后将其与其他图片的对应区域进行比对。如果两个区域相似,则将其记录下来。最后将所有相似的区域的位置信息整理成一个表格返回。其中比对过程中用到了一些图像处理的函数,如np.unique()和cv2_based()。
阅读全文