diff --git a/predict.py b/predict.py index 3e7837c11..b6dadf0cd 100644 --- a/predict.py +++ b/predict.py @@ -14,4 +14,4 @@ probs = result.probs # 分类输出的Probs对象 obb = result.obb # OBB输出的Oriented boxes对象 result.show() # 显示到屏幕 - result.save(filename="result.jpg") # 保存到磁盘 \ No newline at end of file + result.save(filename="result.jpg") # 保存到磁盘 diff --git a/pyproject.toml b/pyproject.toml index eaf5956af..2d502be87 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -90,7 +90,7 @@ dev = [ export = [ "numpy<2.0.0", # TF 2.20 compatibility "onnx>=1.12.0; platform_system != 'Darwin'", # ONNX export - "onnx>=1.12.0,<1.18.0; platform_system == 'Darwin'", # TF inference hanging on MacOS + "onnx>=1.12.0,<1.20.0; platform_system == 'Darwin'", # TF inference hanging on MacOS "coremltools>=8.0; platform_system != 'Windows' and python_version <= '3.13'", # CoreML supported on macOS and Linux "scikit-learn>=1.3.2; platform_system != 'Windows' and python_version <= '3.13'", # CoreML k-means quantization "openvino>=2024.0.0", # OpenVINO export diff --git a/train.py b/train.py index b5e6c222f..6a8745d76 100644 --- a/train.py +++ b/train.py @@ -6,4 +6,4 @@ model = YOLO("yolo11n.yaml").load("yolo11n.pt") # 从YAML构建并转移权重 # 训练模型 -results = model.train(data="coco8.yaml", epochs=100, imgsz=640) \ No newline at end of file +results = model.train(data="coco8.yaml", epochs=100, imgsz=640)