安装Ultralytics
Ultralytics 提供多种安装方法,包括 pip、conda 和 Docker。您可以通过 ultralytics
pip 软件包,或通过克隆 Ultralytics GitHub 存储库 获取最新版本。Docker 也是在隔离容器中运行软件包的一个选项,可避免本地安装。
观看: Ultralytics YOLO 快速入门指南
安装
安装或更新 ultralytics
运行 pip install -U ultralytics
.有关 ultralytics
软件包,请访问 Python 软件包索引 (PyPI).
您还可以安装 ultralytics
直接从 Ultralytics GitHub 存储库.如果您需要最新的开发版本,这将非常有用。确保已安装 Git 命令行工具,然后运行
Conda 可作为 pip 的替代软件包管理器。更多详情,请访问Anaconda。用于更新 conda 软件包的Ultralytics 原料库位于GitHub。
备注
如果在CUDA 环境中安装,最佳做法是安装 ultralytics
, pytorch
和 pytorch-cuda
在同一命令中。这允许 conda 软件包管理器解决任何冲突。或者,安装 pytorch-cuda
最后覆盖CPU pytorch
如有必要,请将该程序包添加到"... "中。
Conda Docker 映像
Ultralytics Conda Docker 映像还可从 DockerHub.这些图像基于 迷你世界3 并为开始使用 ultralytics
在 Conda 环境中。
# Set image name as a variable
t=ultralytics/ultralytics:latest-conda
# Pull the latest ultralytics image from Docker Hub
sudo docker pull $t
# Run the ultralytics image in a container with GPU support
sudo docker run -it --ipc=host --gpus all $t # all GPUs
sudo docker run -it --ipc=host --gpus '"device=2,3"' $t # specify GPUs
克隆 Ultralytics GitHub 存储库 如果你有兴趣参与开发,或希望尝试最新的源代码,请复制到"...... "目录。克隆后,进入目录,以可编辑模式安装软件包 -e
使用 pip。
使用 Docker 执行 ultralytics
软件包在一个隔离的容器中,确保在各种环境下性能一致。通过选择官方 ultralytics
图片来自 Docker Hub使用 Ultralytics 的 Docker 镜像,您可以避免本地安装的复杂性,并获得经过验证的工作环境。Ultralytics 提供五种主要支持的 Docker 镜像,每一种都具有很高的兼容性和效率:
- Dockerfile:推荐用于培训的GPU 镜像。
- Dockerfile-arm64:针对 ARM64 架构进行了优化,适合在 Raspberry Pi 等设备和其他基于 ARM64 的平台上部署。
- Dockerfile-cpu:基于 Ubuntu 的CPU 版本,适用于推理和没有 GPU 的环境。
- Dockerfile-jetson:专为NVIDIA Jetson设备定制,集成了针对这些平台优化的GPU 支持。
- Dockerfile-python :仅包含Python 和必要依赖项的最小镜像,是轻量级应用和开发的理想选择。
- Dockerfile-conda: 基于 Miniconda3 和 conda 安装的
ultralytics
包装
以下是获取并执行最新映像的命令:
# Set image name as a variable
t=ultralytics/ultralytics:latest
# Pull the latest ultralytics image from Docker Hub
sudo docker pull $t
# Run the ultralytics image in a container with GPU support
sudo docker run -it --ipc=host --gpus all $t # all GPUs
sudo docker run -it --ipc=host --gpus '"device=2,3"' $t # specify GPUs
上述命令使用最新的 ultralytics
图像图像 -it
标记分配一个伪 TTY 并保持 stdin 开启,允许与容器进行交互。标记 --ipc=host
标记将 IPC(进程间通信)命名空间设置为主机命名空间,这对进程间共享内存至关重要。进程间通信 --gpus all
flag 可以访问容器内所有可用的 GPU,这对需要GPU 计算的任务至关重要。
注意:要在容器中处理本地计算机上的文件,请使用 Docker 卷将本地目录挂载到容器中:
# Mount local directory to a directory inside the container
sudo docker run -it --ipc=host --gpus all -v /path/on/host:/path/in/container $t
更换 /path/on/host
的目录路径,以及 /path/in/container
与 Docker 容器内的所需路径一致。
有关 Docker 的高级用法,请参阅《Ultralytics Docker 指南》。
参见 ultralytics
pyproject.toml 文件中的依赖项列表。请注意,上述所有示例都安装了所有必需的依赖项。
将Ultralytics 与CLI
Ultralytics 命令行界面CLI)允许用户在无需Python 环境的情况下执行简单的单行命令。CLI 不需要定制或编写Python 代码,只需在终端使用 yolo
命令。有关从命令行使用YOLO 的更多信息,请参阅 CLI 指南.
示例
Ultralytics yolo
命令使用以下语法:
TASK
(可选)是 (侦测, 分部, 纶, 姿势, obb)
- MODE
(必填)是 (训练, 缬氨酸, 预言, 出口, 赛道, 基准)
- ARGS
(可选)为 arg=value
如 imgsz=640
覆盖默认值。
查看全部 ARGS
在充分 配置指南 或使用 yolo cfg
CLI 指挥。
训练检测模型 10 个历元,初始学习率为 0.01:
在图像大小为 320 时,使用预训练的分割模型预测 YouTube 视频:
将 YOLOv11n 分类模型导出为ONNX 格式,图像大小为 224x128(无需 TASK):
使用YOLO11 计数视频或直播流中的物体:
使用YOLO11 姿势模型监控锻炼情况:
使用YOLO11 对指定队列或区域中的对象进行计数:
使用Streamlit 在网络浏览器中执行对象检测、实例分割或姿态估计:
警告
参数必须以 arg=value
对,用等号分割 =
符号,并用空格分隔。请勿使用 --
参数 ,
参数之间。
yolo predict model=yolo11n.pt imgsz=640 conf=0.25
✅yolo predict model yolo11n.pt imgsz 640 conf 0.25
❌(缺失=
)yolo predict model=yolo11n.pt, imgsz=640, conf=0.25
❌(不要使用,
)yolo predict --model yolo11n.pt --imgsz 640 --conf 0.25
❌(不要使用--
)yolo solution model=yolo11n.pt imgsz=640 conf=0.25
❌(使用solutions
不是solution
)
将Ultralytics 与Python
Ultralytics YOLO Python 界面可无缝集成到Python 项目中,从而轻松加载、运行和处理模型输出。Python 界面设计简单,用户可以快速实现对象检测、分割和分类。这使得YOLO Python 接口成为将这些功能集成到Python 项目中的宝贵工具。
例如,用户只需几行代码就可以加载模型、训练模型、评估其性能并将其导出为ONNX 格式。查看Python 指南,了解在Python 项目中使用YOLO 的更多信息。
示例
from ultralytics import YOLO
# Create a new YOLO model from scratch
model = YOLO("yolo11n.yaml")
# Load a pretrained YOLO model (recommended for training)
model = YOLO("yolo11n.pt")
# Train the model using the 'coco8.yaml' dataset for 3 epochs
results = model.train(data="coco8.yaml", epochs=3)
# Evaluate the model's performance on the validation set
results = model.val()
# Perform object detection on an image using the model
results = model("https://ultralytics.com/images/bus.jpg")
# Export the model to ONNX format
success = model.export(format="onnx")
Ultralytics 设置
Ultralytics 库包括一个 SettingsManager
用于对实验进行细粒度控制,允许用户轻松访问和修改设置。这些设置存储在环境用户配置目录下的 JSON 文件中,可在Python 环境中或通过命令行界面CLI)查看或修改。
检查设置
查看设置的当前配置:
查看设置
修改设置
Ultralytics 可以通过以下方式轻松修改设置:
更新设置
在Python 中,使用 update
上的 settings
反对
了解设置
下表概述了Ultralytics 中的可调整设置,包括示例值、数据类型和说明。
名称 | 示例值 | 数据类型 | 说明 |
---|---|---|---|
settings_version |
'0.0.4' |
str |
Ultralytics 设置版本(有别于Ultralytics pip版本) |
datasets_dir |
'/path/to/datasets' |
str |
存储数据集的目录 |
weights_dir |
'/path/to/weights' |
str |
存储模型权重的目录 |
runs_dir |
'/path/to/runs' |
str |
存储实验运行的目录 |
uuid |
'a1b2c3d4' |
str |
当前设置的唯一标识符 |
sync |
True |
bool |
将分析和崩溃同步到Ultralytics HUB的选项 |
api_key |
'' |
str |
Ultralytics HUBAPI 密钥 |
clearml |
True |
bool |
使用以下选项 ClearML记录 |
comet |
True |
bool |
使用Comet ML进行实验跟踪和可视化的选项 |
dvc |
True |
bool |
可选择使用DVC 进行实验跟踪和版本控制 |
hub |
True |
bool |
使用Ultralytics HUB集成选项 |
mlflow |
True |
bool |
使用MLFlow进行实验跟踪的选项 |
neptune |
True |
bool |
可选择使用 Neptune进行实验跟踪 |
raytune |
True |
bool |
使用Ray Tune进行超参数调整的选项 |
tensorboard |
True |
bool |
使用TensorBoard进行可视化的选项 |
wandb |
True |
bool |
使用选项 Weights & Biases记录 |
vscode_msg |
True |
bool |
检测到 VS Code 终端时,启用下载Ultralytics扩展的提示。 |
随着项目或实验的进展,请重新检查这些设置,以确保最佳配置。
常见问题
如何使用 pip 安装Ultralytics ?
使用 pip 安装Ultralytics :
这将安装最新稳定版的 ultralytics
从 PyPI.要直接从 GitHub 安装开发版本,请访问
确保系统已安装 Git 命令行工具。
可以使用 conda 安装Ultralytics YOLO 吗?
是的,使用 conda 安装Ultralytics YOLO :
这种方法是 pip 的最佳替代方案,可确保与其他软件包兼容。对于CUDA 环境,安装 ultralytics
, pytorch
和 pytorch-cuda
共同解决冲突:
有关更多说明,请参阅Conda 快速入门指南。
使用 Docker 运行Ultralytics YOLO 有哪些优势?
Docker 为Ultralytics YOLO 提供了一个隔离、一致的环境,确保跨系统的流畅性能,避免本地安装的复杂性。官方 Docker 映像可在Docker Hub 上获取,并有针对GPU、CPU、ARM64、NVIDIA Jetson 和 Conda 的不同版本。提取并运行最新镜像:
# Pull the latest ultralytics image from Docker Hub
sudo docker pull ultralytics/ultralytics:latest
# Run the ultralytics image in a container with GPU support
sudo docker run -it --ipc=host --gpus all ultralytics/ultralytics:latest
有关 Docker 的详细说明,请参阅Docker 快速入门指南。
如何克隆Ultralytics 仓库进行开发?
克隆Ultralytics 软件仓库,并使用以下工具建立开发环境:
# Clone the ultralytics repository
git clone https://github.com/ultralytics/ultralytics
# Navigate to the cloned directory
cd ultralytics
# Install the package in editable mode for development
pip install -e .
这样就可以为项目做出贡献,或使用最新的源代码进行实验。有关详细信息,请访问Ultralytics GitHub 存储库。
为什么要使用Ultralytics YOLO CLI ?
Ultralytics YOLO CLI 简化了运行对象检测任务的过程,无需Python 代码,可直接从终端使用单行命令进行训练、验证和预测。基本语法如下
例如,训练检测模型:
在完整的CLI 指南中探索更多命令和使用示例。