Bỏ qua nội dung

Bắt đầu nhanh

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. 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 để có phiên bản mới nhất. Docker có thể được sử dụng để thực thi gói trong một vùng chứa bị cô lậ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 ultralytics gói sử dụng pip hoặc cập nhật cài đặt hiện có bằng cách chạy pip install -U ultralytics. Ghé thăm Python Chỉ mục gói (PyPI) để biết thêm chi tiết về ultralytics bưu kiện: https://pypi.org/project/ultralytics/.

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 gói trực tiếp từ GitHub kho lưu trữ. Đ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 cài đặt công cụ dòng lệnh Git trên hệ thống của bạn. @main lệnh cài đặt main nhánh và có thể được sửa đổi thành nhánh khác, tức là @my-branch, hoặc xóa hoàn toàn để mặc định main chi nhánh.

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

Conda là trình quản lý gói thay thế cho pip, cũng có thể được sử dụng để cài đặt. Truy cập Anaconda để biết thêm chi tiết tại https://anaconda.org/conda-forge/ ultralytics . Ultralytics kho lưu trữ dữ liệu để cập nhật gói conda nằm tại https://github.com/conda-forge/ ultralytics -feedstock/ .

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 thực hành tốt nhất là cài đặt ultralytics, pytorchpytorch-cuda trong cùng một lệnh để cho phép trình quản lý gói conda giải quyết bất kỳ xung đột nào hoặc để cài đặt pytorch-cuda cuối cùng để cho phép nó ghi đè lên 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à là 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ữ nếu bạn quan tâm đến việc đóng góp vào quá trình 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 dễ dàng 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 và mượt mà 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 không chỉ tránh được sự phức tạp của việc cài đặt cục bộ mà còn được hưởng lợi từ việc tiếp cận môi trường làm việc đã được xác minh. Ultralytics cung cấp 5 hình ảnh Docker được hỗ trợ chính, mỗi hình ảnh được thiết kế để mang lại khả năng tương thích và hiệu quả cao cho các nền tảng và trường hợp sử dụng khác nhau:

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, cho phé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 với suy luận và môi trường không có GPU.
  • Dockerfile-jetson: Được thiết kế riêng cho NVIDIA Thiết bị 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.

Dưới đâ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à duy trì stdin mở, cho phép bạn 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, điều này 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 đổi /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 để có thể truy cập được.

Để sử dụng Docker nâng cao, bạn có thể tham khảo Hướng dẫn sử dụng Docker 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 nên 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ã. Bạn có thể chỉ cần chạy tất cả các tác vụ từ thiết bị đầu cuối với yolo lệnh. Kiểm tra CLI Hướng dẫn để tìm hiểu thêm về cách sử dụng YOLO từ dòng lệnh.

Ví dụ

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

yolo TASK MODE ARGS

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ột 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

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

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

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

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

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

Cảnh báo

Các đối số phải được thông qua như arg=val cặp, chia đôi bằng một = dấu và được phân cách bằng khoảng cách giữa các cặp. 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 --)

CLI Hướng dẫn

Sử dụng Ultralytics với Python

YOLO 'S Python giao diện cho phé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ế với mục đích đơn giản và dễ sử dụng, 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 trong các dự án của họ. Điều này làm cho YOLO 'S Python giao diện là một công cụ vô giá cho bất kỳ ai muốn 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ó trên một tập hợp xác thực và thậm chí xuất nó sang ONNX định dạng chỉ với một vài dòng mã. Kiểm tra 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 cung cấp một hệ thống quản lý cài đặt mạnh mẽ để cho phép kiểm soát chi tiết các thí nghiệm của bạn. Bằng cách sử dụng SettingsManager được đặt trong ultralytics.utils module, người dùng có thể dễ dàng truy cập và thay đổi cài đặt của họ. Những cài đặt này được lưu trữ trong tệp JSON trong thư mục cấu hình người dùng môi trường và có thể được xem hoặc sửa đổi trực tiếp 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

Để hiểu rõ hơn về cấu hình hiện tại của cài đặt, bạn có thể xem trực tiếp:

Xem cài đặt

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

from ultralytics import settings

# View all settings
print(settings)

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

Ngoài ra, giao diện dòng lệnh cho phép bạn kiểm tra cài đặt bằng một lệnh đơn giản:

yolo settings

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

Ultralytics cho phép người dùng dễ dàng sửa đổi cài đặt của họ. Thay đổi có thể được thực hiện theo những cách sau:

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

Trong vòng Python môi trường, gọi là update phương pháp trên settings phản đối việc thay đổi cài đặt của bạn:

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

Nếu bạn thích sử dụng giao diện dòng lệnh, các lệnh sau đây sẽ cho phép bạn sửa đổi cài đặt của mì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 cung cấp tổng quan về các thiết lập có sẵn để điều chỉnh trong Ultralytics . Mỗi thiết lập được trình bày kèm theo giá trị ví dụ, kiểu dữ liệu và mô tả ngắn gọn.

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 với Ultralytics Phiên bản pip )
datasets_dir '/path/to/datasets' str Thư mục nơi các tập dữ liệu được lưu trữ
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 thí nghiệm chạy được lưu trữ
uuid 'a1b2c3d4' str Mã định danh duy nhất cho các thiết lập hiện tại
sync True bool Có nên đồng bộ hóa phân tích và sự cố với HUB không
api_key '' str Ultralytics Khóa API HUB
clearml True bool Có nên sử dụng ghi nhật ký ClearML không
comet True bool Có nên sử dụng Comet ML để theo dõi và trực quan hóa thử nghiệm không
dvc True bool Có nên sử dụng DVC để theo dõi thử nghiệm và kiểm soát phiên bản không
hub True bool Có nên sử dụng tích hợp Ultralytics HUB không
mlflow True bool Có nên sử dụng MLFlow để theo dõi thử nghiệm không
neptune True bool Có nên sử dụng Neptune để theo dõi thử nghiệm không
raytune True bool Có nên sử dụng Ray Tune để điều chỉnh siêu tham số không
tensorboard True bool Có nên sử dụng TensorBoard để trực quan hóa không
wandb True bool Có nê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, cho phép nhắc tải xuống tiện ích mở rộng Ultralytics -Snippets .

Khi bạn điều hướng qua các dự án hoặc thử nghiệm của mình, hãy nhớ xem lại các cài đặt này để đảm bảo chúng được cấu hình tối ưu cho nhu cầu của bạn.

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, thực hiện lệnh sau:

pip install ultralytics

Đối với bản phát hành ổn định mới nhất, điều này sẽ cài đặt ultralytics gói trực tiếp từ Python Chỉ mục gói (PyPI). Để biết thêm chi tiết, hãy truy cập ultralytics gói trên PyPI.

Ngoài ra, bạn có thể cài đặt phiên bản phát triển mới nhất trực tiếp từ GitHub:

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

Đảm bảo rằng 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ó, bạn có thể cài đặt Ultralytics YOLO sử dụng conda bằng cách chạy:

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 và đảm bảo khả năng tương thích với các gói khác trong môi trường của bạn. Đối với CUDA môi trường, tốt nhất là cài đặt ultralytics, pytorch, Và pytorch-cuda đồng thời giải quyết mọi 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 truy cập 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 ?

Sử dụng Docker để chạy Ultralytics YOLO cung cấp một môi trường biệt lập và nhất quán, đảm bảo hiệu suất mượt mà trên các hệ thống khác nhau. Nó cũng loại bỏ sự phức tạp của cài đặt cục bộ. Hình ảnh Docker chính thức từ Ultralytics có sẵn trên Docker Hub , với các biến thể khác nhau được thiết kế riêng cho GPU , CPU , ARM64, NVIDIA Môi trường Jetson và Conda. Dưới đây là các lệnh để 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 hơn 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, hãy sử dụng các bước sau:

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

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

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

Các Ultralytics YOLO giao diện dòng lệnh ( 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 yêu cầu Python mã. Bạn có thể thực hiện các lệnh một dòng cho các tác vụ như đà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 cho yolo lệnh là:

yolo TASK MODE ARGS

Ví dụ, để đào tạo một mô hình phát hiện với các tham số được chỉ định:

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

Xem Hướng dẫn CLI đầy đủ để khám phá thêm nhiều lệnh và ví dụ sử dụng.

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

Bình luận