このプロジェクトについて

jetson-inferenceは、NVIDIA Jetsonデバイス上でディープラーニングビジョンモデルをデプロイするための教育プロジェクト兼ライブラリです。TensorRTを使用して最適化されたネットワークをC++またはPythonから実行し、モデルのトレーニングと転移学習にはPyTorchを使用します。 このライブラリは、画像分類、物体検出、セマンティックセグメンテーション、ポーズ推定、行動認識、背景除去、単眼深度推定のためのインターフェースを提供します。READMEには、画像とライブカメラフィードを処理するためのC++およびPythonの例、ならびにカメラストリーミング、WebRTC Webアプリケーション、Flask RESTサービス、Plotlyダッシュボードのチュートリアルが含まれています。ROSおよびROS2統合は、関連プロジェクトであるros_deep_learningを通じて提供されます。 Hello AI Worldの教材では、Jetson上で完全に推論を実行し、データセットを収集し、モデルを再トレーニングし、得られたモデルをTensorRTでデプロイする方法を扱っています。同じPyTorchトレーニングコードは、NVIDIAのディスクリートGPUを搭載したPC、サーバー、またはクラウドインスタンスでも使用できます。 このプロジェクトには、ImageNet形式の認識、COCO物体検出、交通およびドライブレコーダーシナリオ、顔検出、CityscapesおよびPascal VOCセグメンテーション、身体および手のポーズ推定、Kinetics行動認識のための事前学習済みモデルが含まれています。ソースビルド、Dockerコンテナ、JetPackセットアップ手順が提供され、Orinデバイス向けにJetPack 6のサポートが記載されています。C++およびPython APIは、libjetson-inferenceおよびlibjetson-utilsを通じて外部プロジェクトにリンクできます。