Theo dõi thí nghiệm với Neptune

Neptune là kho lưu trữ siêu dữ liệu cho MLOps, được xây dựng cho các nhóm thực hiện nhiều thử nghiệm. Nó cung cấp cho bạn một nơi duy nhất để ghi lại, lưu trữ, hiển thị, sắp xếp, so sánh và truy vấn tất cả siêu dữ liệu xây dựng mô hình của bạn.

Ultralytics YOLO11 tích hợp với Neptune để hợp lý hóa việc theo dõi thử nghiệm . Tích hợp này cho phép bạn tự động ghi lại số liệu đào tạo, trực quan hóa dự đoán mô hình và lưu trữ các hiện vật mô hình mà không cần viết mã ghi nhật ký tùy chỉnh.

Các tính năng chính

• Ghi nhật ký tự động : Tự động ghi nhật ký các số liệu đào tạo quan trọng như mất hộp, mất phân loại và mAP .
• Hình ảnh trực quan : Xem các khảm đào tạo và dự đoán xác thực trực tiếp trong Neptune bảng điều khiển.
• Kiểm tra mô hình: Tải lên và kiểm soát phiên bản trọng số mô hình được đào tạo của bạn (best.pt) tự động vào cuối quá trình đào tạo.
• Theo dõi siêu tham số : Ghi lại tất cả các tham số cấu hình để đảm bảo khả năng tái tạo đầy đủ các thí nghiệm của bạn.
• Biểu đồ tương tác : Trực quan hóa ma trận nhầm lẫn và đường cong thu hồi chính xác để phân tích hiệu suất mô hình.

Cài đặt

Để sử dụng Neptune với Ultralytics , bạn sẽ cần phải cài đặt neptune gói khách hàng cùng với ultralytics.

# Install the required packages
pip install ultralytics neptune

# Enable Neptune integration in Ultralytics settings
yolo settings neptune=True

Cấu hình

Trước khi bắt đầu đào tạo, bạn cần kết nối môi trường địa phương của bạn với Neptune dự án. Bạn sẽ cần Mã thông báo API và Tên dự án từ Neptune bảng điều khiển.

1. Nhận chứng chỉ của bạn

1. Đăng nhập vào Neptune .ai .
2. Tạo một dự án mới (hoặc chọn một dự án hiện có).
3. Vào menu người dùng và lấy Mã thông báo API của bạn.

2. Thiết lập biến môi trường

Cách an toàn nhất để xử lý thông tin xác thực là thông qua các biến môi trường. Lưu ý rằng Ultralytics Neptune gọi lại đọc YOLO project lập luận và không sử dụng NEPTUNE_PROJECT. Vượt qua đầy đủ Neptune sên (ví dụ, workspace/name) thông qua project= trong lệnh huấn luyện của bạn; nếu không Neptune sẽ cố gắng sử dụng mặc định theo nghĩa đen "Ultralytics" và cuộc chạy sẽ thất bại.

export NEPTUNE_API_TOKEN="your_long_api_token_here" # required

$Env:NEPTUNE_API_TOKEN = "your_long_api_token_here"  # required

import os

os.environ["NEPTUNE_API_TOKEN"] = "your_long_api_token_here"
os.environ["NEPTUNE_PROJECT"] = "your_workspace/your_project"

Cách sử dụng

Sau khi cấu hình xong, bạn có thể bắt đầu đào tạo YOLO11 các mô hình. Các Neptune tích hợp hoạt động tự động khi neptune gói đã được cài đặt và tích hợp được bật trong cài đặt.

Ví dụ đào tạo

from ultralytics import YOLO

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

# Train the model
# Pass the Neptune project slug as the 'project' argument (workspace/name)
results = model.train(data="coco8.yaml", epochs=10, project="my-workspace/my-project", name="experiment-1")

# Train via CLI
# project must be the Neptune slug (workspace/name); otherwise run creation will fail
yolo train data=coco8.yaml epochs=10 project=my-workspace/my-project name=experiment-1

Hiểu về Tích hợp

Sơ đồ sau đây minh họa cách Ultralytics Đường ống đào tạo tương tác với Neptune để ghi lại nhiều hiện vật và số liệu khác nhau.

graph LR
    A[YOLO Training Loop] --> B{Neptune Callback}
    B -->|Log Scalars| C[Loss, mAP, LR]
    B -->|Log Images| D[Mosaics, Preds]
    B -->|Log Artifacts| E[Model Weights]
    B -->|Log Metadata| F[Hyperparameters]

    C --> G[Neptune Server]
    D --> G
    E --> G
    F --> G

    G --> H[Neptune Web Dashboard]

Logged là gì?

Khi bạn chạy lệnh đào tạo, Neptune tích hợp tự động nắm bắt cấu trúc dữ liệu sau trong lần chạy của bạn:

1. Cấu hình/Siêu tham số : Tất cả các đối số đào tạo (epoch, lr0, trình tối ưu hóa, v.v.) đều được ghi vào phần Cấu hình.
2. Cấu hình/Mô hình : Kiến trúc và định nghĩa mô hình.
3. Các độ đo:
   • Huấn luyện: box_loss, cls_loss, dfl_loss, lr (tốc độ học tập).
   • Các độ đo: precision, recall, mAP50, mAP50-95.
4. Ảnh:
   • Mosaic: Các đợt đào tạo hiển thị dữ liệu tăng cường.
   • Validation: Nhãn thực tế và dự đoán mô hình trên dữ liệu xác thực.
   • Plots: Ma trận nhầm lẫn, đường cong Độ chính xác-Thu hồi.
5. Trọng lượng: Mô hình được đào tạo cuối cùng (best.pt) được tải lên weights thư mục trong Neptune chạy.

Sử dụng nâng cao

Tổ chức chạy

Bạn có thể sử dụng tiêu chuẩn Ultralytics project và name các đối số để tổ chức các lần chạy của bạn trong Neptune .

• project: Phải là Neptune slug dự án workspace/name; đây là những gì lệnh gọi lại truyền tới neptune.init_run.
• name: Hoạt động như mã định danh cho lần chạy cụ thể.

Ghi nhật ký tùy chỉnh

Nếu bạn cần ghi lại các số liệu tùy chỉnh bổ sung cùng với việc ghi nhật ký tự động, bạn có thể truy cập Neptune chạy phiên bản. Lưu ý rằng bạn sẽ cần phải sửa đổi logic huấn luyện hoặc tạo một lệnh gọi lại tùy chỉnh để truy cập đối tượng chạy cụ thể, vì Ultralytics tích hợp xử lý vòng đời chạy nội bộ.

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

Làm thế nào để tôi vô hiệu hóa Neptune khai thác gỗ?

Nếu bạn đã cài đặt neptune nhưng muốn vô hiệu hóa việc ghi nhật ký cho một phiên cụ thể hoặc toàn cầu, bạn có thể sửa đổi YOLO cài đặt.

# Disable Neptune integration
yolo settings neptune=False

Hình ảnh của tôi không tải lên được. Có vấn đề gì vậy?

Đảm bảo rằng mạng của bạn cho phép kết nối tới Neptune máy chủ của. Ngoài ra, việc ghi nhật ký hình ảnh thường diễn ra theo các khoảng thời gian cụ thể (ví dụ: kết thúc kỷ nguyên hoặc kết thúc đào tạo). Nếu bạn ngắt quá trình đào tạo sớm bằng cách sử dụng Ctrl+C, một số hiện vật cuối cùng như ma trận nhầm lẫn hoặc trọng số mô hình tốt nhất có thể không được tải lên.

Tôi có thể đăng nhập vào một Neptune chạy ID?

Tích hợp hiện tại sẽ tự động tạo một lần chạy mới cho mỗi phiên đào tạo. Để tiếp tục ghi nhật ký vào một lần chạy hiện có, thông thường bạn sẽ cần xử lý Neptune khởi tạo thủ công trong Python mã, nằm ngoài phạm vi tích hợp tự động. Tuy nhiên, Ultralytics hỗ trợ việc tiếp tục đào tạo tại địa phương, điều này sẽ tạo ra một bước tiến mới Neptune ĐẾN track các kỷ nguyên được tiếp tục.

Tôi có thể tìm thấy trọng số mô hình ở đâu? Neptune ?

Trong bạn Neptune bảng điều khiển, điều hướng đến Hiện vật hoặc Tất cả siêu dữ liệu phần. Bạn sẽ tìm thấy một weights thư mục chứa của bạn best.pt tệp mà bạn có thể tải xuống để triển khai.

📅 Được tạo 0 ngày trước ✏️ Được cập nhật 0 ngày trước

Bình luận