Ultralytics YOLOv5
概要
YOLOv5uは、物体検出手法における進歩を表すものです。Ultralyticsが開発したYOLOv5モデルの基礎アーキテクチャから派生したYOLOv5uは、YOLOv8モデルで導入されたアンカーフリーかつオブジェクネスフリーなスプリットヘッドを統合しています。この適応によりモデルアーキテクチャが洗練され、物体検出タスクにおける精度と速度のトレードオフが改善されました。経験的な結果とその派生機能に基づき、YOLOv5uは研究および実用アプリケーションの両面で堅牢なソリューションを求めるユーザーにとって効率的な代替手段となります。

Ultralyticsは、YOLOv5モデルのアンカーフリー版を提供しています。オリジナルのYOLOv5リポジトリで学習されたモデルは、Ultralyticsライブラリでは使用できません。
YOLOv5モデルをUltralytics Platformで直接探索および実行できます。
主な特徴
-
アンカーフリー・スプリットUltralyticsヘッド: 従来の物体検出モデルは、事前に定義されたアンカーボックスを使用して物体の位置を予測していました。しかし、YOLOv5uはこの手法を現代化しています。アンカーフリーなスプリットUltralyticsヘッドを採用することで、より柔軟で適応性の高い検出メカニズムが保証され、結果として多様なシナリオでのパフォーマンスが向上します。
-
最適化された精度と速度のトレードオフ: 速度と精度はしばしば相反する関係にあります。しかし、YOLOv5uはこのトレードオフに挑戦しています。調整されたバランスを提供し、精度を損なうことなくリアルタイムの検出を実現します。この機能は、自律走行車、ロボティクス、リアルタイム映像解析など、迅速な応答が求められるアプリケーションにとって特に重要です。
-
多様な学習済みモデル: タスクごとに異なるツールセットが必要であることを理解し、YOLOv5uは多数の学習済みモデルを提供しています。推論、検証、学習のいずれに重点を置く場合でも、あなたにぴったりのモデルが見つかります。この多様性により、汎用的なソリューションを使用するだけでなく、独自の課題に対して特別に微調整されたモデルを利用できます。
サポートされているタスクとモード
様々な学習済み重みを持つYOLOv5uモデルは、物体検出タスクにおいて優れた性能を発揮します。幅広いモードをサポートしており、開発からデプロイメントまで多様なアプリケーションに適しています。
| モデルタイプ | 事前学習済み重み | タスク | 推論 (Inference) | 検証 | 学習 | Export |
|---|---|---|---|---|---|---|
| YOLOv5u | yolov5nu, yolov5su, yolov5mu, yolov5lu, yolov5xu, yolov5n6u, yolov5s6u, yolov5m6u, yolov5l6u, yolov5x6u | 物体検出 | ✅ | ✅ | ✅ | ✅ |
この表は、YOLOv5uモデルの各バリエーションの詳細な概要であり、物体検出タスクへの適合性、および推論、検証、学習、エクスポートといった様々な運用モードへの対応を示しています。この包括的なサポートにより、ユーザーは幅広い物体検出シナリオにおいてYOLOv5uモデルの能力を最大限に活用できます。
性能メトリクス
See Detection Docs for usage examples with these models trained on COCO, which include 80 pretrained classes.
| モデル | YAML | サイズ (ピクセル) | mAPval 50-95 | 速度 CPU ONNX (ms) | 速度 A100 TensorRT (ms) | パラメータ (M) | FLOPs (B) |
|---|---|---|---|---|---|---|---|
| yolov5nu.pt | yolov5n.yaml | 640 | 34.3 | 73.6 | 1.06 | 2.6 | 7.7 |
| yolov5su.pt | yolov5s.yaml | 640 | 43.0 | 120.7 | 1.27 | 9.1 | 24.0 |
| yolov5mu.pt | yolov5m.yaml | 640 | 49.0 | 233.9 | 1.86 | 25.1 | 64.2 |
| yolov5lu.pt | yolov5l.yaml | 640 | 52.2 | 408.4 | 2.50 | 53.2 | 135.0 |
| yolov5xu.pt | yolov5x.yaml | 640 | 53.2 | 763.2 | 3.81 | 97.2 | 246.4 |
| yolov5n6u.pt | yolov5n6.yaml | 1280 | 42.1 | 211.0 | 1.83 | 4.3 | 7.8 |
| yolov5s6u.pt | yolov5s6.yaml | 1280 | 48.6 | 422.6 | 2.34 | 15.3 | 24.6 |
| yolov5m6u.pt | yolov5m6.yaml | 1280 | 53.6 | 810.9 | 4.36 | 41.2 | 65.7 |
| yolov5l6u.pt | yolov5l6.yaml | 1280 | 55.7 | 1470.9 | 5.47 | 86.1 | 137.4 |
| yolov5x6u.pt | yolov5x6.yaml | 1280 | 56.8 | 2436.5 | 8.98 | 155.4 | 250.7 |
使用例
この例では、基本的なYOLOv5の学習および推論の例を紹介します。これらやその他のモードに関する完全なドキュメントについては、推論、学習、検証、およびエクスポートのドキュメントページを参照してください。
PyTorch の事前学習済み *.pt モデル、および設定 *.yaml ファイルを YOLO() クラスに渡すことで、Pythonでモデルインスタンスを作成できます。
from ultralytics import YOLO
# Load a COCO-pretrained YOLOv5n model
model = YOLO("yolov5n.pt")
# Display model information (optional)
model.info()
# Train the model on the COCO8 example dataset for 100 epochs
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)
# Run inference with the YOLOv5n model on the 'bus.jpg' image
results = model("path/to/bus.jpg")引用と謝辞
Ultralyticsは、モデルの急速な進化のため、YOLOv5に関する正式な研究論文を出版していません。私たちは静的なドキュメントを作成することよりも、技術を進歩させ、使いやすくすることに注力しています。YOLOのアーキテクチャ、機能、使用方法に関する最新情報については、GitHubリポジトリおよびドキュメントを参照してください。
研究でYOLOv5またはYOLOv5uを使用する場合は、次のようにUltralytics YOLOv5リポジトリを引用してください。
@software{yolov5,
title = {Ultralytics YOLOv5},
author = {Glenn Jocher},
year = {2020},
version = {7.0},
license = {AGPL-3.0},
url = {https://github.com/ultralytics/yolov5},
doi = {10.5281/zenodo.3908559},
orcid = {0000-0001-5950-6979}
}YOLOv5モデルは、AGPL-3.0およびEnterpriseライセンスの下で提供されていることに注意してください。
FAQ
Ultralytics YOLOv5uとは何ですか?また、YOLOv5とどのように異なりますか?
Ultralytics YOLOv5uは、YOLOv5の高度なバージョンであり、リアルタイム物体検出タスクの精度と速度のトレードオフを改善する、アンカーフリーかつオブジェクネスフリーなスプリットヘッドを統合しています。従来のYOLOv5とは異なり、YOLOv5uはアンカーフリーの検出メカニズムを採用しており、多様なシナリオにおいてより柔軟で適応性が高くなっています。その機能の詳細については、YOLOv5の概要を参照してください。
アンカーフリーのUltralyticsヘッドは、YOLOv5uでどのように物体検出パフォーマンスを向上させますか?
YOLOv5uのアンカーフリーUltralyticsヘッドは、事前に定義されたアンカーボックスへの依存を排除することで物体検出パフォーマンスを向上させます。これにより、さまざまな物体のサイズや形状をより効率的に扱える、柔軟で適応性の高い検出メカニズムが実現します。この強化は、精度と速度のバランスのとれたトレードオフに直接寄与し、YOLOv5uをリアルタイムアプリケーションに適したものにしています。アーキテクチャの詳細については、主な機能セクションを参照してください。
学習済みYOLOv5uモデルを異なるタスクやモードで使用できますか?
はい、学習済みYOLOv5uモデルは、物体検出などの様々なタスクで使用できます。これらのモデルは、推論、検証、学習、エクスポートなど、複数のモードをサポートしています。この柔軟性により、ユーザーはさまざまな運用要件にわたってYOLOv5uモデルの機能を活用できます。詳細な概要については、サポートされているタスクとモードセクションを確認してください。
YOLOv5uモデルのパフォーマンス指標は、プラットフォームごとにどのように異なりますか?
YOLOv5uモデルのパフォーマンス指標は、使用されるプラットフォームやハードウェアによって異なります。例えば、YOLOv5nuモデルはCOCOデータセットで34.3のmAPを達成し、CPU (ONNX) で73.6 ms、A100 TensorRTで1.06 msの速度を実現します。様々なYOLOv5uモデルの詳細なパフォーマンス指標は、パフォーマンス指標セクションに記載されており、さまざまなデバイス間での包括的な比較が可能です。
Ultralytics Python APIを使用してYOLOv5uモデルを学習する方法は?
学習済みモデルをロードし、データセットを指定して学習コマンドを実行することで、YOLOv5uモデルを学習できます。簡単な例を以下に示します。
from ultralytics import YOLO
# Load a COCO-pretrained YOLOv5n model
model = YOLO("yolov5n.pt")
# Display model information (optional)
model.info()
# Train the model on the COCO8 example dataset for 100 epochs
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)より詳細な手順については、使用例セクションを参照してください。