iisyos / z_fighter_server

z_fighter_server

z_fighter のためのバックエンドです。 画像urlを受け取り、機械学習の結果を返します。

使用パッケージ

Turi Create

Image Classification を使用し、画像類似度判定のモデルを作成

FastAPI

PythonでAPIサーバーをたてるために使用

Serverless Framework

• ECRにimageをpush
• lambdaにデプロイ
• APIGatewayに紐付け 等をCIするために使用

使用法

1. コンテナのビルド

$ docker-comopose up -d --build

2.学習データの収集

$ docker-comopose exec lambda_api /bin/zsh
$ python src/script/zFIghtersImageCollector.py

3.学習モデルの作成

$ python src/script/createModel.py

4.localでAPI起動

$ python src/app.py

5.テスト

$ curl -X 'POST' \
  'http://localhost:8081/v1/predict' \
  -H 'accept: application/json' \
  -H 'Content-Type: application/json' \
  -d '{
  "url": "url"
}'