ヨロビジョン深圳
深セン
今すぐ参加

Intel OpenVINOでの推論用にUltralytics YOLOv8をエクスポートおよび最適化

Nuvola Ladi

2分で読めます

2024年5月24日

OpenVINOを使用して、推論用にUltralytics YOLOv8モデルを最適化します。PyTorchモデルをONNXに変換し、リアルタイムアプリケーション向けに最適化するためのガイドに従ってください。

このブログ記事では、Ultralytics YOLOv8の事前学習済みまたはカスタム学習済みモデルをエクスポートし、OpenVINOを使用した推論用に最適化する方法を見ていきます。CPUまたはGPUのいずれであっても、Intelベースのシステムを使用している場合、このガイドでは、最小限の労力でモデルを大幅に高速化する方法を紹介します。

OpenVINOでYOLOv8を最適化する理由

OpenVINOでYOLOv8モデルを最適化すると、推論タスクの速度が最大3倍向上する可能性があり、特にIntel CPUで実行している場合に効果的です。このパフォーマンスの向上は、物体検出からセグメンテーション、セキュリティシステムまで、リアルタイムアプリケーションにおいて非常に大きな違いをもたらします。

YOLOv8モデルのエクスポートと最適化の手順

プロセスの理解

まず最初に、プロセスを分解してみましょう。PyTorchモデルをONNXに変換し、OpenVINOを使用して最適化します。このプロセスにはいくつかの簡単なステップが含まれており、TensorFlow、PyTorch、Caffe、ONNXなどのさまざまなモデルと形式に適用できます。

モデルのエクスポート

Ultralyticsのドキュメントを見ると、YOLOv8モデルのエクスポートには、Ultralyticsフレームワークのexportメソッドを使用することがわかります。このメソッドを使用すると、モデルをPyTorchからONNXに変換し、最終的にOpenVINO用に最適化できます。その結果、Intelの強力なハードウェアを活用して、モデルの実行速度が大幅に向上します。

依存関係のインストール

エクスポートスクリプトを実行する前に、必要な依存関係がすべてインストールされていることを確認する必要があります。これには、Ultralyticsライブラリ、ONNX、およびOpenVINOが含まれます。これらのパッケージのインストールは、Pythonパッケージインストーラーであるpipを使用して簡単に行うことができます。

エクスポートスクリプトの実行

環境が設定されたら、エクスポートスクリプトを実行できます。このスクリプトは、PyTorchモデルをONNXに変換し、次にOpenVINOに変換します。プロセスは簡単で、エクスポートを処理するために単一の関数を呼び出すだけです。Ultralyticsフレームワークを使用すると、モデルを簡単に変換および最適化できるため、最小限の手間で最高のパフォーマンスを得ることができます。

__wf_reserved_inherit
図1. Nicolai Nielsen氏がエクスポートスクリプトの実行方法を解説。

パフォーマンスの比較

エクスポート後、元のモデルと最適化されたモデルのパフォーマンスを比較することが重要です。両方のモデルの推論時間をベンチマークすることで、パフォーマンスの向上が明確にわかります。通常、OpenVINOモデルは、元のPyTorchモデルと比較して、推論時間が大幅に短縮されます。これは、パフォーマンスの向上が最も顕著な大規模モデルの場合に特に当てはまります。

現実世界のアプリケーションと利点

OpenVINOによるYOLOv8モデルの最適化は、リアルタイム処理を必要とするアプリケーションに特に有効です。以下にいくつかの例を示します。

  • セキュリティシステム: リアルタイムの物体検出は、セキュリティ担当者に即座に警告を発し、安全性と応答性を高めます。
  • 自動運転車: 推論速度の向上は、自動運転システムの応答性を高め、安全性と信頼性を向上させます。
  • 医療: 診断ツール向けの迅速な画像処理は、より迅速な結果を提供することで命を救うことができ、タイムリーな介入を可能にします。

これらの最適化を実装することで、パフォーマンスが向上するだけでなく、アプリケーションの信頼性と効率も向上します。これにより、ユーザーエクスペリエンスの向上、生産性の向上、より革新的なソリューションにつながる可能性があります。

まとめ

OpenVINO用にYOLOv8モデルをエクスポートおよび最適化することは、より高速で効率的なAIアプリケーションのためにIntelハードウェアを活用するための強力な方法です。いくつかの簡単なステップだけで、モデルのパフォーマンスを向上させ、それを現実世界のシナリオに効果的に適用できます。

Ultralyticsのチュートリアルとガイドをチェックして、AIプロジェクトをさらに強化してください。GitHubリポジトリにアクセスし、Ultralyticsコミュニティに参加して、より多くの洞察とアップデートを入手してください。一緒に革新を起こしましょう!

モデルの最適化は、単に速度を上げるだけでなく、新たな可能性を切り開き、AIソリューションが堅牢かつ効率的で、将来に対応できることを保証するものです。 

AIの未来を
共に築きましょう!

未来の機械学習で、新たな一歩を踏み出しましょう。

無料ではじめる
クリップボードにコピーしました