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..b5c6248d8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -88,7 +88,7 @@ dev = [ "minijinja>=2.0.0", # render docs macros without mkdocs-macros-plugin ] export = [ - "numpy<2.0.0", # TF 2.20 compatibility + "numpy<3.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 "coremltools>=8.0; platform_system != 'Windows' and python_version <= '3.13'", # CoreML supported on macOS and Linux 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)