5
2

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 5 years have passed since last update.

Golang + AWS Rekognitionで画像解析

Last updated at Posted at 2017-07-22

AWS Rekognitionで画像の中に含まれているオブジェクトを解析するAPIがあります。それをGolangで使ってみました。

import (
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/rekognition"
	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/credentials"
	"fmt"
	"io/ioutil"
)

func MyDetectLabels() {
	imagePath := "hamburger.jpg"
	//
	// 画像をbyte配列にする
	// SDKのコメントにbase64にエンコードせよとか書いてあるけどしなくてよい
	// むしろしたらエラーになります・・・
	//
	imageBytes, err := ioutil.ReadFile(imagePath)
	if err != nil {
		fmt.Println(err.Error())
		return
	}
	//
	// Imageオブジェクトにbyte配列を指定
	//
	image := &rekognition.Image {
		Bytes: imageBytes,
	}
	//
	// IAMユーザーのアクセスキーとシークレットを指定し、セッションを作る
	// IAMユーザーはAmazonRekognitionReadOnlyAccessポリシーを持つこと
	//
	accessKeyId := "*****"
	secretAccessKey := "*****"
	sess := session.Must(session.NewSession(&aws.Config{
		Region: aws.String("us-west-2"),
		Credentials: credentials.NewStaticCredentials(accessKeyId, secretAccessKey, ""),
	}))
	// 解析オブジェクトを生成
	reko := rekognition.New(sess)
	//
	// 画像解析し、画像に含まれている情報を返す
	//
	input := &rekognition.DetectLabelsInput{}
	input.SetImage(image)
	input.SetMaxLabels(10) // 最大取得ラベル数
	input.SetMinConfidence(50) // 取得する最低信頼度(%)
	output, err := reko.DetectLabels(input)
	if err != nil {
		fmt.Println(err.Error())
		return
	}
	for _, label := range output.Labels {
		fmt.Printf("ラベル名:%s 信頼度:%f\n", *label.Name, *label.Confidence)
	}
}

DetectLabelsの機能しか使わないなら、IAMのユーザーポリシーは"rekognition:DetectLabels"のみでも良いはず。

hum02.jpg

ビールと一緒にハンバーガー食べた時の画像の解析結果。

ラベル名:Burger 信頼度:98.644783
ラベル名:Food 信頼度:98.644783
ラベル名:Bottle 信頼度:96.044090
ラベル名:Alcohol 信頼度:82.943893
ラベル名:Beer 信頼度:82.943893
ラベル名:Beverage 信頼度:82.943893
ラベル名:Drink 信頼度:82.943893
ラベル名:Glass 信頼度:57.666523

ハンバーガーはもちろん、ビールも解析できている。ボトルもグラスも。しかし、SetMinConfidenceを0にしてみても、ケチャップは検出できなかった。でも何の食べ物が写っているかわかるだけでもすごいし、こんな高度な画像解析が個人開発者でも使えるようになったのもすごい。いろいろ使い道がありそうですね。

ukiuki256.png

5
2
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
5
2

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?