|
@@ -108,7 +108,7 @@ def draw_ellipses_on_image(image, masks_pred, threshold=0.5, color=(0, 255, 0),
|
|
|
rot_matrix=np.array([[np.cos(angle_red), -np.sin(angle_red)],[-np.sin(angle_red), np.cos(angle_red)]])
|
|
rot_matrix=np.array([[np.cos(angle_red), -np.sin(angle_red)],[-np.sin(angle_red), np.cos(angle_red)]])
|
|
|
points_rotated=np.dot(points_centered, rot_matrix.T)
|
|
points_rotated=np.dot(points_centered, rot_matrix.T)
|
|
|
|
|
|
|
|
- polar_angles=np.arctan2(points_rotated[:,1],points_rotated[:,1])
|
|
|
|
|
|
|
+ polar_angles=np.arctan2(points_rotated[:,1],points_rotated[:,0])
|
|
|
|
|
|
|
|
min_angle_idx=np.argmin(polar_angles)
|
|
min_angle_idx=np.argmin(polar_angles)
|
|
|
max_angle_idx=np.argmax(polar_angles)
|
|
max_angle_idx=np.argmax(polar_angles)
|