Cách huấn luyện các mô hình tùy chỉnh của bạn với Ultralytics HUB
Khám phá cách huấn luyện 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ề nền tảng trực quan cung cấp cái nhìn tổng quan về tích hợp liền mạch, theo dõi thời gian thực và huấn luyện trên đám mây.
Trong hành trình không ngừng nỗ lực hướng tới đổi mới dựa trên AI, hãy cùng chúng tôi tìm hiểu sâu hơn về Ultralytics HUB, một nền tảng được thiết kế để đơn giản hóa việc huấn luyện các model Ultralytics YOLO tùy chỉnh. Cho dù bạn là một lập trình viên dày dạn kinh nghiệm hay người mới bắt đầu, Ultralytics HUB đều mang đến trải nghiệm liền mạch để tạo và quản lý các model thị giác máy tính. Hãy cùng khám phá cách huấn luyện các model YOLO tùy chỉnh của bạn chỉ trong vài cú nhấp chuột.
Link to this sectionUltralytics 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à model của bạn, giúp việc huấn luyện và triển khai các model machine learning trở nên dễ dàng hơn.
Nền tảng này tích hợp với nhiều công cụ khác nhau, hỗ trợ cập nhật 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 chuyên sâu về kỹ thuật và quy trình làm việc từng bước, hãy xem bài viết trên Medium của chúng tôi về cách Huấn luyện Model Thị giác Máy tính trên Đám mây.
Link to this sectionBắt đầu với Ultralytics HUB#
Link to this sectionGiao 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 khả năng truy cập dễ dàng vào tập dữ liệu, các dự án, model và các tích hợp. Mặc dù vẫn đang trong giai đoạn beta, nền tảng này đã sở hữu một loạt các tính năng mạnh mẽ.
Link to this sectionTập dữ liệu và model#
Ultralytics HUB được tải sẵn các tập dữ liệu tiêu chuẩn như VOC, COCO và Simpsons, vốn thường được sử dụng để làm benchmark. Các tập dữ liệu này khá đồ sộ, với COCO bao gồm 80 lớp và 140.000 hình ảnh. Tuy nhiên, nền tảng 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 riêng mình hoặc kết nối với các công cụ bên ngoài như Roboflow để gán nhãn hình ảnh và chuẩn bị tập dữ liệu.
Link to this sectionTạo dự án#
Việc tạo một dự án mới trong Ultralytics HUB vô cùng đơn giản. Dưới đây là hướng dẫn từng bước để bạn bắt đầu:
- Tạo Dự án: Bắt đầu bằng cách đặt tên cho dự án của bạn. Trong ví dụ này, hãy gọi nó là "Object Detection."
- Chọn Tập dữ liệu: Chọn một tập dữ liệu từ các tùy chọn có sẵn. Để đơn giản, chúng ta sẽ sử dụng tập dữ liệu Simpsons với 14.000 hình ảnh.
- Chọn Model: Chọn một model YOLO. Ultralytics HUB hỗ trợ nhiều model khác nhau, bao gồm YOLOv5 và YOLOv8. Với mục đích của ví dụ này, chúng ta sẽ chọn model YOLOv8 nano.
- Điều chỉnh Siêu tham số: Tùy chỉnh các thiết lập như số lượng epoch, kích thước hình ảnh và batch size. Bạn cũng có thể chọn sử dụng GPU hay CPU và cấu hình các tùy chọn bộ nhớ đệm (caching).
Link to this sectionHuấn luyện model#
Sau khi dự án đã được thiết lập, việc huấn luyện model chỉ cách bạn một cú nhấp chuột. Ultralytics HUB cung cấp nhiều tùy chọn huấn luyện:
- Huấn luyện cục bộ: Cài đặt Ultralytics trên máy cục bộ của bạn và chạy script huấn luyện.
- Google Colab: Mở một notebook Google Colab đã được cấu hình sẵn với mã nguồn cần thiết. Tùy chọn này không yêu cầu lập trình và chạy liền mạch trên đám mây.
- HUB Cloud: Huấn luyện trên Đám mây Ultralytics HUB cung cấp giải pháp không cần viết mã (no-code) để huấn luyện các model YOLO, rất phù hợp cho những người không chuyên về lập trình và các chủ doanh nghiệp. Quy trình làm việc bao gồm tải lên tập dữ liệu, chọn model và thiết lập cloud instance, giúp việc tinh chỉnh các model đã huấn luyện trước và xuất chúng cho các ứng dụng khác nhau trở nên dễ dàng.
Để huấn luyện model trong Google Colab:
- Cài đặt Ultralytics: Chạy lệnh để cài đặt Ultralytics trong notebook.
- Thiết lập và API Key: Cấu hình thiết lập và nhập API key của bạn.
- Bắt đầu Huấn luyện: Thực thi các lệnh huấn luyện, và model của bạn sẽ bắt đầu quá trình huấn luyện.
Hình 1. Nicolai Nielsen mô tả cách huấn luyện model trên Ultralytics HUB.
Link to this sectionTheo dõi tiến độ huấn luyện#
Ultralytics HUB cung cấp khả năng theo dõi thời gian thực về tiến độ huấn luyện model của bạn. Bạn có thể giám sát các chỉ số quan trọng như độ chính xác (accuracy), precision, recall và các hàm loss. Nền tảng cũng trực quan hóa dữ liệu huấn luyện, cho phép bạn thấy cách model của mình cải thiện theo thời gian.
Link to this sectionTính năng nâng cao và tích hợp#
Link to this sectionHuấn luyện trên đám mây và tích hợp với Roboflow#
Ultralytics HUB đã giới thiệu huấn luyện trên đám mây, cho phép người dùng huấn luyện model 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 để gán nhãn hình ảnh và chuẩn bị tập dữ liệu. Sự tích hợp này giúp hợp lý hóa quy trình từ khâu chuẩn bị dữ liệu đến khi huấn luyện model.
Link to this sectionỨng dụng di động#
Ứng dụng Ultralytics HUB mang khả năng phát hiện đối tượng thời gian thực đến thiết bị di động của bạn. Sử dụng các model đã được huấn luyện 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 rất hoàn hảo để kiểm thử model trong các kịch bản thực tế và trình diễn khả năng của các model mà bạn đã huấn luyện.
Link to this sectionTóm tắt#
Ultralytics HUB là một bước ngoặt trong thế giới thị giác máy tính, giúp việc huấn luyện các model YOLO tùy chỉnh trở nên dễ dàng hơn bao giờ hết. Giao diện thân thiện, các tính năng mạnh mẽ và khả năng tích hợp liền mạch biến nó thành một công cụ không thể thiếu đối với các lập trình viên và nhà nghiên cứu. Cho dù bạn muốn huấn luyện model cục bộ, trên đám mây hay khi đang di chuyển với ứng dụng di động, Ultralytics HUB đều đáp ứng tốt mọi nhu cầu.
Vậy còn chần chờ gì nữa? Hãy khám phá Ultralytics HUB ngay hôm nay và mở khóa tiềm năng của các dự án machine learning của bạn chỉ với vài cú nhấp chuột!
