Quellcode durchsuchen

fix arc_syart and arc_end

Your Name vor 4 Tagen
Ursprung
Commit
f60bf648d4
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. 1 1
      models/line_detect/trainer.py

+ 1 - 1
models/line_detect/trainer.py

@@ -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)]])
                     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)
                     max_angle_idx=np.argmax(polar_angles)