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..50550f96f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -94,7 +94,7 @@ export = [ "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 - "tensorflow>=2.0.0,<=2.19.0", # TF bug https://github.com/ultralytics/ultralytics/issues/5161 + "tensorflow>=2.0.0,<=2.20.0", # TF bug https://github.com/ultralytics/ultralytics/issues/5161 "tensorflowjs>=2.0.0", # TF.js export, automatically installs tensorflow "tensorstore>=0.1.63; platform_machine == 'aarch64' and python_version >= '3.9'", # for TF Raspberry Pi exports "h5py!=3.11.0; platform_machine == 'aarch64'", # fix h5py build issues due to missing aarch64 wheels in 3.11 release 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)