/classify-dogcat_django

【Public (forked)】画像判定アプリ

Primary LanguagePython

画像判定アプリ

1. 概要

犬 or 猫どちらに該当するかを判定してくれるアプリ

2. 環境

  • Python 3.6.10
  • Django 3.0.4

3. 大まかな手順

1. 仮想環境にてTensorFlowの導入

$ conda create -n djangoai tensorflow
$ source activate djangoai

2. データの収集

$ pip install flickrapi

→犬と猫を取得するためのスクリプトdl.pyを作成

  • 収集したデータをNumpy配列に変換するgenerate_data.pyを作成 →トレーニングデータとテストデータに分割したimagefiles.npyを吐き出す。

3. モデル構築

  1. データをロードする
  2. モデルを定義する
  3. 最適化手法を定義する
  4. トレーニングする
  5. 精度評価する

vgg16_transfer.py を実行し、vgg16_transfer.h5(h5形式)としてモデルを保存する

4. Webアプリ開発

1. Djangoの導入

$ pip install Django

2. プロジェクトとアプリケーションの作成

$ django-admin startproject aiapps
$ cd aiapps
$ python manage.py startapp dogcat

3. Webサイト構築 Django Bootstrap4モジュールのインストール

$ pip install django-bootstrap4

以降、ルーティング設定やビュー設定など、Webアプリ化を行う。