YOLOv5 v6.2 ile Sınıflandırma İş Akışlarını Basitleştirme

YOLOv5 nesne algılama alanındaki son teknolojiyi yeni zirvelere taşıyor! Yeni sınıflandırma modelleri, eğitim tekrarlanabilirliği ve Apple Metal Performance Shader (MPS) Desteğinden ClearML ve Deci ile entegrasyonlara kadar, size yeni YOLOv5 v6.2 sürümünü sunuyoruz.

Önemli YOLOv5 Güncellemeler

Şubat 2022'deki son sürümümüzden bu yana favori YOLO Vision AI mimarinizi geliştirmek için çalışıyoruz. Bunlar, en son YOLOv5 v6.2'deki en önemli güncellemelerdir:

• Sınıflandırma Modelleri: YOLOv5-cls ImageNet tarafından önceden eğitilmiş sınıflandırma modelleri artık ilk kez kullanılabilir.
• ClearML Günlük kaydı: Açık kaynak deney izleyicisi ile entegrasyon ClearML. Pip install clearml ile kurulum, entegrasyonu etkinleştirecek ve kullanıcıların ClearML adresindeki her antrenman koşusunu takip etmesine olanak tanıyacaktır. Bu da kullanıcıların koşuları takip etmesine ve karşılaştırmasına ve hatta koşuları uzaktan planlamasına olanak tanır.
• GPU Benchmarkları Dışa Aktar: python utils/benchmarks.py --weights yolov5s.pt --device 0 ile GPU benchmarkları için veya --device CPU ile CPU benchmarkları için tüm YOLOv5 dışa aktarma formatlarını benchmarklayın (mAP ve hız).
• Eğitim Tekrar Üretilebilirliği: torch >=1.12.0 ile tekGPU YOLOv5 eğitimi artık tamamen tekrarlanabilir ve yeni bir --seed argümanı kullanılabilir (varsayılan seed=0).
• Apple Metal Performance Shader (MPS) Desteği: --device mps ile Apple M1/M2 cihazları için MPS desteği (tam işlevsellik torch güncellemelerini pytorch/ pytorch#77764'de bekliyor).

Yeni Sınıflandırma Modelleri

Bu sürümle ana hedefimiz, mevcut nesne algılama modellerimiz gibi basit YOLOv5 sınıflandırma iş akışlarını tanıtmaktır. Yeni v6.2 YOLOv5-cls modelleri sadece bir başlangıç, bunları mevcut algılama modellerimizle birlikte geliştirmeye devam edeceğiz. Bu çabaya katkılarınızı çok isteriz!

Bu sürüm, Şubat 2022'deki son sürümümüzden bu yana 41 katılımcının 401 PR'sini içermektedir. Sınıflandırma eğitimi, doğrulama, tahmin ve dışa aktarma (11 formatın tümüne) ekler ve ayrıca ImageNet tarafından önceden eğitilmiş YOLOv5m-cls, ResNet (18, 34, 50, 101) ve EfficientNet (b0-b3) modelleri sağlar.

YOLOv5-cls sınıflandırma modellerini ImageNet üzerinde 4xA100 örneği kullanarak 90 epoch boyunca eğittik ve karşılaştırmak için ResNet ve EfficientNet modellerini aynı varsayılan eğitim ayarlarıyla birlikte eğittik. Tüm modelleri CPU hız testleri için ONNX FP32'ye ve GPU hız testleri için TensorRT FP16'ya aktardık. Kolay tekrarlanabilirlik için tüm hız testlerini Google Colab Pro 'da gerçekleştirdik.  

Ultralytics'dan Sırada Ne Var?

Bir sonraki sürümümüz olan v6.3'ün Eylül 2022'de yayınlanması ve YOLOv5 adresine resmi örnek segmentasyon desteği getirmesi planlanmaktadır. Bu yılın sonlarında yayınlanacak olan v7.0 sürümü ise sınıflandırma, tespit ve segmentasyon olmak üzere 3 görevin tamamında mimarileri güncelleyecektir.

Güncel kalmak ve bu sürüm hakkında daha fazla bilgi edinmek için YOLOv5 açık kaynaklı GitHub depomuzu ziyaret edin.