Theo dõi và trực quan hóa thí nghiệm YOLO với Weights & Biases

Các mô hình phát hiện đối tượng như Ultralytics YOLO26 đã trở thành một phần không thể thiếu trong nhiều ứng dụng thị giác máy tính. Tuy nhiên, việc huấn luyện, đánh giá và triển khai các mô hình phức tạp này đặt ra nhiều thách thức. Việc theo dõi các chỉ số huấn luyện quan trọng, so sánh các phiên bản mô hình, phân tích hành vi mô hình và phát hiện các vấn đề đòi hỏi sự thiết lập công cụ và quản lý thí nghiệm đáng kể.

Hướng dẫn này giới thiệu khả năng tích hợp Ultralytics YOLO26 với Weights & Biases để nâng cao hiệu quả theo dõi thí nghiệm, lưu checkpoint mô hình và trực quan hóa hiệu suất mô hình. Hướng dẫn cũng bao gồm các chỉ dẫn để thiết lập tích hợp, huấn luyện, tinh chỉnh và trực quan hóa kết quả bằng cách sử dụng các tính năng tương tác của Weights & Biases.

Weights & Biases

Weights & Biases là một nền tảng MLOps tiên tiến được thiết kế để theo dõi, trực quan hóa và quản lý các thí nghiệm học máy. Nền tảng này có tính năng tự động ghi lại các chỉ số huấn luyện để đảm bảo tính tái lập đầy đủ của thí nghiệm, giao diện người dùng tương tác để phân tích dữ liệu hiệu quả và các công cụ quản lý mô hình tối ưu để triển khai trong nhiều môi trường khác nhau.

Huấn luyện YOLO26 với Weights & Biases

Bạn có thể sử dụng Weights & Biases để mang lại sự hiệu quả và tự động hóa cho quy trình huấn luyện YOLO26 của mình. Việc tích hợp cho phép bạn theo dõi các thí nghiệm, so sánh các mô hình và đưa ra các quyết định dựa trên dữ liệu để cải thiện các dự án thị giác máy tính của bạn.

Cài đặt

Để cài đặt các gói cần thiết, hãy chạy:

# Install the required packages for Ultralytics YOLO and Weights & Biases
pip install -U ultralytics wandb

# Enable W&B logging for Ultralytics
yolo settings wandb=True

Để biết hướng dẫn chi tiết và các phương pháp tốt nhất liên quan đến quy trình cài đặt, hãy đảm bảo xem Hướng dẫn cài đặt YOLO26 của chúng tôi. Trong khi cài đặt các gói cần thiết cho YOLO26, nếu bạn gặp bất kỳ khó khăn nào, hãy tham khảo Hướng dẫn các vấn đề thường gặp để tìm giải pháp và mẹo.

Cấu hình Weights & Biases

Sau khi cài đặt các gói cần thiết, bước tiếp theo là thiết lập môi trường Weights & Biases của bạn. Điều này bao gồm việc tạo tài khoản Weights & Biases và nhận khóa API cần thiết để kết nối mượt mà giữa môi trường phát triển của bạn và nền tảng W&B.

Bắt đầu bằng việc khởi tạo môi trường Weights & Biases trong workspace của bạn. Bạn có thể thực hiện việc này bằng cách chạy lệnh sau và làm theo các hướng dẫn hiển thị.

import wandb

# Initialize your Weights & Biases environment
wandb.login(key="YOUR_API_KEY")

Điều hướng đến trang ủy quyền Weights & Biases để tạo và lấy khóa API của bạn. Sử dụng khóa này khi được nhắc để xác thực môi trường của bạn với W&B.

Cách sử dụng: Huấn luyện YOLO26 với Weights & Biases

Trước khi đi sâu vào các hướng dẫn sử dụng cho việc huấn luyện mô hình YOLO26 với Weights & Biases, hãy đảm bảo xem qua phạm vi các mô hình YOLO26 do Ultralytics cung cấp. Điều này sẽ giúp bạn chọn mô hình phù hợp nhất cho các yêu cầu dự án của mình.

from ultralytics import YOLO

# Load a YOLO model
model = YOLO("yolo26n.pt")

# Train and Fine-Tune the Model
model.train(data="coco8.yaml", epochs=5, project="ultralytics", name="yolo26n")

Các đối số W&B

# Enable Weights & Biases logging
yolo settings wandb=True

# Disable Weights & Biases logging
yolo settings wandb=False

Hiểu về đầu ra

Sau khi chạy đoạn mã sử dụng ở trên, bạn có thể nhận được các kết quả đầu ra chính sau đây:

• Việc thiết lập một lần chạy mới với ID duy nhất, cho biết sự bắt đầu của quá trình huấn luyện.
• Tóm tắt ngắn gọn về cấu trúc của mô hình, bao gồm số lượng lớp và tham số.
• Cập nhật định kỳ về các chỉ số quan trọng như box loss, cls loss, dfl loss, precision, recall và điểm mAP trong mỗi epoch huấn luyện.
• Khi kết thúc huấn luyện, các chỉ số chi tiết bao gồm tốc độ suy luận của mô hình và các chỉ số độ chính xác tổng thể sẽ được hiển thị.
• Các liên kết đến bảng điều khiển Weights & Biases để phân tích và trực quan hóa chuyên sâu quá trình huấn luyện, cùng với thông tin về vị trí tệp log cục bộ.

Xem Bảng điều khiển Weights & Biases

Sau khi chạy đoạn mã sử dụng, bạn có thể truy cập bảng điều khiển Weights & Biases (W&B) thông qua liên kết được cung cấp trong kết quả đầu ra. Bảng điều khiển này cung cấp cái nhìn toàn diện về quá trình huấn luyện mô hình của bạn với YOLO26.

Các tính năng chính của Bảng điều khiển Weights & Biases

• Theo dõi chỉ số theo thời gian thực: Quan sát các chỉ số như loss, độ chính xác và điểm kiểm định khi chúng thay đổi trong quá trình huấn luyện, mang lại thông tin chi tiết tức thì cho việc tinh chỉnh mô hình. Xem cách các thí nghiệm được theo dõi bằng Weights & Biases.

• Tối ưu hóa siêu tham số (Hyperparameter): Weights & Biases hỗ trợ tinh chỉnh các tham số quan trọng như learning rate, batch size, và nhiều tham số khác, giúp nâng cao hiệu suất của YOLO26. Điều này giúp bạn tìm ra cấu hình tối ưu cho tập dữ liệu và tác vụ cụ thể của mình.

• Phân tích so sánh: Nền tảng cho phép so sánh song song các lần chạy huấn luyện khác nhau, rất cần thiết để đánh giá tác động của các cấu hình mô hình khác nhau và hiểu được những thay đổi nào giúp cải thiện hiệu suất.

• Trực quan hóa tiến độ huấn luyện: Các biểu diễn đồ họa của các chỉ số quan trọng cung cấp sự hiểu biết trực quan về hiệu suất của mô hình qua các epoch. Xem cách Weights & Biases giúp bạn trực quan hóa kết quả kiểm định.

• Giám sát tài nguyên: Theo dõi mức sử dụng CPU, GPU và bộ nhớ để tối ưu hóa hiệu quả của quy trình huấn luyện và xác định các điểm nghẽn tiềm ẩn trong quy trình làm việc của bạn.

• Quản lý Artifacts mô hình: Truy cập và chia sẻ các checkpoint mô hình, tạo điều kiện thuận lợi cho việc triển khai và cộng tác với các thành viên trong nhóm trên các dự án phức tạp.

• Xem kết quả suy luận với lớp phủ hình ảnh (Image Overlay): Trực quan hóa kết quả dự đoán trên hình ảnh bằng cách sử dụng các lớp phủ tương tác trong Weights & Biases, cung cấp cái nhìn rõ ràng và chi tiết về hiệu suất mô hình trên dữ liệu thực tế. Để biết thêm thông tin chi tiết, hãy xem khả năng phủ hình ảnh của Weights & Biases.

Bằng cách sử dụng các tính năng này, bạn có thể theo dõi, phân tích và tối ưu hóa hiệu quả quá trình huấn luyện mô hình YOLO26 của mình, đảm bảo hiệu suất và hiệu quả tốt nhất có thể cho các tác vụ phát hiện đối tượng của bạn.

Tóm tắt

Hướng dẫn này đã giúp bạn khám phá khả năng tích hợp Ultralytics YOLO với Weights & Biases. Nó minh họa khả năng của sự tích hợp này trong việc theo dõi và trực quan hóa kết quả huấn luyện và dự đoán mô hình một cách hiệu quả. Bằng cách tận dụng các tính năng mạnh mẽ của W&B, bạn có thể hợp lý hóa quy trình học máy của mình, đưa ra các quyết định dựa trên dữ liệu và cải thiện hiệu suất mô hình của bạn.

Để biết thêm chi tiết về cách sử dụng, hãy truy cập tài liệu chính thức của Weights & Biases hoặc khám phá bài thuyết trình của Soumik Rakshit từ YOLO VISION 2023 về sự tích hợp này.

Ngoài ra, hãy nhớ xem qua trang hướng dẫn tích hợp của Ultralytics để tìm hiểu thêm về các tích hợp thú vị khác như MLflow và Comet ML.

Câu hỏi thường gặp (FAQ)

Làm thế nào để tôi tích hợp Weights & Biases với Ultralytics YOLO26?

Để tích hợp Weights & Biases với Ultralytics YOLO26:

1. Cài đặt các gói cần thiết:

   pip install -U ultralytics wandb
   yolo settings wandb=True

2. Đăng nhập vào tài khoản Weights & Biases của bạn:

   import wandb
   
   wandb.login(key="YOUR_API_KEY")

3. Huấn luyện mô hình YOLO26 của bạn với tính năng ghi log W&B đã được bật:

   from ultralytics import YOLO
   
   model = YOLO("yolo26n.pt")
   model.train(data="coco8.yaml", epochs=5, project="ultralytics", name="yolo26n")

Điều này sẽ tự động ghi log các chỉ số, siêu tham số và artifacts mô hình vào dự án W&B của bạn.

Các tính năng chính của việc tích hợp Weights & Biases với YOLO26 là gì?

Các tính năng chính bao gồm:

• Theo dõi chỉ số theo thời gian thực trong quá trình huấn luyện
• Các công cụ tối ưu hóa siêu tham số
• Phân tích so sánh các lần chạy huấn luyện khác nhau
• Trực quan hóa tiến độ huấn luyện thông qua biểu đồ
• Giám sát tài nguyên (CPU, GPU, mức sử dụng bộ nhớ)
• Quản lý và chia sẻ artifacts mô hình
• Xem kết quả suy luận với các lớp phủ hình ảnh

Những tính năng này giúp theo dõi các thí nghiệm, tối ưu hóa các mô hình và cộng tác hiệu quả hơn trong các dự án YOLO26.

Làm thế nào để tôi có thể xem bảng điều khiển Weights & Biases cho quá trình huấn luyện YOLO26 của mình?

Sau khi chạy tập lệnh huấn luyện với tích hợp W&B:

1. Một liên kết đến bảng điều khiển W&B của bạn sẽ được cung cấp trong kết quả đầu ra của bảng điều khiển.
2. Nhấp vào liên kết hoặc truy cập wandb.ai và đăng nhập vào tài khoản của bạn.
3. Điều hướng đến dự án của bạn để xem các chỉ số chi tiết, hình ảnh trực quan hóa và dữ liệu hiệu suất mô hình.

Bảng điều khiển cung cấp thông tin chi tiết về quá trình huấn luyện mô hình của bạn, cho phép bạn phân tích và cải thiện các mô hình YOLO26 của mình một cách hiệu quả.

Tôi có thể tắt ghi log Weights & Biases cho quá trình huấn luyện YOLO26 không?

Có, bạn có thể tắt ghi log W&B bằng lệnh sau:

yolo settings wandb=False

Để bật lại tính năng ghi log, hãy sử dụng:

yolo settings wandb=True

Điều này cho phép bạn kiểm soát thời điểm bạn muốn sử dụng tính năng ghi log W&B mà không cần sửa đổi các tập lệnh huấn luyện của mình.

Weights & Biases giúp tối ưu hóa các mô hình YOLO26 như thế nào?

Weights & Biases giúp tối ưu hóa các mô hình YOLO26 bằng cách:

1. Cung cấp hình ảnh trực quan hóa chi tiết về các chỉ số huấn luyện
2. Cho phép so sánh dễ dàng giữa các phiên bản mô hình khác nhau
3. Cung cấp các công cụ để tinh chỉnh siêu tham số
4. Cho phép phân tích cộng tác về hiệu suất mô hình
5. Tạo điều kiện chia sẻ dễ dàng các artifacts mô hình và kết quả

Những tính năng này giúp các nhà nghiên cứu và nhà phát triển lặp lại nhanh hơn và đưa ra các quyết định dựa trên dữ liệu để cải thiện các mô hình YOLO26 của họ.