import numpy as np import matplotlib.pyplot as plt import calc_way import get_data import calc_slope_line import cv2 import model import os model = model.Model() alpha = model.alpha beta = model.beta img_path = r'C:\Users\Administrator\Desktop\BYD\20250520\frame__9600_2_yolo.jpg' txt_name = "C:\\Users\\Administrator\\Desktop\\BYD\\20250520\\frame_9600_2.jpg_zuobiao.txt" output_folder = 'C:\\Users\\Administrator\\Desktop\\BYD\\Visual measurement\\pic\\9600' """ 计算图像中识别到的路沿距离车辆坐标原点的距离、高度信息 参数：保存数据的txt文件路径 返回值： """ # def vs_measurement(txt_name): # # os.makedirs(output_folder, exist_ok=True) # x_bot, y_bot, x_top, y_top = get_data.get_data(txt_name) # # x_bot = np.array(x_bot) # y_bot = np.array(y_bot) # x_top = np.array(x_top) # y_top = np.array(y_top) # slope_top, intercept_top ,r2 = calc_slope_line.linear_regression(x_top,y_top) # # x_zero, y_zero = calc_way.calc_zeros() # x_zero = np.array(x_zero) # y_zero = np.array(y_zero) # slope_zero, intercept_zero, r2_zero = calc_slope_line.linear_regression(x_zero, y_zero) # # # # 绘制原始数据 # # plt.scatter(x_top,y_top, color='blue', label='orgin') # # # # # 绘制拟合线 # # y_pred = slope_top * x_top + intercept_top # # plt.plot(x_top, y_pred, color='red', label='fix') # # # # plt.xlabel('X') # # plt.ylabel('Y') # # plt.legend() # # plt.show() # # Z = 0 # Y = 0 # max_Zw = 0 # max_Zw_index = 0 # min_Zw = 0 # min_Zw_index = 0 # max_Yw = 0 # max_Yw_index = 0 # min_Yw = 0 # min_Yw_index = 0 # # for i in range(len(x_bot)): # image = cv2.imread(img_path) # 默认读取BGR格式 # if image is None: # print("Error: 无法读取图像，请检查路径！") # exit() # # point1 = (int(x_zero[0]), int(960 - y_zero[0])) # point2 = (int(x_zero[-1]), int(960 - y_zero[-1])) # cv2.line(image, point1, point2, (0, 0, 255), 2) # # k = calc_slope_line.get_k(alpha,beta,x_bot[i],y_bot[i]) # b = calc_slope_line.get_b(x_bot[i],y_bot[i],k) # x = (intercept_top - b) / (k - slope_top) # y = k * x + b # Zw = calc_way.calc_height(x_bot[i],y_bot[i], x, y, alpha, beta) # Xw, Yw = calc_way.calc_distance(x_bot[i],y_bot[i], alpha, beta) # if i == 0: # max_Zw = Zw # min_Zw = Zw # max_Yw = Yw # min_Yw = Yw # else: # if Zw > max_Zw: # max_Zw = Zw # max_Zw_index = i # if Zw < min_Zw: # min_Zw = Zw # min_Zw_index = i # if Yw > max_Yw: # max_Yw = Yw # max_Yw_index = i # if Yw < min_Yw: # min_Yw = Yw # min_Yw_index = i # point1 = (x_bot[i], 960-y_bot[i]) # point2 = (int(x), 960-int(y)) # cv2.line(image, point1, point2, (0, 255, 0), 1) # text = f"Xw,Yw:{int(Xw),int(Yw)},Zw:{int(Zw)}" # position = (int(x), 960-int(y)) # font = cv2.FONT_HERSHEY_SIMPLEX # font_scale = 1.0 # color = (0, 0, 255) # thickness = 2 # cv2.putText(image, text, position, font, font_scale, color, thickness, cv2.LINE_AA) # output_path = os.path.join(output_folder, f'{i+1}.jpg') # cv2.imwrite(output_path, image) # Z = Z + Zw # Y = Y + Yw # file_path = os.path.join(output_folder, 'data.txt') # # with open(file_path, 'w', encoding='utf-8') as file: # file.write("该图片数据如下\n") # file.write(f"路沿平均高度为：{}\n") # file.write(f"最大高度为图{max_Zw_index+1},高度为：{max_Zw}\n") # file.write(f"最小高度为图{min_Zw_index+1},高度为：{min_Zw}\n") # file.write(f"路沿距离车辆平均距离为：{-Y/len(x_bot)}\n") # file.write(f"最远距离为图{min_Yw_index + 1},距离为：{min_Yw}\n") # file.write(f"最近距离为图{max_Yw_index + 1},距离为：{max_Yw}\n") def vs_measurement(txt_name): # os.makedirs(output_folder, exist_ok=True) x_bot, y_bot, x_top, y_top = get_data.get_data(txt_name) x_bot = np.array(x_bot) y_bot = np.array(y_bot) x_top = np.array(x_top) y_top = np.array(y_top) slope_top, intercept_top ,r2 = calc_slope_line.linear_regression(x_top,y_top) Z = 0 Y = 0 max_Zw = 0 max_Zw_index = 0 min_Zw = 0 min_Zw_index = 0 max_Yw = 0 max_Yw_index = 0 min_Yw = 0 min_Yw_index = 0 for i in range(len(x_bot)): k = calc_slope_line.get_k(alpha,beta,x_bot[i],y_bot[i]) b = calc_slope_line.get_b(x_bot[i],y_bot[i],k) x = (intercept_top - b) / (k - slope_top) y = k * x + b Zw = calc_way.calc_height(x_bot[i],y_bot[i], x, y, alpha, beta) Xw, Yw = calc_way.calc_distance(x_bot[i],y_bot[i], alpha, beta) if i == 0: max_Zw = Zw min_Zw = Zw max_Yw = Yw min_Yw = Yw else: if Zw > max_Zw: max_Zw = Zw max_Zw_index = i if Zw < min_Zw: min_Zw = Zw min_Zw_index = i if Yw > max_Yw: max_Yw = Yw max_Yw_index = i if Yw < min_Yw: min_Yw = Yw min_Yw_index = i Z = Z + Zw Y = Y + Yw return Z/len(x_bot),-max_Yw if __name__ == '__main__': vs_measurement(txt_name)