Đơn giản hóa quy trình phân loại với Ultralytics YOLOv5 v6.2

YOLOv5 đang đưa công nghệ phát hiện đối tượng lên một tầm cao mới! Từ các mô hình phân loại mới, khả năng tái tạo huấn luyện và Apple Metal Performance Shader ( MPS ) Hỗ trợ tích hợp với ClearML và Deci, chúng tôi giới thiệu đến bạn sản phẩm mới YOLOv5 Phiên bản phát hành v6.2.

Quan trọng YOLOv5 Cập nhật

Chúng tôi đã nỗ lực cải thiện mục yêu thích của bạn YOLO Kiến trúc AI của Vision kể từ bản phát hành mới nhất của chúng tôi vào tháng 2 năm 2022. Đây là những cập nhật quan trọng nhất trong bản mới nhất YOLOv5 Phiên bản 6.2:

• Mô hình phân loại: YOLOv5 -cls ImageNet -Các mô hình phân loại được đào tạo trước hiện đã có sẵn lần đầu tiên.
  ‍
• Ghi nhật ký ClearML : Tích hợp với trình theo dõi thử nghiệm nguồn mở ClearML . Cài đặt bằng pip install clearml sẽ cho phép tích hợp và cho phép người dùng track mỗi buổi tập chạy trong ClearML . Điều này lần lượt cho phép người dùng track và so sánh các lần chạy và thậm chí lên lịch chạy từ xa.
  ‍
• Điểm chuẩn xuất khẩu GPU : Điểm chuẩn ( mAP và tốc độ) tất cả YOLOv5 xuất định dạng với python utils/benchmarks.py --weights yolov5s.pt --device 0 cho GPU điểm chuẩn hoặc --thiết bị CPU vì CPU chuẩn mực.
  ‍
• Khả năng tái tạo đào tạo: Đơn GPU YOLOv5 đào tạo với torch >=1.12.0 hiện có thể tái tạo hoàn toàn và có thể sử dụng đối số --seed mới (seed mặc định=0).
  ‍
• Hỗ trợ Apple Metal Performance Shader ( MPS ): MPS hỗ trợ cho các thiết bị Apple M1/M2 với --device mps (chức năng đầy đủ đang chờ xử lý) torch cập nhật trong pytorch / pytorch #77764 ).

Các Mô hình Phân loại Mới

Mục tiêu chính của chúng tôi với bản phát hành này là giới thiệu YOLOv5 quy trình phân loại, giống như các mô hình phát hiện đối tượng hiện có của chúng tôi. Phiên bản v6.2 mới YOLOv5 Các mô hình -cls chỉ là bước khởi đầu, chúng tôi sẽ tiếp tục cải thiện chúng trong tương lai cùng với các mô hình phát hiện hiện có. Chúng tôi rất mong nhận được sự đóng góp của bạn cho nỗ lực này!

Bản phát hành này kết hợp 401 PR từ 41 người đóng góp kể từ bản phát hành cuối cùng của chúng tôi vào tháng 2 năm 2022. Nó bổ sung đào tạo Phân loại , xác thực, dự đoán và xuất (cho tất cả 11 định dạng) và cũng cung cấp ImageNet -các mô hình YOLOv5m-cls, ResNet (18, 34, 50, 101) và EfficientNet (b0-b3) được đào tạo trước.

Chúng tôi đã đào tạo YOLOv5 -cls mô hình phân loại trên ImageNet trong 90 kỷ nguyên sử dụng phiên bản 4xA100 và chúng tôi đã huấn luyện các mô hình ResNet và EfficientNet cùng với các thiết lập huấn luyện mặc định giống nhau để so sánh. Chúng tôi đã xuất tất cả các mô hình sang ONNX FP32 cho CPU kiểm tra tốc độ và TensorRT FP16 cho GPU kiểm tra tốc độ. Chúng tôi đã chạy tất cả các bài kiểm tra tốc độ trên Google Colab Pro để dễ dàng tái tạo.  

Tiếp theo là gì Ultralytics ?

Bản phát hành tiếp theo của chúng tôi, v6.3, dự kiến ra mắt vào tháng 9 năm 2022 và sẽ mang lại hỗ trợ phân đoạn phiên bản chính thức cho YOLOv5 , với bản phát hành v7.0 chính thức vào cuối năm nay, cập nhật kiến trúc trên cả 3 nhiệm vụ - phân loại, phát hiện và phân đoạn.

Ghé thăm chúng tôi YOLOv5 kho lưu trữ GitHub nguồn mở để cập nhật và tìm hiểu thêm về bản phát hành này.