Chuyển đến nội dung

YOLO -NAS

Cập nhật quan trọng

Xin lưu ý rằng Deci , những người sáng tạo ban đầu của YOLO -NAS, đã được mua lại bởi NVIDIA . Do đó, các mô hình này không còn được Deci duy trì tích cực nữa. Ultralytics tiếp tục hỗ trợ việc sử dụng các mô hình này, nhưng không mong đợi bất kỳ bản cập nhật nào nữa từ nhóm ban đầu.

Tổng quan

Được phát triển bởi Deci AI , YOLO -NAS là một mô hình nền tảng phát hiện đối tượng đột phá. Đây là sản phẩm của công nghệ Tìm kiếm Kiến trúc Nơ-ron tiên tiến, được thiết kế tỉ mỉ để khắc phục những hạn chế của các công nghệ trước đây. YOLO mô hình. Với những cải tiến đáng kể trong hỗ trợ lượng tử hóa và sự đánh đổi giữa độ chính xác và độ trễ, YOLO -NAS là bước tiến lớn trong việc phát hiện vật thể.

Ảnh ví dụ mô hìnhTổng quan về YOLO -NAS. YOLO -NAS sử dụng các khối nhận biết lượng tử hóa và lượng tử hóa chọn lọc để đạt hiệu suất tối ưu. Mô hình, khi được chuyển đổi sang phiên bản lượng tử hóa INT8, có độ chính xác giảm thiểu tối đa, một cải tiến đáng kể so với các mô hình khác. Những tiến bộ này tạo nên một kiến trúc vượt trội với khả năng phát hiện đối tượng chưa từng có và hiệu suất vượt trội.

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

  • Khối cơ bản thân thiện với lượng tử hóa: YOLO -NAS giới thiệu một khối cơ bản mới thân thiện với lượng tử hóa, giải quyết một trong những hạn chế đáng kể của trước đây YOLO các mô hình.
  • Đào tạo và lượng tử hóa tinh vi: YOLO -NAS tận dụng các chương trình đào tạo nâng cao và lượng tử hóa sau đào tạo để nâng cao hiệu suất.
  • Tối ưu hóa AutoNAC và đào tạo trước: YOLO -NAS sử dụng tối ưu hóa AutoNAC và được đào tạo trước trên các tập dữ liệu nổi bật như COCO , Objects365 và Roboflow 100. Quá trình đào tạo trước này khiến nó cực kỳ phù hợp cho các tác vụ phát hiện đối tượng tiếp theo trong môi trường sản xuất.

Các mô hình được huấn luyện trước

Trải nghiệm sức mạnh của công nghệ phát hiện đối tượng thế hệ tiếp theo với hệ thống được đào tạo trước YOLO -Các mô hình NAS được cung cấp bởi Ultralytics . Các mẫu máy này được thiết kế để mang lại hiệu suất vượt trội về cả tốc độ lẫn độ chính xác. Hãy lựa chọn từ nhiều tùy chọn phù hợp với nhu cầu cụ thể của bạn:

Hiệu suất

Mô hìnhmAPĐộ Trễ (ms)
YOLO -NAS S47.53.21
YOLO -NAS M51.555.85
YOLO -NAS L52.227.87
YOLO -NAS S INT-847.032.36
YOLO -NAS M INT-851.03.78
YOLO -NAS L INT-852.14.78

Mỗi biến thể mô hình được thiết kế để cung cấp sự cân bằng giữa Độ chính xác trung bình ( mAP ) và độ trễ, giúp bạn tối ưu hóa tác vụ phát hiện đối tượng về cả hiệu suất và tốc độ.

Ví dụ sử dụng

Ultralytics đã làm YOLO -Các mô hình NAS dễ dàng tích hợp vào Python ứng dụng thông qua của chúng tôi ultralytics gói python. Gói này cung cấp Python API thân thiện với người dùng để hợp lý hóa quy trình.

Các ví dụ sau đây cho thấy cách sử dụng YOLO -Các mô hình NAS với ultralytics gói để suy luận và xác thực:

Ví dụ về Suy luận và Xác thực

Trong ví dụ này chúng tôi xác thực YOLO -NAS-s trên COCO8 tập dữ liệu.

Ví dụ

Ví dụ này cung cấp mã suy luận và xác thực đơn giản cho YOLO -NAS. Để xử lý kết quả suy luận, hãy xem Dự đoán chế độ. Để sử dụng YOLO -NAS với các chế độ bổ sung xem ValXuất. YOLO-NAS trên ultralytics gói không hỗ trợ huấn luyện.

PyTorch đã được huấn luyện trước *.pt các tệp mô hình có thể được chuyển đến NAS() class để tạo một thể hiện mô hình trong python:

from ultralytics import NAS

# Load a COCO-pretrained YOLO-NAS-s model
model = NAS("yolo_nas_s.pt")

# Display model information (optional)
model.info()

# Validate the model on the COCO8 example dataset
results = model.val(data="coco8.yaml")

# Run inference with the YOLO-NAS-s model on the 'bus.jpg' image
results = model("path/to/bus.jpg")

Các lệnh CLI có sẵn để chạy trực tiếp các mô hình:

# Load a COCO-pretrained YOLO-NAS-s model and validate it's performance on the COCO8 example dataset
yolo val model=yolo_nas_s.pt data=coco8.yaml

# Load a COCO-pretrained YOLO-NAS-s model and run inference on the 'bus.jpg' image
yolo predict model=yolo_nas_s.pt source=path/to/bus.jpg

Các Tác vụ và Chế độ được Hỗ trợ

Chúng tôi cung cấp ba biến thể của YOLO - Các model NAS: Nhỏ (s), Trung bình (m) và Lớn (l). Mỗi phiên bản được thiết kế để đáp ứng các nhu cầu tính toán và hiệu suất khác nhau:

  • YOLO -NAS-s : Được tối ưu hóa cho môi trường có tài nguyên tính toán hạn chế nhưng hiệu quả là yếu tố then chốt.
  • YOLO -NAS-m : Cung cấp phương pháp cân bằng, phù hợp để phát hiện đối tượng mục đích chung với độ chính xác cao hơn.
  • YOLO -NAS-l : Được thiết kế riêng cho các tình huống đòi hỏi độ chính xác cao nhất, trong đó tài nguyên tính toán ít bị hạn chế.

Dưới đây là tổng quan chi tiết về từng mô hình, bao gồm các liên kết đến trọng số đã được huấn luyện trước, các tác vụ mà chúng hỗ trợ và khả năng tương thích của chúng với các chế độ hoạt động khác nhau.

Loại mô hìnhTrọng số được huấn luyện trướcCác tác vụ được hỗ trợSuy luậnXác thựcHuấn luyệnXuất
YOLO -NAS-syolo_nas_s.ptPhát Hiện Đối Tượng
YOLO -NAS-myolo_nas_m.ptPhát Hiện Đối Tượng
YOLO -NAS-lyolo_nas_l.ptPhát Hiện Đối Tượng

Trích dẫn và Lời cảm ơn

Nếu bạn sử dụng YOLO -NAS trong công trình nghiên cứu hoặc phát triển của bạn, vui lòng trích dẫn SuperGradients:

@misc{supergradients,
      doi = {10.5281/ZENODO.7789328},
      url = {https://zenodo.org/records/7789328},
      author = {Aharon,  Shay and {Louis-Dupont} and {Ofri Masad} and Yurkova,  Kate and {Lotem Fridman} and {Lkdci} and Khvedchenya,  Eugene and Rubin,  Ran and Bagrov,  Natan and Tymchenko,  Borys and Keren,  Tomer and Zhilko,  Alexander and {Eran-Deci}},
      title = {Super-Gradients},
      publisher = {GitHub},
      journal = {GitHub repository},
      year = {2021},
}

Chúng tôi bày tỏ lòng biết ơn tới Deci AI nhóm SuperGradients vì những nỗ lực của họ trong việc tạo ra và duy trì nguồn tài nguyên có giá trị này cho cộng đồng thị giác máy tính . Chúng tôi tin rằng YOLO -NAS, với kiến trúc cải tiến và khả năng phát hiện đối tượng vượt trội, sẽ trở thành một công cụ quan trọng cho cả nhà phát triển và nhà nghiên cứu.

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

Là gì YOLO -NAS và nó cải thiện như thế nào so với trước đây YOLO mô hình?

YOLO -NAS, được phát triển bởi Deci AI là một mô hình phát hiện đối tượng tiên tiến tận dụng công nghệ Tìm kiếm Kiến trúc Thần kinh (NAS) tiên tiến. Nó giải quyết những hạn chế của các mô hình trước đây YOLO bằng cách giới thiệu các tính năng như khối cơ bản thân thiện với lượng tử hóa và các lược đồ huấn luyện tinh vi. Điều này mang lại những cải thiện đáng kể về hiệu suất, đặc biệt là trong môi trường có tài nguyên tính toán hạn chế. YOLO -NAS cũng hỗ trợ lượng tử hóa, duy trì độ chính xác cao ngay cả khi chuyển đổi sang phiên bản INT8, nâng cao khả năng phù hợp với môi trường sản xuất. Để biết thêm chi tiết, vui lòng xem phần Tổng quan .

Làm thế nào tôi có thể tích hợp YOLO -Các mô hình NAS vào tôi Python ứng dụng?

Bạn có thể dễ dàng tích hợp YOLO -Các mô hình NAS vào Python ứng dụng sử dụng ultralytics gói. Đây là một ví dụ đơn giản về cách tải một gói được đào tạo trước YOLO -Mô hình NAS và thực hiện suy luận:

from ultralytics import NAS

# Load a COCO-pretrained YOLO-NAS-s model
model = NAS("yolo_nas_s.pt")

# Validate the model on the COCO8 example dataset
results = model.val(data="coco8.yaml")

# Run inference with the YOLO-NAS-s model on the 'bus.jpg' image
results = model("path/to/bus.jpg")

Để biết thêm thông tin, hãy tham khảo Các ví dụ về suy luận và xác thực.

Các tính năng chính của là gì? YOLO -NAS và tại sao tôi nên cân nhắc sử dụng nó?

YOLO -NAS giới thiệu một số tính năng chính giúp nó trở thành lựa chọn vượt trội cho các nhiệm vụ phát hiện đối tượng:

  • Khối cơ bản thân thiện với lượng tử hóa: Kiến trúc nâng cao giúp cải thiện hiệu suất mô hình với độ giảm độ chính xác tối thiểu sau lượng tử hóa.
  • Huấn luyện và lượng tử hóa phức tạp: Sử dụng các lược đồ huấn luyện nâng cao và các kỹ thuật lượng tử hóa sau huấn luyện.
  • Tối ưu hóa AutoNAC và đào tạo trước: Sử dụng tối ưu hóa AutoNAC và được đào tạo trước trên các tập dữ liệu nổi bật như COCO , Objects365 và Roboflow 100.

Các tính năng này đóng góp vào độ chính xác cao, hiệu suất hiệu quả và phù hợp để triển khai trong môi trường sản xuất. Tìm hiểu thêm trong phần Các tính năng chính.

Những nhiệm vụ và chế độ nào được hỗ trợ bởi YOLO -Mô hình NAS?

YOLO -Các mô hình NAS hỗ trợ nhiều tác vụ và chế độ phát hiện đối tượng khác nhau như suy luận, xác thực và xuất dữ liệu. Chúng không hỗ trợ đào tạo. Các mô hình được hỗ trợ bao gồm YOLO -NAS-s, YOLO -NAS-m, và YOLO -NAS-l, mỗi loại được thiết kế riêng cho các khả năng tính toán và nhu cầu hiệu suất khác nhau. Để biết tổng quan chi tiết, vui lòng tham khảo phần Nhiệm vụ và Chế độ được Hỗ trợ .

Có được đào tạo trước không? YOLO -Các mẫu NAS hiện có và tôi có thể truy cập chúng bằng cách nào?

Đúng, Ultralytics cung cấp đào tạo trước YOLO - Các mô hình NAS mà bạn có thể truy cập trực tiếp. Các mô hình này được đào tạo trước trên các tập dữ liệu như COCO , đảm bảo hiệu suất cao về cả tốc độ và độ chính xác. Bạn có thể tải xuống các mô hình này bằng các liên kết được cung cấp trong phần Mô hình được đào tạo trước . Dưới đây là một số ví dụ:



📅 Đã tạo 2 năm trước ✏️ Cập nhật 1 tháng trước
glenn-jocherRizwanMunawarY-T-GLaughing-qMatthewNoyce

Bình luận