「家の鍵、閉めたっけ?」をAIでなくそう!


当記事は以下のパクリです。

「エアコン消したっけ?」をAIでなくそう!
https://qiita.com/it31415/items/f3726b547795904837ef

概要

AutoMLをつかって、家の鍵が閉まってるかを見分けるAIをつくりました。

はじめに

あんまり有用性はないけど、AutoMLのポテンシャルを測るにはいいかと思います。「エアコン消したっけ?」をAIでなくそう!の記事書いた人がKeras好きそうなので AutoKeras 使ってほしいですね。

準備

今回は3種類のドアの鍵で試してみました。

写真を撮る

自分の家の鍵の開閉を見分けたいので、開閉の写真をそれぞれ100枚ずつぐらい撮りました(連写)。
カメラを取り付けて定位置から認識させようと思っているため、そこまでロバストにする必要性はないと考えています。

ドアの種類 Open Close
種類1
種類2
種類3
door_lock/
  ├ open/
  │    └ image1.jpg
  │    └ image2.jpg
  │         ︙
  │    └ image100.jpg
  └ close/
       └ image1.jpg
       └ image2.jpg
            ︙
       └ image100.jpg
door_lock.zip

door_lock は任意の好きな名前です。 open, close は画像のラベル名です。あとは door_lock フォルダを圧縮してください。

実行環境

TPUが無料で使えるAutoML Vision上で実行しました。TPUをつかうとGPUより数十倍も学習速度が速くなりとても便利です。

データセットの作成

データセット名は任意の好きな名前をいれてください。先程作成したzipファイルを選択してください。

学習

無料で学習できます。

結果

200枚ぐらいの画像で学習してテストしたところ、正解率は99.9%でした。

種類1

種類2

種類3

おわりに

データセットを こちら に用意しましたのでご自由にお使いください。鍵が開いてるか閉じてるかだけのちがいだったので、見分けるのは難しいかなあと思っていましたが、結構いい精度になりました。ほとんど同じ位置からの写真だったので簡単だったのかもしれません。