You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

52 lines
1.5 KiB

import numpy as np
import matplotlib.pyplot as plt
import calc_way
import get_data
import calc_slope_line
import cv2
import model
model = model.Model()
alpha = model.alpha
beta = model.beta
img_path = r'C:\Users\Administrator\Desktop\BYD\error\20250620\20250620\pic1.jpg'
def draw_lines(img_path):
image = cv2.imread(img_path)
if image is None:
print("Error: 无法读取图像,请检查路径!")
exit()
x_zero, y_zero = calc_way.calc_zeros_xto0(862+60)
print(x_zero)
print(y_zero)
point1 = (int(x_zero[0]),960-int(y_zero[0]))
point2 = (int(x_zero[-1]),960-int(y_zero[-1]))
cv2.line(image, point1, point2, (0, 0, 255), 1)
x_zero, y_zero = calc_way.calc_zeros_yto0(0)
print(x_zero)
print(y_zero)
point1 = (int(x_zero[0]),960-int(y_zero[0]))
point2 = (int(x_zero[-1]),960-int(y_zero[-1]))
cv2.line(image, point1, point2, (0, 0, 255), 1)
x_zero, y_zero = calc_way.calc_zeros_xto0(862+300)
print(x_zero)
print(y_zero)
point1 = (int(x_zero[0]),960-int(y_zero[0]))
point2 = (int(x_zero[-1]),960-int(y_zero[-1]))
cv2.line(image, point1, point2, (0, 0, 255), 1)
x_zero, y_zero = calc_way.calc_zeros_yto0(0-480)
print(x_zero)
print(y_zero)
point1 = (int(x_zero[0]),960-int(y_zero[0]))
point2 = (int(x_zero[-1]),960-int(y_zero[-1]))
cv2.line(image, point1, point2, (0, 0, 255), 1)
cv2.imshow('image', image)
cv2.waitKey(0)
cv2.destroyAllWindows()
draw_lines(img_path)