Giter Club home page Giter Club logo

deeppose's Introduction

Deep Pose

I think more tuning is possible. If you have some advice, please tell me!

Features

  • Chainer implementation
  • Subsequent stages
  • Image viewer on web browsers. (Flask and flask-socketio are needed)

Supported Datasets

  • FLIC
  • LPS

Testing Environments

Ubuntu 14.04

  • Python 2.7
  • Chainer 1.9.1
  • OpenCV 2.4.8
  • Flask 0.11.1
  • Flask_SocketIO 2.4

Arch Linux

  • Python 3.5
  • Chainer 1.9.1
  • OpenCV 3.1.0
  • Flask 0.10.1
  • Flask_SocketIO 2.2

Windows 7 (not recommend)

  • Python 2.7
  • Chainer 1.9.1
  • OpenCV 3.1.0
  • Flask 0.11.1
  • Flask_SocketIO 2.5

Training

First, download FLIC FULL and FLIC PLUS to some directory, and set the path to settings.json. And also fix CASCADE_PATHS to detect faces for your environment.

To start training, please execute the following command.

./scripts/train.py --stage 0

For subsequent stage training, --joint_idx argument is needed.

./scripts/train.py --stage 1 --joint_idx 0
./scripts/train.py --stage 1 --joint_idx 1
./scripts/train.py --stage 2 --joint_idx 0  # and so on

--resume argument is also supported.

To check current training state, please open http://localhost:8889/ (port number can be changed by settings.json). The error rate graph and visualized images can be seen.

If you want to use GPU, please set GPU parameter in settings.json to a positive number.

Use trained models

Execute the following command, and open http://localhost:8889/.

./scripts/use_model.py

Settings is common with training (settings.json).

Results

First stage

Subsequent stages are training now.

Async Mode Setting

This project uses Python threading or multiprocessing package and it can be configured by ASYNC_MODE in settings.json. On Linux process mode is better due to the speed, but on Windows only thread mode is valid.

TODO

  • Tune training parameters (learning rate, bounding box sigma and so on).
  • Replace multiprocess.Queue and Event to threading's ones on the thread mode.

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.