feeding

vision/angle_detector.py

@@ -1,12 +1,12 @@
 # vision/angle_detector.py
 import sys
 import os
-from vision.anger_caculate import predict_obb_best_angle
+from vision.obb_angle_model.obb_angle import detect_two_box_angle
 
 # 添加项目根目录到Python路径
-sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
+# sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
 
-def get_current_door_angle(model=None, image=None, image_path=None):
+def get_current_door_angle(model,image=None, image_path=None):
     """
     通过视觉系统获取当前出砼门角度
     :param model: 模型实例
@@ -16,10 +16,10 @@ def get_current_door_angle(model=None, image=None, image_path=None):
     """
     try:
         # 调用实际的角度检测函数
-        angle_deg, _ = predict_obb_best_angle(
-            model=model,
-            image=image,
-            image_path=image_path
+        angle_deg, _ = detect_two_box_angle(
+            model_path=model,
+            rgb_frame=image
+            # ,image_path=image_path
         )
         return angle_deg
     except Exception as e: