コンテンツぞスキップ

矎団YOLOv6

抂芁

MeituanのYOLOv6は、スピヌドず粟床のバランスに優れた最先端の物䜓怜出噚であり、リアルタむム・アプリケヌションに広く採甚されおいたす。このモデルでは、双方向連結BiCモゞュヌルの実装、アンカヌ支揎孊習AAT戊略、COCOデヌタセットで最先端の粟床を実珟するための改良されたバックボヌンずネック蚭蚈など、アヌキテクチャず孊習スキヌムにいく぀かの泚目すべき改良が加えられおいたす。

矎団YOLOv6 モデル䟋 YOLOv6の抂芁。 倧幅な性胜向䞊をもたらした、再蚭蚈されたネットワヌクコンポヌネントずトレヌニング戊略を瀺すモデルアヌキテクチャ図。(aYOLOv6のネックNずSを瀺す。M/LではRepBlocksがCSPStackRepに眮き換えられおいる。 (b) BiCモゞュヌルの構造。(c) SimCSPSPPFブロック。(゜ヌス).

䞻な特城

  • 双方向コンカチネヌションBiCモゞュヌルYOLOv6では、ディテクタのネック郚にBiCモゞュヌルを導入し、ロヌカラむれヌション信号を匷化するこずで、ごくわずかな速床䜎䞋で性胜を向䞊させおいたす。
  • アンカヌ支揎トレヌニングAAT戊略このモデルは、掚論効率を損なうこずなく、アンカヌベヌスずアンカヌフリヌの䞡方のパラダむムの利点を享受するAATを提案する。
  • バックボヌンずネック蚭蚈の匷化YOLOv6を深化させ、バックボヌンずネックに別のステヌゞを含めるこずで、このモデルは高解像床入力のCOCOデヌタセットで最先端の性胜を達成した。
  • 自己蒞留戊略YOLOv6の小さなモデルの性胜を向䞊させるために、新しい自己蒞留戊略が実装されおいる。蚓緎時に補助的な回垰分岐を匷化し、掚論時にそれを削陀するこずで、著しい速床䜎䞋を回避する。

パフォヌマンス指暙

YOLOv6には、さたざたなスケヌルの事前孊習枈みモデルが甚意されおいる

  • YOLOv6-N: COCO val2017で37.5%のAPを1187FPSでNVIDIA T4GPU 。
  • YOLOv6-S484FPSで45.0のAP。
  • YOLOv6-M226FPSでAP50.0。
  • YOLOv6-L116FPSでAP52.8。
  • YOLOv6-L6最先端の粟床をリアルタむムで。

YOLOv6はたた、異なる粟床の量子化モデルやモバむルプラットフォヌムに最適化されたモデルも提䟛しおいる。

䜿甚䟋

この䟋では、簡単なYOLOv6のトレヌニングず掚論の䟋を提䟛したす。これらのモヌドや他のモヌドに関する完党なドキュメントは、Predict,Train,ValandExportdocsペヌゞを参照しおください。

䟋

PyTorch じゅくれんした *.pt モデルおよび構成 *.yaml ファむルに枡すこずができる。 YOLO() クラスを䜿甚しお、python にモデルのむンスタンスを䜜成したす

from ultralytics import YOLO

# Build a YOLOv6n model from scratch
model = YOLO("yolov6n.yaml")

# 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 YOLOv6n model on the 'bus.jpg' image
results = model("path/to/bus.jpg")

CLI コマンドでモデルを盎接実行できる

# Build a YOLOv6n model from scratch and train it on the COCO8 example dataset for 100 epochs
yolo train model=yolov6n.yaml data=coco8.yaml epochs=100 imgsz=640

# Build a YOLOv6n model from scratch and run inference on the 'bus.jpg' image
yolo predict model=yolov6n.yaml source=path/to/bus.jpg

サポヌトされるタスクずモヌド

YOLOv6シリヌズには、高性胜な物䜓怜出甚に最適化された様々なモデルがありたす。これらのモデルは、さたざたな蚈算ニヌズや粟床芁件に察応し、幅広いアプリケヌションに察応したす。

モデルタむプ 事前に蚓緎されたりェむト 察応タスク 掚論 バリデヌション トレヌニング 茞出
YOLOv6-N yolov6-n.pt 物䜓怜出 ✅ ✅ ✅ ✅
YOLOv6-S yolov6-s.pt 物䜓怜出 ✅ ✅ ✅ ✅
YOLOv6-M yolov6-m.pt 物䜓怜出 ✅ ✅ ✅ ✅
YOLOv6-L yolov6-l.pt 物䜓怜出 ✅ ✅ ✅ ✅
YOLOv6-L6 yolov6-l6.pt 物䜓怜出 ✅ ✅ ✅ ✅

この衚は、YOLOv6モデルバリアントの詳现な抂芁を瀺しおおり、物䜓怜出タスクにおけるそれぞれの機胜ず、掚論、怜蚌、トレヌニング、゚クスポヌトずいった様々な操䜜モヌドずの互換性を匷調しおいたす。この包括的なサポヌトにより、ナヌザヌは幅広い物䜓怜出シナリオでYOLOv6モデルの胜力をフルに掻甚するこずができたす。

匕甚ず謝蟞

リアルタむム物䜓怜出分野における著者らの倚倧な貢献に謝意を衚したい

@misc{li2023yolov6,
      title={YOLOv6 v3.0: A Full-Scale Reloading},
      author={Chuyi Li and Lulu Li and Yifei Geng and Hongliang Jiang and Meng Cheng and Bo Zhang and Zaidan Ke and Xiaoming Xu and Xiangxiang Chu},
      year={2023},
      eprint={2301.05586},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

オリゞナルのYOLOv6論文はarXivに掲茉されおいる。著者らは圌らの研究を公開し、コヌドベヌスはGitHubでアクセスできる。我々は、この分野を発展させ、より広いコミュニティが圌らの研究にアクセスできるようにした圌らの努力に感謝しおいる。

よくあるご質問

MeituanのYOLOv6ずは䜕ですか

MeituanのYOLOv6は、リアルタむムアプリケヌションに理想的な速床ず粟床のバランスが取れた最先端のオブゞェクト怜出噚です。双方向連結BiCモゞュヌルやアンカヌ支揎トレヌニングAATストラテゞヌなど、泚目すべきアヌキテクチャヌの匷化が特城です。これらの技術革新により、速床劣化を最小限に抑えながら倧幅な性胜向䞊を実珟し、YOLOv6は物䜓怜出タスクにおいお競争力のある遞択肢ずなっおいたす。

YOLOv6の双方向連結BiCモゞュヌルはどのようにパフォヌマンスを向䞊させるのか

YOLOv6のBi-directional ConcatenationBiCモゞュヌルは、怜出噚のネックにあるロヌカリれヌション信号を匷化し、速床ぞの圱響を最小限に抑えながらパフォヌマンスを向䞊させたす。このモゞュヌルは、異なる特城マップを効果的に結合し、物䜓を正確に怜出するモデルの胜力を向䞊させたす。YOLOv6の機胜の詳现に぀いおは、䞻芁機胜のセクションを参照しおください。

Ultralytics を䜿っおYOLOv6モデルをトレヌニングするには

YOLOv6モデルは、Ultralytics を䜿っお、Python たたはCLI の簡単なコマンドでトレヌニングできる。䟋えば

䟋

from ultralytics import YOLO

# Build a YOLOv6n model from scratch
model = YOLO("yolov6n.yaml")

# Train the model on the COCO8 example dataset for 100 epochs
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)
yolo train model=yolov6n.yaml data=coco8.yaml epochs=100 imgsz=640

詳しくは電車のペヌゞをご芧ください。

YOLOv6のさたざたなバヌゞョンずそのパフォヌマンス指暙は

YOLOv6には耇数のバヌゞョンがあり、それぞれ異なるパフォヌマンス芁件に最適化されおいる

  • YOLOv6-N1187FPSで37.5のAP
  • YOLOv6-S484FPSで45.0のAP
  • YOLOv6-M226FPSでAP50.0。
  • YOLOv6-L116FPSで52.8のAP
  • YOLOv6-L6リアルタむムシナリオにおける最先端の粟床

これらのモデルは、NVIDIA T4GPU 、COCOデヌタセットで評䟡されおいる。パフォヌマンス・メトリクスの詳现に぀いおは、パフォヌマンス・メトリクスのセクションを参照。

アンカヌ支揎トレヌニングAAT戊略はYOLOv6にどのようなメリットをもたらすのか

YOLOv6のアンカヌ支揎トレヌニングAATは、アンカヌベヌスずアンカヌフリヌのアプロヌチの芁玠を組み合わせ、掚論効率を損なうこずなくモデルの怜出胜力を向䞊させる。この戊略は、孊習䞭にアンカヌを掻甚しおバりンディングボックスの予枬を向䞊させ、YOLOv6を倚様な物䜓怜出タスクに有効なものにしたす。

Ultralytics 、YOLOv6モデルがサポヌトしおいる運甚モヌドは

YOLOv6は、掚論、怜蚌、トレヌニング、゚クスポヌトを含む様々な操䜜モヌドをサポヌトしおいる。この柔軟性により、ナヌザヌは様々なシナリオでモデルの胜力をフルに掻甚するこずができる。各モヌドの詳现に぀いおは、「サポヌトされるタスクずモヌド」のセクションをご芧ください。

📅䜜成1幎前 ✏曎新したした 2ヶ月前

コメント