找到两个向量,并找出它们之间的角度

时间:2019-11-30 20:32:24

标签: python algorithm math geometry box2d

我有一辆车辆和一个代表行人的点,我能够找到向量a和b(请参见下图),然后尝试计算它们之间的角度

我使用已知的公式cos(角度)=幅度(vector_a)*幅度(vector_a))/ dot_product(vector_a,vector_b),但是当我计算余弦时,它比一个大得多,因此无法计算角度...我在做什么错了?

    def detect_pedestrian(car,pedestrian):
        #Find a vector which connects pedestrian with a car
        pedestrian_position = pedestrian.body.position
        car_position = car.body.position

        #Find vectors a and b
        car_vector = Box2D.b2Body.__GetTransform(car.body).R.GetYAxis()
        pedestrian_vector = pedestrian_position - car_position

        #Calculate magnitudes of both vector
        pedestrian_vector_magnitude = np.linalg.norm(pedestrian_vector)
        car_vector_magnitude = np.linalg.norm(car_vector)

        #Normalize both vectors
        pedestrian_vector_normalized = pedestrian_vector/pedestrian_vector_magnitude
        car_vector_normalized = car_vector/car_vector_magnitude

        #Calculate angle
        magnitudes = pedestrian_vector_magnitude * car_vector_magnitude
        dot_product = b2Dot(pedestrian_vector_normalized, car_vector_normalized)
        cosine = magnitudes/dot_product
        arccos = np.arccos(cosine)

        print(cosine)

enter image description here

非常感谢您

1 个答案:

答案 0 :(得分:0)

您需要汽车的标题和行人相对于汽车的位置。如果您以世界坐标(即绝对位置)和汽车速度来确定行人和汽车的位置,则代码很简单:

import numpy as np

def detect_pedestrian(car_vector, pedestrian_vector, car_velocity):
    pedestrian_relative = pedestrian_vector - car_vector
    pedestrian_length = np.linalg.norm(pedestrian_relative)
    velocity_length = np.linalg.norm(car_velocity)
    dot_product = np.dot(pedestrian_relative, car_velocity)
    cosine = dot_product / (pedestrian_length * velocity_length)
    return(np.arccos(cosine))

car = np.array([5, 2])
pedestrian = np.array([2, 3])
v = np.array([0, 2])

angle_rad = detect_pedestrian(car, pedestrian, v)
print(f"Pedestrian angle: {np.degrees(angle_rad)}°")
# Pedestrian angle: 71.56505117707799°