オブジェクト検出 - MXNet - Amazon SageMaker AI

翻訳は機械翻訳により提供されています。提供された翻訳内容と英語版の間で齟齬、不一致または矛盾がある場合、英語版が優先します。

オブジェクト検出 - MXNet

Amazon SageMaker AI オブジェクト検出 - MXNet アルゴリズムは、単一のディープニューラルネットワークを使用してイメージ内のオブジェクトを検出して分類します。このアルゴリズムは、入力としてイメージを取得し、イメージシーン内のオブジェクトのすべてのインスタンスを識別する、教師あり学習アルゴリズムです。オブジェクトは、指定されたコレクション内のいずれかのクラスに分類され、そのクラスに属しているという信頼性スコアが付けられます。イメージ内のその場所とスケールは、長方形の境界ボックスで示されます。シングルショットマルチボックス検出器 (SSD) フレームワークが使用され、VGGResNet の 2 つの基本ネットワークがサポートされます。ネットワークはゼロからトレーニングすることも、ImageNet データセットで事前トレーニングされたモデルでトレーニングすることもできます。

オブジェクト検出アルゴリズムの入出力インターフェイス

SageMaker AI オブジェクト検出アルゴリズムは、ファイルモードでのトレーニング用に RecordIO (application/x-recordio) とイメージ (image/png、、application/x-image) の両方のコンテンツタイプをサポートしimage/jpeg、パイプモードでのトレーニング用に RecordIO (application/x-recordio) をサポートします。ただし、RecordIO ファイルを作成せずに、拡張マニフェスト形式を使用して、イメージファイル (image/pngimage/jpegapplication/x-image) を使用してパイプモードでトレーニングすることもできます。Amazon SageMaker AI オブジェクト検出アルゴリズムに推奨される入力形式は、Apache MXNet RecordIO です。ただし、.jpg または .png の形式で RAW イメージを使用することもできます。このアルゴリズムでは、推論に application/x-image のみがサポートされます。

注記

既存の深層学習フレームワークとの相互運用性を維持するために、これは他の Amazon SageMaker AI アルゴリズムで一般的に使用される protobuf データ形式とは異なります。

データ形式の詳細については、オブジェクト検出サンプルノートブックを参照してください。

RecordIO 形式でトレーニングする

トレーニングに RecordIO 形式を使用する場合は、CreateTrainingJob リクエストの InputDataConfig パラメータの値としてトレーニングチャネルと検証チャネルの両方を指定します。1 つの RecordIO (.rec) ファイルを train チャネルで指定し、もう 1 つの RecordIO ファイルを validation チャネルで指定します。両方のチャネルのコンテンツタイプを application/x-recordio に設定します。RecordIO ファイルを生成する方法の例は、オブジェクト検出サンプルノートブックにあります。MXNet の GluonCV のツールを使用して、PASCAL ビジュアルオブジェクトクラスCommon Objects in Context (COCO) などの一般的なデータセット用の RecordIO ファイルを生成することもできます。

イメージ形式を使用してトレーニングする

トレーニングにイメージ形式を使用する場合は、trainvalidationtrain_annotationvalidation_annotation チャネルを CreateTrainingJob リクエストの InputDataConfig パラメータの値として指定します。トレーニングチャネルおよび検証チャネルの個々のイメージデータ (.jpg または .png ファイル) を指定します。注釈データには、JSON 形式を使用できます。train_annotation チャネルと validation_annotation チャネルで対応する .json ファイルを指定します。イメージタイプに基づいて、4 つすべてのチャネルのコンテンツタイプを image/png または image/jpeg に設定します。データセットに .jpg と .png の両方のイメージが含まれている場合は、コンテンツタイプ application/x-image を使用することもできます。次は、.json ファイルの例です。

{ "file": "your_image_directory/sample_image1.jpg", "image_size": [ { "width": 500, "height": 400, "depth": 3 } ], "annotations": [ { "class_id": 0, "left": 111, "top": 134, "width": 61, "height": 128 }, { "class_id": 0, "left": 161, "top": 250, "width": 79, "height": 143 }, { "class_id": 1, "left": 101, "top": 185, "width": 42, "height": 130 } ], "categories": [ { "class_id": 0, "name": "dog" }, { "class_id": 1, "name": "cat" } ] }

各イメージは注釈用に .json ファイルを必要とし、.json ファイルは対応するイメージと同じ名前である必要があります。上記の .json ファイルの名前は "sample_image1.json" になります。注釈 .json ファイルには 4 つのプロパティが含まれています。"file" プロパティはイメージファイルの相対パスを指定します。例えば、トレーニングイメージとそれに対応する .json ファイルが s3://your_bucket/train/sample_image and s3://your_bucket/train_annotation に保存されている場合、train チャネルと train_annotation チャネルのパスを、それぞれ s3://your_bucket/train and s3://your_bucket/train_annotation に指定します。

.json ファイルでは、sample_image1.jpg という名前のイメージの相対パスは sample_image/sample_image1.jpg になります。"image_size" プロパティはイメージ全体の寸法を指定します。SageMaker AI オブジェクト検出アルゴリズムは現在、3 チャネルイメージのみをサポートしています。"annotations" プロパティは、イメージ内のオブジェクトのカテゴリと境界ボックスを指定します。各オブジェクトには、"class_id" インデックスと 4 つの境界ボックス座標 ("left""top""width""height") によって注釈が付けられています。"left" (x 座標) と "top" (y 座標) の値は、境界ボックスの左上隅を表します。"width" (x 座標) と "height" (y 座標) の値は、境界ボックスの大きさを表します。原点 (0, 0) は、イメージ全体の左上隅です。1 つのイメージ内に複数のオブジェクトがある場合は、すべての注釈を単一の .json ファイルに含める必要があります。"categories" プロパティには、クラスインデックスとクラス名の間のマッピングが保存されます。クラスインデックスは連続して番号付けされ、番号付けは 0 から始まります。注釈 .json ファイルの場合、"categories" プロパティはオプションです。

拡張マニフェストイメージ形式でトレーニングする

拡張マニフェスト形式を使用すると、RecordIO ファイルを作成しなくても、イメージファイルを使用してパイプモードでトレーニングを行うことができます。CreateTrainingJob リクエストの InputDataConfig パラメータの値として、トレーニングチャネルと検証チャネルの両方を指定する必要があります。形式を使用している間、イメージとそれに対応する注釈のリストを含む S3 マニフェストファイルを生成する必要があります。マニフェストファイル形式は、各行が 1 つのサンプルを表す JSON Lines 形式になっている必要があります。イメージは、その S3 の場所を指す 'source-ref' タグを使用して指定します。注釈は、CreateTrainingJob リクエストで指定されている "AttributeNames" パラメータ値の下に入力します。metadata タグの下に追加のメタデータを含めることもできますが、これらはアルゴリズムによって無視されます。次の例では、"AttributeNames["source-ref", "bounding-box"] リストに含まれています。

{"source-ref": "s3://your_bucket/image1.jpg", "bounding-box":{"image_size":[{ "width": 500, "height": 400, "depth":3}], "annotations":[{"class_id": 0, "left": 111, "top": 134, "width": 61, "height": 128}, {"class_id": 5, "left": 161, "top": 250, "width": 80, "height": 50}]}, "bounding-box-metadata":{"class-map":{"0": "dog", "5": "horse"}, "type": "groundtruth/object-detection"}} {"source-ref": "s3://your_bucket/image2.jpg", "bounding-box":{"image_size":[{ "width": 400, "height": 300, "depth":3}], "annotations":[{"class_id": 1, "left": 100, "top": 120, "width": 43, "height": 78}]}, "bounding-box-metadata":{"class-map":{"1": "cat"}, "type": "groundtruth/object-detection"}}

オブジェクト検出アルゴリズムをトレーニングする場合、入力ファイルの "AttributeNames" の順番が重要になります。パイプ指定されたデータは特定の順で受け入れられます。image が最初になり、その後に annotations が続きます。したがって、この例の「AttributeNames」では、"source-ref" が最初になり、その後に "bounding-box" が続きます。拡張マニフェストでオブジェクト検出を使用する場合は、RecordWrapperType パラメータの値を "RecordIO" として設定する必要があります。

拡張マニフェストファイルの詳細については、トレーニングジョブの拡張マニフェストファイルを参照してください。

段階的トレーニング

SageMaker AI で以前にトレーニングしたモデルのアーティファクトを使用して、新しいモデルのトレーニングをシードすることもできます。段階的トレーニングでは、同じモデルまたは類似のデータを使用して新しいモデルをトレーニングする際のトレーニング時間が短縮されます。SageMaker AI オブジェクト検出モデルは、SageMaker AI でトレーニングされた別の組み込みオブジェクト検出モデルでのみシードできます。

事前トレーニング済みモデルを使用するには、CreateTrainingJob リクエストで、InputDataConfig パラメータに ChannelName を "model" と指定します。モデルチャネルの ContentTypeapplication/x-sagemaker-model に設定します。モデルチャネルにアップロードする新しいモデルと事前トレーニング済みモデルの両方の入力ハイパーパラメータの設定は、base_network および num_classes 入力パラメータの設定と同じである必要があります。これらのパラメータはネットワークアーキテクチャーを定義します。事前トレーニング済みのモデルファイルには、SageMaker AI によって出力された圧縮されたモデルアーティファクト (.tar.gz 形式) を使用します。入力データには、RecordIO 形式またはイメージ形式を使用できます。

段階的トレーニングの詳細とその使用方法については、Amazon SageMaker AI で増分トレーニングを使用するを参照してください。

オブジェクト検出アルゴリズムの EC2 インスタンスに関する推奨事項

オブジェクト検出アルゴリズムは P2、P3、G4dn、G5 GPU インスタンスファミリーをサポートします。大きなバッチサイズのトレーニングにはメモリが多い GPU インスタンスを使用することをお勧めします。分散型トレーニングでは複数 GPU および複数マシン設定でオブジェクト検出アルゴリズムを実行することができます。

推論には、CPU インスタンス (C5、M5 など) と GPU インスタンス (P3、G4dn など) の両方を使用できます。

オブジェクト検出サンプルノートブック

SageMaker AI オブジェクト検出アルゴリズムを使用して でモデルをトレーニングおよびホストする方法を示すサンプルノートブックの場合

シングルショットマルチボックスディテクターアルゴリズムを使用した Caltech Birds (CUB 200 2011) データセットについては、Amazon SageMaker AI Object Detection for Bird Species」を参照してください。SageMaker AI でサンプルを実行するために使用できる Jupyter Notebook インスタンスを作成してアクセスする手順については、「」を参照してくださいAmazon SageMaker ノートブックインスタンス。ノートブックインスタンスを作成して開いたら、SageMaker AI Examples タブを選択して、すべての SageMaker AI サンプルのリストを表示します。オブジェクト検出アルゴリズムを使用したオブジェクト検出例のノートブックは、「Introduction to Amazon Algorithms」セクションにあります。ノートブックを開くには、その [Use (使用)] タブをクリックして [Create copy (コピーを作成)] を選択します。

Amazon SageMaker AI オブジェクト検出アルゴリズムの詳細については、次のブログ記事を参照してください。