[Amazon Rekognition] Custom Labelsのデータセットを高速に作成するアプリを作ってみました

SIN

2020.03.13

この記事は公開されてから1年以上経過しています。情報が古い可能性がありますので、ご注意ください。

1 はじめに

CX事業本部の平内(SIN)です。

Amazon Rekognition の Custom Labelsでは、少数の画像セットで、ビジネスシーンに合せた独自のオブジェクトの検出が可能です。

そして、データセットの作成（ラベルの付加）には、独自に準備されたエディタや、SageMaker Ground Truth が利用可能です。どちらも、非常に優れたUIで、簡単にアノテーション作業が可能です。

しかし、大量のデータセットを生成するとなると、やはり手間がかかることは否定できません。

今回は、このデータセットの作成（写真撮影とアノテーション付与）が少しでも楽にならないかと言う事で、アプリを作成して見ました。

2 動作の様子

最初に、作成したアプリを使用している様子です。

Chromeでindex.htmlを開くと、下記のような画面になります。カメラは、USB接続されたWebカメラ(Logitech C920)を使用しています。

カメラで撮影した画像は、OpenCVで処理して　リアルタイムで矩形を検出します。表示された矩形がいい感じとなった時、「SAVE」ボタンを押すだけで、撮影とアノテーションの終わったManifestが生成されS3に送信されます。

S3に保存された画像と、マニフェストを指定して、Create Datasetすると、データセットの作成は完了です。

3 構成

構成は、以下のとおりです。

① 撮影したデータから、画像(jpeg)とManifestを生成します。
② SAVEボタンで、上記のデータが、S3に送信されます。
③ 上記S3バケットを指定してデータセットが作成できます。
④ S3バケットへのパーミッションは、Cognitoで付与されています。

4 環境構築

アプリの環境構築要領は、以下のとおりです。

(1) Github

コードは、Githubに置かれていますので、cloneして下さい。以降の作業が完了した時点で、index.htmlをchromeで開くことで利用可能です。

$ git clone https://github.com/furuya02/DataSetCreater.git
$ cd DataSetCreater

(2) S3

送信先として、バージニアのバケットが必要です。準備したバケット名をindex.htmlに設定して下さい。

index.html

const bucketName = "my-bucket-name";
const s3 = new S3(bucketName);

バケットには、CORSの設定が必要です。

CORSの例

<!--?xml version="1.0" encoding="UTF-8"?-->


*
GET
PUT
DELETE
3000
*

また、バケットに、Rakognitionからアクセスできるようにパーミッションの設定が必要です。以下は、一例ですが、バケット名を編集する必要があります。

Permitionの一例

{
"Version": "2012-10-17",
"Statement": [
{
"Sid": "AWSRekognitionS3AclBucketRead20191011",
"Effect": "Allow",
"Principal": {
"Service": "rekognition.amazonaws.com"
},
"Action": [
"s3:GetBucketAcl",
"s3:GetBucketLocation"
],
"Resource": "arn:aws:s3:::my-bucket-name"
},
{
"Sid": "AWSRekognitionS3GetBucket20191011",
"Effect": "Allow",
"Principal": {
"Service": "rekognition.amazonaws.com"
},
"Action": [
"s3:GetObject",
"s3:GetObjectAcl",
"s3:GetObjectVersion",
"s3:GetObjectTagging"
],
"Resource": "arn:aws:s3:::my-bucket-name/*"
},
{
"Sid": "AWSRekognitionS3ACLBucketWrite20191011",
"Effect": "Allow",
"Principal": {
"Service": "rekognition.amazonaws.com"
},
"Action": "s3:GetBucketAcl",
"Resource": "arn:aws:s3:::my-bucket-name"
},
{
"Sid": "AWSRekognitionS3PutObject20191011",
"Effect": "Allow",
"Principal": {
"Service": "rekognition.amazonaws.com"
},
"Action": "s3:PutObject",
"Resource": "arn:aws:s3:::my-bucket-name/*",
"Condition": {
"StringEquals": {
"s3:x-amz-acl": "bucket-owner-full-control"
}
}
}
]
}

なお、データセットを作成する際に、パーミッションの例が、コンソールに表示されますので、それをコピーして設定することも出来ます。

(3) Cognito

Cognitoで未認証でも利用可能なIdentityPoolIdを作成します。

作成したIdentityPoolIdをs3.jsに設定して下さい。

s3.js

AWS.config.credentials = new AWS.CognitoIdentityCredentials({
IdentityPoolId: 'us-east-1:xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx',
});

Roleには、S3バケットへのアクセス権を付与します。

※リソース名を作成したバケットに合わせてください。

{
"Version": "2012-10-17",
"Statement": [
{
"Effect": "Allow",
"Action": [
"mobileanalytics:PutEvents",
"cognito-sync:*"
],
"Resource": [
"*"
]
},
{
"Effect": "Allow",
"Action": [
"s3:PutObject",
"s3:GetObject",
"s3:DeleteObject",
"s3:ListBucket"
],
"Resource": [
"arn:aws:s3:::my-bucket-name",
"arn:aws:s3:::my-bucket-name/*"
]
}
]
}