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
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) |