More than 3 years have passed since last update.

5分で出来る!? FastAPIとOpenCVで顔検出APIを作成してHerokuで公開

Posted at 2020-06-09

はじめに

こんにちは。本記事は自サイトのものを書き写したものになりますが、広く使ってもらえそうな内容だと思ったので思い切ってQiitaに投稿してみました。

さて、コンピュータビジョンを勉強していて、この技術をどのように具体的に使えるようにするのが良いか模索しています。その中でAPIを作るという手法は色々な環境・言語で共通モジュール的に使えて良い案の１つと思います。勿論デメリットもありますが。
APIを簡単に作れるPythonフレームワークが無いか探していたところ、FastAPIというものを見つけたので、FastAPIを使って顔検出APIを作成し、それをせっかくなのでHerokuで公開まで一気に行きたいと思います。

さて、5分で出来るでしょうか（笑）タイマースタート！

FastAPIとは

FastAPIは、標準のPython型ヒントをベースにPython 3.6+でAPIを構築するための、モダンで高速（ハイパフォーマンス）なWebフレームワークです。とても使いやすいフレームワークなので是非とも沢山の人に利用してほしいです。
公式サイトはhttps://fastapi.tiangolo.com/です。

顔検出APIとは？

顔検出したい画像をアップロードすると、JSONで検出できた数や矩形座標を返すAPIを作成します。その結果を利用して、よくある感じで枠を付けてみるとこんな感じになります。

画像：http://gahag.net/

前提

予め以下を実施して下さい。

Herokuユーザ登録済み(Freeプラン)
Githubユーザ登録済み

以下は実施した方がいいと思いますが、今回は必須ではありません。

デバッグ用にPCにPythonインストール
VSCodeなどのデバッグ環境インストール

手順

GitHubリモートリポジトリ作成
Herokuにデプロイ
API完成!?

①GitHubリモートリポジトリ作成

ソースコードの詳細は後程自サイトにて書いていきたいと思います。今回使用するソースコードはGithubにあります。自分のGithubリモートリポジトリを作成してソースを置いて下さい。

構成は以下のようになっています。

.
├── cascades
│   └── haarcascade_frontalface_default.xml
├── detect
│   └── detect_face.py
├── .gitignore
├── Aptfile
├── Procfile
├── const.py
├── main.py
├── requirements.txt
└── women.jpg

「cascades/haarcascade_frontalface_default.xml」は顔検出できるカスケード識別器の学習済みファイルです。「detect/detect_face.py」には顔検出するロジックが書いてあります。「.gitignore」はgit管理したくないものを書いてます。「Aptfile」はHerokuでOpenCVを使うのに必要なファイルです。詳細は後述しています。「Procfile」はHerokuでFastAPIを動かすのに必要なファイルです。「const.py」はCONSTを使うのに楽なので使用しています。「main.py」はuvicorn により実行されるファイルです。「requirements.txt」は依存ライブラリを管理するファイルです。HerukuでDeployするために必要です。「women.jpg」はサンプル用画像です。

②Herokuにデプロイ

さて、ソースの準備が出来たので、HerokuでAPI公開の準備をします。
まずはCreatNewAppをします。

アプリ名を付けます。

Deployタブを選択して、自身のGithubアカウントと接続した後、デプロイするリポジトリを選択します。

OpenCVは「opencv-contrib-python」パッケージをインストールしていますが、これだけだとimportエラーが出てしまいます。「ImportError: libSM.so.6: cannot open shared object file: No such file or directory」そのため、「buildpacksの追加」と「Aptfileファイル作成」が必要になります。Aptfileはソース一式の中に用意してありますので、Herokuでbuildpacksの追加のみ実施します。SettingsタブのBuildpacksにて「https://github.com/heroku/heroku-buildpack-apt」を追加します。