Bỏ qua nội dung

Cài đặt Ultralytics

Ultralytics cung cấp nhiều phương pháp cài đặt khác nhau, bao gồm pip, conda và Docker. Bạn có thể cài đặt YOLO thông qua ultralytics gói pip cho bản phát hành ổn định mới nhất hoặc bằng cách sao chép Ultralytics Kho lưu trữ GitHub đối với phiên bản mới nhất. Docker cũng là một tùy chọn để chạy gói trong một vùng chứa bị cô lập, giúp tránh cài đặt cục bộ.



Đồng hồ: Ultralytics YOLO Hướng dẫn bắt đầu nhanh

Cài đặt

PyPI - Python Phiên bản

Cài đặt hoặc cập nhật ultralytics gói sử dụng pip bằng cách chạy pip install -U ultralytics. Để biết thêm chi tiết về ultralytics gói, ghé thăm Python Chỉ mục gói (PyPI).

PyPI - Phiên bản Tải xuống

# Install the ultralytics package from PyPI
pip install ultralytics

Bạn cũng có thể cài đặt ultralytics trực tiếp từ Ultralytics Kho lưu trữ GitHub. Điều này có thể hữu ích nếu bạn muốn có phiên bản phát triển mới nhất. Đảm bảo bạn đã cài đặt công cụ dòng lệnh Git, sau đó chạy:

# Install the ultralytics package from GitHub
pip install git+https://github.com/ultralytics/ultralytics.git@main

Conda có thể được sử dụng như một trình quản lý gói thay thế cho pip. Để biết thêm chi tiết, hãy truy cập Anaconda . Ultralytics kho lưu trữ dữ liệu để cập nhật gói conda có sẵn tại GitHub .

Phiên bản Conda Tải xuống Conda Công thức làm Conda Nền tảng Conda

# Install the ultralytics package using conda
conda install -c conda-forge ultralytics

Ghi chú

Nếu bạn đang cài đặt trong một CUDA môi trường, cách tốt nhất là cài đặt ultralytics, pytorch, Và pytorch-cuda trong cùng một lệnh. Điều này cho phép trình quản lý gói conda giải quyết mọi xung đột. Ngoài ra, hãy cài đặt pytorch-cuda cuối cùng để ghi đè CPU -cụ thể pytorch gói nếu cần thiết.

# Install all packages together using conda
conda install -c pytorch -c nvidia -c conda-forge pytorch torchvision pytorch-cuda=11.8 ultralytics

Hình ảnh Conda Docker

Ultralytics Hình ảnh Conda Docker cũng có sẵn từ DockerHub. Những hình ảnh này dựa trên Miniconda3 và cung cấp một cách đơn giản để bắt đầu sử dụng ultralytics trong môi trường 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

Sao chép Ultralytics Kho lưu trữ GitHub nếu bạn quan tâm đến việc đóng góp cho sự phát triển hoặc muốn thử nghiệm với mã nguồn mới nhất. Sau khi sao chép, hãy điều hướng vào thư mục và cài đặt gói ở chế độ có thể chỉnh sửa -e sử dụng pip.

GitHub cam kết cuối cùng Hoạt động cam kết GitHub

# 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 .

Sử dụng Docker để thực hiện ultralytics đóng gói trong một thùng chứa riêng biệt, đảm bảo hiệu suất nhất quán trong nhiều môi trường khác nhau. Bằng cách chọn một trong những ultralytics hình ảnh từ Trung tâm Docker, bạn tránh được sự phức tạp của việc cài đặt cục bộ và có quyền truy cập vào môi trường làm việc đã được xác minh. Ultralytics cung cấp năm hình ảnh Docker được hỗ trợ chính, mỗi hình ảnh được thiết kế để có khả năng tương thích và hiệu quả cao:

Phiên bản hình ảnh Docker Docker kéo

  • Tệp Docker: GPU hình ảnh được đề xuất để đào tạo.
  • Dockerfile-arm64: Được tối ưu hóa cho kiến trúc ARM64, phù hợp để triển khai trên các thiết bị như Raspberry Pi và các nền tảng khác dựa trên ARM64.
  • Dockerfile- cpu : Dựa trên Ubuntu CPU -Phiên bản duy nhất, phù hợp cho suy luận và môi trường không có GPU.
  • Dockerfile-jetson: Được thiết kế riêng cho các thiết bị NVIDIA Jetson , tích hợp GPU hỗ trợ được tối ưu hóa cho các nền tảng này.
  • Dockerfile- python : Hình ảnh tối thiểu chỉ với Python và các phụ thuộc cần thiết, lý tưởng cho các ứng dụng nhẹ và phát triển.
  • Dockerfile-conda: Dựa trên Miniconda3 với cài đặt conda của ultralytics bưu kiện.

Sau đây là các lệnh để lấy hình ảnh mới nhất và thực thi nó:

# 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

Lệnh trên khởi tạo một vùng chứa Docker với phiên bản mới nhất ultralytics hình ảnh. Các -it cờ gán một pseudo-TTY và giữ stdin mở, cho phép tương tác với container. --ipc=host cờ đặt không gian tên IPC (Giao tiếp giữa các quy trình) cho máy chủ, điều này rất cần thiết để chia sẻ bộ nhớ giữa các quy trình. --gpus all cờ cho phép truy cập vào tất cả các GPU có sẵn bên trong vùng chứa, rất quan trọng đối với các tác vụ yêu cầu GPU tính toán.

Lưu ý: Để làm việc với các tệp trên máy cục bộ của bạn trong vùng chứa, hãy sử dụng ổ đĩa Docker để gắn thư mục cục bộ vào vùng chứa:

# 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

Thay thế /path/on/host với đường dẫn thư mục trên máy cục bộ của bạn và /path/in/container với đường dẫn mong muốn bên trong vùng chứa Docker.

Để sử dụng Docker nâng cao, hãy khám phá Hướng dẫn Docker của Ultralytics .

Xem ultralytics pyproject.toml tệp để biết danh sách các phụ thuộc. Lưu ý rằng tất cả các ví dụ trên đều cài đặt tất cả các phụ thuộc bắt buộc.

Mẹo

Yêu cầu PyTorch thay đổi tùy theo hệ điều hành và CUDA yêu cầu, vì vậy hãy cài đặt PyTorch Đầu tiên hãy làm theo hướng dẫn tại PyTorch .

PyTorch Hướng dẫn cài đặt

Sử dụng Ultralytics với CLI

Các Ultralytics giao diện dòng lệnh ( CLI ) cho phép các lệnh đơn giản trên một dòng mà không cần Python môi trường. CLI không yêu cầu tùy chỉnh hoặc Python mã; chạy tất cả các tác vụ từ thiết bị đầu cuối với yolo lệnh. Để biết thêm về cách sử dụng YOLO từ dòng lệnh, hãy xem CLI Hướng dẫn.

Ví dụ

Ultralytics yolo lệnh sử dụng cú pháp sau:

yolo TASK MODE ARGS
- TASK (tùy chọn) là một trong (phát hiện, phân khúc, phân loại, tư thế, obb) - MODE (bắt buộc) là một trong (xe lửa, giá trị, dự đoán, xuất khẩu, theo dõi, chuẩn mực) - ARGS (tùy chọn) là arg=value cặp như imgsz=640 ghi đè lên các giá trị mặc định.

Xem tất cả ARGS trong đầy đủ Hướng dẫn cấu hình hoặc với yolo cfg CLI yêu cầu.

Đào tạo mô hình phát hiện trong 10 kỷ nguyên với tốc độ học ban đầu là 0,01:

yolo train data=coco8.yaml model=yolo11n.pt epochs=10 lr0=0.01

Dự đoán video YouTube bằng mô hình phân đoạn được đào tạo trước ở kích thước hình ảnh 320:

yolo predict model=yolo11n-seg.pt source='https://youtu.be/LNwODJXcvt4' imgsz=320

Xác thực mô hình phát hiện được đào tạo trước với kích thước lô là 1 và kích thước hình ảnh là 640:

yolo val model=yolo11n.pt data=coco8.yaml batch=1 imgsz=640

Xuất mô hình phân loại YOLOv11n sang ONNX định dạng với kích thước hình ảnh là 224x128 (không yêu cầu NHIỆM VỤ):

yolo export model=yolo11n-cls.pt format=onnx imgsz=224,128

Đếm các đối tượng trong video hoặc luồng trực tiếp bằng cách sử dụng YOLO11 :

yolo solutions count show=True

yolo solutions count source="path/to/video.mp4" # specify video file path

Theo dõi các bài tập luyện tập bằng cách sử dụng YOLO11 tư thế người mẫu:

yolo solutions workout show=True

yolo solutions workout source="path/to/video.mp4" # specify video file path

# Use keypoints for ab-workouts
yolo solutions workout kpts="[5, 11, 13]" # left side
yolo solutions workout kpts="[6, 12, 14]" # right side

Sử dụng YOLO11 để đếm các đối tượng trong một hàng đợi hoặc khu vực được chỉ định:

yolo solutions queue show=True

yolo solutions queue source="path/to/video.mp4" # specify video file path

yolo solutions queue region="[(20, 400), (1080, 400), (1080, 360), (20, 360)]" # configure queue coordinates

Thực hiện phát hiện đối tượng, phân đoạn thể hiện hoặc ước tính tư thế trong trình duyệt web bằng Streamlit :

yolo solutions inference

yolo solutions inference model="path/to/model.pt" # use model fine-tuned with Ultralytics Python package

Chạy các lệnh đặc biệt để xem phiên bản, xem cài đặt, chạy kiểm tra và nhiều hơn nữa:

yolo help
yolo checks
yolo version
yolo settings
yolo copy-cfg
yolo cfg
yolo solutions help

Cảnh báo

Các đối số phải được thông qua như arg=value cặp, chia đôi bằng một = dấu và phân cách bằng khoảng trắng. Không sử dụng -- tiền tố đối số hoặc dấu phẩy , giữa các lập luận.

  • yolo predict model=yolo11n.pt imgsz=640 conf=0.25
  • yolo predict model yolo11n.pt imgsz 640 conf 0.25 ❌ (thiếu =)
  • yolo predict model=yolo11n.pt, imgsz=640, conf=0.25 ❌ (không sử dụng ,)
  • yolo predict --model yolo11n.pt --imgsz 640 --conf 0.25 ❌ (không sử dụng --)
  • yolo solution model=yolo11n.pt imgsz=640 conf=0.25 ❌ (sử dụng solutions, không solution)

CLI Hướng dẫn

Sử dụng Ultralytics với Python

Các Ultralytics YOLO Python giao diện cung cấp tích hợp liền mạch vào Python dự án, giúp dễ dàng tải, chạy và xử lý đầu ra của mô hình. Được thiết kế để đơn giản hóa, Python giao diện cho phép người dùng nhanh chóng triển khai phát hiện đối tượng , phân đoạn và phân loại. Điều này làm cho YOLO Python giao diện là một công cụ vô giá để kết hợp các chức năng này vào Python dự án.

Ví dụ, người dùng có thể tải một mô hình, đào tạo nó, đánh giá hiệu suất của nó và xuất nó sang ONNX định dạng chỉ với một vài dòng mã. Khám phá Hướng dẫn Python để tìm hiểu thêm về cách sử dụng YOLO trong bạn Python dự án.

Ví dụ

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")

Python Hướng dẫn

Ultralytics Cài đặt

Các Ultralytics thư viện bao gồm một SettingsManager để kiểm soát chi tiết các thí nghiệm, cho phép người dùng truy cập và sửa đổi cài đặt dễ dàng. Được lưu trữ trong tệp JSON trong thư mục cấu hình người dùng của môi trường, các cài đặt này có thể được xem hoặc sửa đổi trong Python môi trường hoặc thông qua Giao diện dòng lệnh ( CLI ).

Kiểm tra cài đặt

Để xem cấu hình hiện tại của cài đặt của bạn:

Xem cài đặt

Sử dụng Python để xem cài đặt của bạn bằng cách nhập settings đối tượng từ ultralytics module. In và trả về các thiết lập bằng các lệnh sau:

from ultralytics import settings

# View all settings
print(settings)

# Return a specific setting
value = settings["runs_dir"]

Giao diện dòng lệnh cho phép bạn kiểm tra cài đặt của mình bằng:

yolo settings

Sửa đổi Cài đặt

Ultralytics giúp bạn dễ dàng thay đổi cài đặt theo những cách sau:

Cập nhật cài đặt

TRONG Python , sử dụng update phương pháp trên settings sự vật:

from ultralytics import settings

# Update a setting
settings.update({"runs_dir": "/path/to/runs"})

# Update multiple settings
settings.update({"runs_dir": "/path/to/runs", "tensorboard": False})

# Reset settings to default values
settings.reset()

Để sửa đổi cài đặt bằng giao diện dòng lệnh:

# Update a setting
yolo settings runs_dir='/path/to/runs'

# Update multiple settings
yolo settings runs_dir='/path/to/runs' tensorboard=False

# Reset settings to default values
yolo settings reset

Hiểu về Cài đặt

Bảng dưới đây tổng quan về các thiết lập có thể điều chỉnh trong Ultralytics , bao gồm các giá trị mẫu, kiểu dữ liệu và mô tả.

Tên Giá trị ví dụ Kiểu dữ liệu Sự miêu tả
settings_version '0.0.4' str Ultralytics phiên bản cài đặt (khác biệt với Ultralytics Phiên bản pip )
datasets_dir '/path/to/datasets' str Thư mục nơi lưu trữ các tập dữ liệu
weights_dir '/path/to/weights' str Thư mục nơi lưu trữ trọng số mô hình
runs_dir '/path/to/runs' str Thư mục nơi lưu trữ các lần chạy thử nghiệm
uuid 'a1b2c3d4' str Mã định danh duy nhất cho các thiết lập hiện tại
sync True bool Tùy chọn đồng bộ hóa phân tích và sự cố với Ultralytics HUB
api_key '' str Khóa API Ultralytics HUB
clearml True bool Tùy chọn sử dụng ghi nhật ký ClearML
comet True bool Tùy chọn sử dụng Comet ML để theo dõi và trực quan hóa thử nghiệm
dvc True bool Tùy chọn sử dụng DVC để theo dõi thử nghiệm và kiểm soát phiên bản
hub True bool Tùy chọn sử dụng tích hợp Ultralytics HUB
mlflow True bool Tùy chọn sử dụng MLFlow để theo dõi thử nghiệm
neptune True bool Tùy chọn sử dụng Neptune để theo dõi thử nghiệm
raytune True bool Tùy chọn sử dụng Ray Tune để điều chỉnh siêu tham số
tensorboard True bool Tùy chọn sử dụng TensorBoard để trực quan hóa
wandb True bool Tùy chọn sử dụng ghi nhật ký Weights & Biases
vscode_msg True bool Khi phát hiện thiết bị đầu cuối VS Code, sẽ hiển thị lời nhắc tải xuống tiện ích mở rộng Ultralytics -Snippets .

Hãy xem lại các thiết lập này khi bạn thực hiện các dự án hoặc thử nghiệm để đảm bảo cấu hình tối ưu.

CÂU HỎI THƯỜNG GẶP

Làm thế nào để tôi cài đặt Ultralytics sử dụng pip?

Cài đặt Ultralytics với pip sử dụng:

pip install ultralytics

Điều này cài đặt bản phát hành ổn định mới nhất của ultralytics gói từ PyPI. Để cài đặt phiên bản phát triển trực tiếp từ GitHub:

pip install git+https://github.com/ultralytics/ultralytics.git

Đảm bảo công cụ dòng lệnh Git được cài đặt trên hệ thống của bạn.

Tôi có thể cài đặt không? Ultralytics YOLO sử dụng conda?

Có, cài đặt Ultralytics YOLO sử dụng conda với:

conda install -c conda-forge ultralytics

Phương pháp này là một giải pháp thay thế tuyệt vời cho pip, đảm bảo khả năng tương thích với các gói khác. Đối với CUDA môi trường, cài đặt ultralytics, pytorch, Và pytorch-cuda cùng nhau giải quyết xung đột:

conda install -c pytorch -c nvidia -c conda-forge pytorch torchvision pytorch-cuda=11.8 ultralytics

Để biết thêm hướng dẫn, hãy xem hướng dẫn bắt đầu nhanh của Conda .

Những lợi thế của việc sử dụng Docker để chạy là gì? Ultralytics YOLO ?

Docker cung cấp một môi trường riêng biệt, nhất quán cho Ultralytics YOLO , đảm bảo hiệu suất mượt mà trên các hệ thống và tránh sự phức tạp trong cài đặt cục bộ. Hình ảnh Docker chính thức có sẵn trên Docker Hub , với các biến thể cho GPU , CPU , ARM64, NVIDIA Jetson và Conda. Để kéo và chạy hình ảnh mới nhất:

# 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

Để biết hướng dẫn chi tiết về Docker, hãy xem hướng dẫn bắt đầu nhanh về Docker .

Làm thế nào để tôi sao chép Ultralytics kho lưu trữ để phát triển?

Sao chép Ultralytics kho lưu trữ và thiết lập môi trường phát triển với:

# 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 .

Điều này cho phép đóng góp vào dự án hoặc thử nghiệm với mã nguồn mới nhất. Để biết chi tiết, hãy truy cập kho lưu trữ GitHub của Ultralytics .

Tại sao tôi nên sử dụng Ultralytics YOLO CLI ?

Các Ultralytics YOLO CLI đơn giản hóa việc chạy các tác vụ phát hiện đối tượng mà không cần Python mã, cho phép các lệnh một dòng để đào tạo, xác thực và dự đoán trực tiếp từ thiết bị đầu cuối của bạn. Cú pháp cơ bản là:

yolo TASK MODE ARGS

Ví dụ, để đào tạo một mô hình phát hiện:

yolo train data=coco8.yaml model=yolo11n.pt epochs=10 lr0=0.01

Khám phá thêm các lệnh và ví dụ sử dụng trong Hướng dẫn CLI đầy đủ.

📅 Được tạo ra cách đây 1 năm ✏️ Đã cập nhật cách đây 1 ngày

Bình luận