RTDETRv2 so với YOLOv9#

Lĩnh vực thị giác máy tính đã chứng kiến sự khác biệt thú vị trong triết lý kiến trúc, chủ yếu là giữa Mạng thần kinh tích chập (CNN) và các mô hình dựa trên Transformer. Khi so sánh RTDETRv2 và YOLOv9, các nhà phát triển thực chất đang đánh giá sự cân bằng giữa cơ chế chú ý toàn cục (global attention) và thông tin gradient có thể lập trình (programmable gradient information). Cả hai mô hình đều đại diện cho đỉnh cao của các mô hình tương ứng, đẩy lùi ranh giới của công nghệ phát hiện đối tượng thời gian thực.

Giới thiệu về các Model#

RTDETRv2: Real-Time Detection Transformer#

Được phát triển bởi các nhà nghiên cứu tại Baidu, RTDETRv2 được xây dựng dựa trên RT-DETR gốc bằng cách giới thiệu "Bag-of-Freebies" để tăng cường cho Real-Time Detection Transformer cơ sở. Nó giải quyết nút thắt cổ chai truyền thống của các transformer—tốc độ suy luận (inference speed)—giúp chúng trở nên khả thi cho các ứng dụng thời gian thực.

• Tác giả: Wenyu Lv, Yian Zhao, Qinyao Chang, Kui Huang, Guanzhong Wang, và Yi Liu
• Tổ chức: Baidu
• Ngày: 24-07-2024
• Links: Arxiv, GitHub

Một đặc điểm xác định của RTDETRv2 là thiết kế end-to-end NMS-free nguyên bản. Bằng cách loại bỏ hoàn toàn Non-Maximum Suppression (NMS) trong quá trình hậu xử lý, mô hình giúp ổn định độ trễ suy luận và đơn giản hóa quy trình triển khai. Cơ chế chú ý toàn cục cho phép mô hình vượt trội trong việc hiểu các cảnh phức tạp và đám đông dày đặc, vì nó đánh giá toàn bộ ngữ cảnh hình ảnh cùng một lúc.

Tìm hiểu thêm về RTDETRv2

YOLOv9: Thông tin Gradient có thể lập trình#

YOLOv9, một kiến trúc dựa trên CNN hiệu quả cao, giải quyết vấn đề nút thắt thông tin vốn có trong các mạng thần kinh sâu. Nó giới thiệu Programmable Gradient Information (PGI) và Generalized Efficient Layer Aggregation Network (GELAN).

• Tác giả: Chien-Yao Wang và Hong-Yuan Mark Liao
• Tổ chức: Institute of Information Science, Academia Sinica
• Ngày: 21 tháng 2 năm 2024
• Links: Arxiv, GitHub

YOLOv9 dựa trên nền tảng mạng neural tích chập đã được kiểm chứng nhưng tối đa hóa hiệu suất tham số. Bằng cách giữ lại thông tin quan trọng trong quá trình truyền thuận, nó đảm bảo cập nhật trọng số đáng tin cậy, tạo ra một model cực kỳ nhẹ nhưng có độ chính xác cao. Tuy nhiên, không giống như RTDETRv2, YOLOv9 vẫn dựa vào quá trình hậu xử lý NMS tiêu chuẩn.

Tìm hiểu thêm về YOLOv9

Hiệu suất và hiệu quả tài nguyên#

Khi đánh giá các model này cho môi trường production, việc cân bằng giữa mAP (mean Average Precision) và chi phí tính toán là rất quan trọng. Bảng dưới đây minh họa hiệu suất của chúng trên tập dữ liệu MS COCO.

Yêu cầu bộ nhớ và hiệu quả huấn luyện#

Các Transformer như RTDETRv2 vốn đòi hỏi nhiều bộ nhớ trong quá trình huấn luyện, thường yêu cầu lượng bộ nhớ CUDA đáng kể và lịch trình huấn luyện dài hơn để hội tụ hoàn toàn. Ngược lại, các kiến trúc CNN như YOLOv9 và các Ultralytics YOLO models khác mang lại mức sử dụng bộ nhớ thấp hơn đáng kể, cho phép các lập trình viên huấn luyện với kích thước batch lớn hơn trên phần cứng cấp độ phổ thông.

Lợi thế của Ultralytics: Hệ sinh thái và tính dễ sử dụng#

Mặc dù việc nghiên cứu các kho lưu trữ độc lập như trang GitHub chính thức của RTDETRv2 hoặc YOLOv9 có thể mang tính giáo dục cao, các môi trường production đòi hỏi sự ổn định, dễ sử dụng và hệ sinh thái được bảo trì tốt. Việc tích hợp các model này thông qua Ultralytics Python API mang lại trải nghiệm lập trình liền mạch.

API hợp nhất và tính linh hoạt#

Framework Ultralytics giúp trừu tượng hóa các độ phức tạp của việc nạp dữ liệu, tăng cường dữ liệu và huấn luyện phân tán. Hơn nữa, trong khi RTDETRv2 gốc chỉ tập trung nghiêm ngặt vào detection, hệ sinh thái Ultralytics cho phép người dùng dễ dàng chuyển đổi giữa Object Detection, Instance Segmentation và Pose Estimation.

from ultralytics import RTDETR, YOLO

# Train a YOLOv9 model on custom data
model_yolo = YOLO("yolov9c.pt")
model_yolo.train(data="coco8.yaml", epochs=50, imgsz=640)

# Easily switch to RT-DETR for complex scene evaluation
model_rtdetr = RTDETR("rtdetr-l.pt")
results = model_rtdetr.predict("https://ultralytics.com/images/bus.jpg")

# Export to production-ready formats like TensorRT
model_yolo.export(format="engine")

Với tài liệu mạnh mẽ, theo dõi thử nghiệm tự động và khả năng xuất liền mạch sang các định dạng như ONNX, TensorRT và OpenVINO, Ultralytics giảm đáng kể thời gian từ prototype đến production.

Các trường hợp sử dụng lý tưởng#

Nơi RTDETRv2 vượt trội#

Nhờ cơ chế chú ý toàn cục, RTDETRv2 là một cường quốc cho xử lý phía máy chủ (server-side processing) và các môi trường mà ngữ cảnh toàn cục là tối quan trọng. Nó vượt trội trong:

• Chẩn đoán hình ảnh y tế: Xác định các bất thường tinh vi nơi ngữ cảnh xung quanh là rất quan trọng.
• Giám sát trên không: Phát hiện các đối tượng nhỏ trong cảnh quay drone độ phân giải cao mà không bị thiên kiến không gian của các phép tích chập CNN truyền thống.
• Phân tích đám đông dày đặc: Theo dõi các cá nhân nơi tình trạng che khuất nghiêm trọng thường gây nhầm lẫn cho các mô hình dựa trên neo (anchor-based).

Nơi YOLOv9 vượt trội#

YOLOv9 là nhà vô địch về triển khai tại biên (edge deployments) hạn chế tài nguyên. Hiệu quả tính toán của nó làm cho nó trở nên lý tưởng cho:

• Robot: Điều hướng thời gian thực và tránh chướng ngại vật nơi yêu cầu độ trễ tối thiểu.
• IoT Thành Phố Thông Minh: Triển khai trên các thiết bị edge như NVIDIA Jetson để giám sát giao thông.
• Kiểm tra công nghiệp: Kiểm soát chất lượng dây chuyền lắp ráp tốc độ cao đòi hỏi tốc độ khung hình trên giây (FPS) cao.

Tương lai: Sự xuất hiện của Ultralytics YOLO26#

Mặc dù YOLOv9 và RTDETRv2 đại diện cho những bước nhảy vọt lớn, bối cảnh công nghệ đã tiến hóa nhanh chóng. Đối với các hệ thống triển khai hiện đại, Ultralytics YOLO26 mới ra mắt đại diện cho sự kết hợp hoàn hảo của cả hai triết lý kiến trúc.

Bằng cách lấy những khía cạnh tốt nhất của transformer và CNN, YOLO26 thiết lập một tiêu chuẩn mới:

• Thiết kế End-to-End NMS-Free: Giống như RTDETRv2, YOLO26 là end-to-end nguyên bản, loại bỏ hoàn toàn hậu xử lý NMS để có quy trình triển khai nhanh hơn, đơn giản hơn và có khả năng dự đoán cao.
• Bộ tối ưu hóa MuSGD: Lấy cảm hứng từ các kỹ thuật huấn luyện mô hình ngôn ngữ lớn (LLM) (như Kimi K2 của Moonshot AI), YOLO26 sử dụng sự kết hợp giữa SGD và Muon. Điều này mang lại sự ổn định huấn luyện chưa từng có và hội tụ nhanh cho thị giác máy tính.
• Suy luận CPU nhanh hơn tới 43%: Không giống như các transformer nặng nề, YOLO26 được tối ưu hóa mạnh mẽ cho điện toán biên và các thiết bị không có GPU.
• Loại bỏ DFL: Việc loại bỏ Distribution Focal Loss làm đơn giản hóa đáng kể đồ thị mô hình, đảm bảo xuất hoàn hảo sang các thiết bị biên công suất thấp và các Đơn vị xử lý thần kinh (NPU) nhúng.
• ProgLoss + STAL: Các hàm mất mát cải tiến này giúp tăng cường đáng kể khả năng nhận dạng đối tượng nhỏ, một tính năng quan trọng cho các tập dữ liệu IoT và trên không.

Đối với các nhóm đang tìm cách bắt đầu một dự án thị giác máy tính mới, chúng tôi thực sự khuyên bạn nên đánh giá YOLO26. Nó cung cấp sự tinh tế không cần NMS của một transformer với tốc độ cực nhanh và hiệu quả huấn luyện của một kiến trúc YOLO được tối ưu hóa cao.

Tìm hiểu thêm về YOLO26

Tóm tắt#

Việc chọn giữa RTDETRv2 và YOLOv9 phần lớn phụ thuộc vào phần cứng triển khai và nhu cầu độ chính xác cụ thể của bạn. RTDETRv2 cung cấp độ chính xác và khả năng nhận thức ngữ cảnh hàng đầu cho các ứng dụng dựa trên máy chủ, trong khi YOLOv9 cung cấp hiệu quả vượt trội cho các thiết bị biên.

Tuy nhiên, bằng cách tận dụng hệ sinh thái Ultralytics trưởng thành, các lập trình viên có thể dễ dàng thử nghiệm với cả hai. Hơn nữa, với sự ra mắt của các model mới hơn như YOLO11 và YOLO26 end-to-end nguyên bản, việc tìm kiếm sự cân bằng hoàn hảo giữa suy luận tốc độ cao, hỗ trợ tác vụ đa dạng và mức tiêu thụ bộ nhớ thấp chưa bao giờ dễ dàng hơn thế.