Code Monkey home page Code Monkey logo

detect_people's Introduction

人検知のための機械学習モデル構築

画像内に人がいるかどうかを判定するための機械学習model構築方法。
学習データセットはMS COCOを使用しています。(学業以外の用途で使用しない方がよい。著作権関連注意)
学習にはtensorflowを使用し、Mobilenetからの転移学習を行う。 ラズパイで動作させるために、tflite用のモデるを作成する。

必要ライブラリ

anacondaで環境作るのおすすめです。
ライブラリに関してはenv/env.ymlに記載してあります。次のコマンドで環境を作ってください。

conda create -n people -f env/env.yml
conda activate people

フォルダ説明

  • eval.py : 学習済みモデルを評価するファイル。あるディレクトリにある画像データを読み込み、学習済みモデルで判定する。
  • train_mobile.py : 学習を行うファイル。学習データセットMSCOCOを読み込み、Mobilenetから転移学習を行う。学習が終わったら、tflite用のモデルも作成する。
  • make_data.py : SCOCOからデータセットをダウンロードし、そのデータセットから人が映っている画像とそうでない画像を抽出するを行うファイル。
  • quntize.py : 学習モデルを量子化するファイル。学習済みモデルを読み込み、量子化を行う。多分エラーが起きるから使わない方がいい。
  • show.py :データセット内の画像を確認するファイル。

使い方

  1. make_data.pyを実行し、学習データセットをダウンロードする。
  2. train_mobile.pyを実行し、学習を行う。学習が行われると、model_mobile_good.h5というtensorflow用のモデルが作成される。そのモデルをtflite用に変換したmodel_mobile.tfliteというモデルも作成される。モデルはmodelディレクトリに保存される。
  3. eval.pyを実行し、学習済みモデルを評価する。あるディレクトリに評価したい画像を適当に入れとく。
  4. train_mobile.pyで作成されたtflite用のモデルをラズパイに転送し、実行する。

detect_people's People

Watchers

YoshiyaOsm avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.