Cách đào tạo các mô hình tùy chỉnh của bạn với Ultralytics HUB

Nuvola Ladi

3 phút đọc

Ngày 31 tháng 5 năm 2024

Khám phá cách đào tạo các mô hình YOLO tùy chỉnh một cách dễ dàng với Ultralytics HUB. Hướng dẫn từng bước này vào nền tảng trực quan cung cấp sự giám sát vào tích hợp liền mạch, theo dõi thời gian thực và đào tạo đám mây.

Khi chúng tôi liên tục phấn đấu cho sự đổi mới do AI thúc đẩy, hãy tham gia cùng chúng tôi để xem xét kỹ hơn Ultralytics HUB , một nền tảng được thiết kế để đơn giản hóa việc đào tạo các mô hình YOLO tùy chỉnh của Ultralytics . Cho dù bạn là một nhà phát triển dày dạn kinh nghiệm hay người mới bắt đầu, Ultralytics HUB đều cung cấp trải nghiệm liền mạch để tạo và quản lý các mô hình thị giác máy tính. Hãy cùng khám phá cách đào tạo các mô hình YOLO tùy chỉnh của bạn chỉ bằng vài cú nhấp chuột.

Ultralytics HUB là gì?

Ultralytics HUB là một nền tảng toàn diện được thiết kế để xử lý các tập dữ liệu, dự án và mô hình của bạn, giúp việc đào tạo và triển khai các mô hình học máy dễ dàng hơn. 

Nó tích hợp với nhiều nền tảng khác nhau, hỗ trợ cập nhật theo thời gian thực thông qua ứng dụng di động (có sẵn trên cả Android và iOS) và liên tục phát triển với các tính năng mới. Để biết thêm thông tin chi tiết về kỹ thuật và quy trình làm việc từng bước, hãy xem bài viết Medium của chúng tôi về cách Đào tạo Mô hình Thị giác Máy tính trên Đám mây .

Bắt đầu với Ultralytics HUB

Giao diện thân thiện với người dùng

Điều đầu tiên bạn sẽ nhận thấy về Ultralytics HUB là giao diện trực quan của nó. Trang chủ cung cấp quyền truy cập dễ dàng vào các tập dữ liệu , dự án, mô hìnhtích hợp . Mặc dù vẫn đang trong giai đoạn beta, nền tảng này đã tự hào có một loạt các tính năng mạnh mẽ.

Bộ dữ liệu và mô hình

Ultralytics HUB được tải sẵn các tập dữ liệu chuẩn như VOC , COCO và Simpsons, thường được sử dụng để đánh giá chuẩn. Các tập dữ liệu này rất lớn, với COCO có 80 lớp và 140.000 hình ảnh. Tuy nhiên, nền tảng này cũng hỗ trợ các tập dữ liệu nhỏ hơn, khiến nó trở nên lý tưởng cho người dùng ở mọi cấp độ. Bạn có thể tải lên tập dữ liệu tùy chỉnh của mình hoặc kết nối với các công cụ bên ngoài như Roboflow để chú thích hình ảnh và chuẩn bị tập dữ liệu.

Tạo một dự án

Tạo một dự án mới trong Ultralytics HUB cực kỳ đơn giản. Sau đây là hướng dẫn từng bước để bạn bắt đầu:

  1. Tạo một dự án: Bắt đầu bằng cách đặt tên cho dự án của bạn. Đối với ví dụ này, hãy gọi là "Phát hiện đối tượng".
  2. Chọn một Bộ dữ liệu: Chọn một bộ dữ liệu từ các tùy chọn có sẵn. Để đơn giản, chúng tôi sẽ sử dụng bộ dữ liệu Simpsons với 14.000 hình ảnh.
  3. Chọn một mô hình: Chọn một mô hình YOLO. Ultralytics HUB hỗ trợ nhiều mô hình khác nhau, bao gồm YOLOv5YOLOv8 . Đối với mục đích của ví dụ này, chúng tôi sẽ sử dụng mô hình nano YOLOv8.
  4. Điều chỉnh siêu tham số: Điều chỉnh các thiết lập như số kỷ nguyên, kích thước hình ảnh và kích thước lô. Bạn cũng có thể chọn sử dụng GPU hay CPU và cấu hình tùy chọn lưu trữ đệm.

Đào tạo mô hình

Sau khi dự án của bạn được thiết lập, việc đào tạo mô hình chỉ cần một cú nhấp chuột. Ultralytics HUB cung cấp nhiều tùy chọn đào tạo:

  • Đào tạo tại chỗ: Cài đặt Ultralytics trên máy cục bộ của bạn và chạy tập lệnh đào tạo.
  • Google Colab : Mở sổ tay Google Colab được cấu hình sẵn với mã cần thiết. Tùy chọn này không yêu cầu mã hóa và chạy liền mạch trên đám mây.
  • HUB Cloud: Ultralytics HUB Cloud Training cung cấp giải pháp không cần mã để đào tạo các mô hình YOLO, hoàn hảo cho những người không phải là lập trình viên và chủ doanh nghiệp. Quy trình làm việc bao gồm tải lên tập dữ liệu, lựa chọn mô hình và thiết lập phiên bản đám mây, giúp dễ dàng tinh chỉnh các mô hình được đào tạo trước và xuất chúng cho nhiều ứng dụng khác nhau.

Để đào tạo mô hình trong Google Colab:

  1. Cài đặt Ultralytics: Chạy lệnh để cài đặt Ultralytics vào máy tính xách tay .
  2. Thiết lập và Khóa API: Cấu hình thiết lập và nhập khóa API của bạn.
  3. Bắt đầu đào tạo: Thực hiện các lệnh đào tạo và mô hình của bạn sẽ bắt đầu đào tạo.
__wf_reserved_thừa kế
Hình 1. Nicolai Nielsen phác thảo cách đào tạo mô hình trên Ultralytics HUB.

Theo dõi tiến độ đào tạo

Ultralytics HUB cung cấp tính năng theo dõi thời gian thực tiến trình đào tạo mô hình của bạn. Bạn có thể theo dõi các số liệu chính như độ chính xác, độ chính xác, độ thu hồi và hàm mất mát. Nền tảng này cũng trực quan hóa dữ liệu đào tạo, cho phép bạn xem mô hình của mình cải thiện như thế nào theo thời gian.

Các tính năng và tích hợp nâng cao

Đào tạo đám mây và tích hợp RoboFlow

Ultralytics HUB đã giới thiệu đào tạo đám mây , cho phép người dùng đào tạo mô hình trực tiếp trên đám mây mà không cần bất kỳ thiết lập cục bộ nào. Nền tảng này cũng tích hợp với Roboflow để chú thích hình ảnh và chuẩn bị tập dữ liệu. Tích hợp này hợp lý hóa quy trình làm việc từ chuẩn bị dữ liệu đến đào tạo mô hình.

Ứng dụng di động

Ứng dụng Ultralytics HUB mang đến khả năng phát hiện đối tượng theo thời gian thực cho thiết bị di động của bạn. Sử dụng các mô hình được đào tạo trước, ứng dụng có thể phát hiện các đối tượng từ tập dữ liệu COCO với độ chính xác cao, chạy ở tốc độ 30 khung hình/giây trên iPhone 14 Pro. Tính năng này hoàn hảo để thử nghiệm các mô hình trong các tình huống thực tế và chứng minh khả năng của các mô hình được đào tạo của bạn.

Kết thúc

Ultralytics HUB là một công cụ thay đổi cuộc chơi trong thế giới thị giác máy tính, giúp việc đào tạo các mô hình YOLO tùy chỉnh trở nên dễ dàng hơn bao giờ hết. Giao diện thân thiện với người dùng, các tính năng mạnh mẽ và tích hợp liền mạch định vị nó là một công cụ không thể thiếu đối với các nhà phát triển và nhà nghiên cứu. Cho dù bạn đang muốn đào tạo các mô hình cục bộ, trên đám mây hay khi đang di chuyển bằng ứng dụng di động, Ultralytics HUB đều có thể đáp ứng nhu cầu của bạn.

Vậy còn chần chừ gì nữa? Hãy tham gia Ultralytics HUB ngay hôm nay và khai phá tiềm năng của các dự án học máy của bạn chỉ bằng vài cú nhấp chuột!

Hãy cùng xây dựng tương lai
của AI cùng nhau!

Bắt đầu hành trình của bạn với tương lai của máy học

Bắt đầu miễn phí
Liên kết đã được sao chép vào clipboard