xiongyi/Feeding_control_system
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

完成中交三航控制系统

Browse Source
This commit is contained in:
cdeyw
2025-09-26 11:50:42 +08:00
parent 2e779ea8e1
commit 486645a3aa
31 changed files with 32 additions and 1685 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
Fedding.py
View File

@ -12,8 +12,8 @@ import sys
sys.path.append(os.path.join(os.path.dirname(__file__), 'src', 'vision'))
# 导入视觉处理模块
from src.vision.anger_caculate import predict_obb_best_angle
from src.vision.resize_tuili_image_main import classify_image_weighted, load_global_rois, crop_and_resize, YOLO
from vision.anger_caculate import predict_obb_best_angle
from vision.resize_tuili_image_main import classify_image_weighted, load_global_rois, crop_and_resize, YOLO
class FeedingControlSystem:
@ -124,10 +124,10 @@ class FeedingControlSystem:
self.alignment_check_interval = 0.5 # 对齐检查间隔(秒)
# 模型路径配置
self.angle_model_path = "src/vision/angle.pt"
self.overflow_model_path = "src/vision/overflow.pt"
self.alignment_model_path = "src/vision/alig.pt" # 模具车对齐检测模型
self.roi_file_path = "src/vision/roi_coordinates/1_rois.txt"
self.angle_model_path = "vision/models/angle.pt"
self.overflow_model_path = "vision/models/overflow.pt"
self.alignment_model_path = "vision/models/alig.pt" # 模具车对齐检测模型
self.roi_file_path = "vision/roi_coordinates/1_rois.txt"
# 模型实例
self.angle_model = None # 夹角检测模型实例
@ -147,7 +147,6 @@ class FeedingControlSystem:
def set_camera_config(self, camera_type="ip", ip=None, port=None, username=None, password=None, channel=1):
"""
设置摄像头配置
:param camera_type: 摄像头类型 "usb" 或 "ip"
:param ip: 网络摄像头IP地址
:param port: 网络摄像头端口
:param username: 网络摄像头用户名