https://github.com/yomi4486/cat_discrimination
猫の種類の判別を行うAPIです。
https://github.com/yomi4486/cat_discrimination
adam-optimizer ai fastapi keras python3 tensorflow
Last synced: about 2 months ago
JSON representation
猫の種類の判別を行うAPIです。
- Host: GitHub
- URL: https://github.com/yomi4486/cat_discrimination
- Owner: yomi4486
- Created: 2024-05-06T13:31:16.000Z (about 2 years ago)
- Default Branch: main
- Last Pushed: 2024-11-28T06:36:23.000Z (over 1 year ago)
- Last Synced: 2025-02-07T03:25:04.266Z (over 1 year ago)
- Topics: adam-optimizer, ai, fastapi, keras, python3, tensorflow
- Language: Python
- Homepage:
- Size: 113 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# 画像から猫の種類を判別するAPIです
Intel Xeon E3-1231v3(Core i7-4770と同等性能)で、0.3秒で判別できます。
モデルはオックスフォード大学が公開しているペットのデータセットを利用して訓練しました。
https://www.robots.ox.ac.uk/~vgg/data/pets/
必要なライブラリのインストールは下記コマンドでできます
```sh
make install
```
下記コマンドでimagesフォルダ内に配置した画像を仕分け（アノテーションを作成）して、学習を行います。
```sh
make train
```
h5モデルが問題なく使用可能な状態になっていれば、下記コマンドでAPIの起動ができます。
```sh
make run
```
`test.py`でAPIの動作確認ができます。判別したい画像を`test_img`フォルダに配置して、引数にその画像の名前を指定して実行すれば結果が瞬時に帰ってきます。
テスト用の画像が1枚入っているので、下記コマンドで試してみてください。
```sh
python3 test.py 1.jpeg
```
参考記事
https://qiita.com/study_ryoma/items/a61ba10a3cd79acc5749