Code Monkey home page Code Monkey logo

wpal-network's Introduction

Weakly-supervised Pedestrian Attribute Localization Network

By Ken Yu, under guidance of Dr. Zhang Zhang and Prof. Kaiqi Huang.

Weakly-supervised Pedestrian Attribute Localization Network (WPAL-network) is a Convolutional Neural Network (CNN) structure designed for recognizing attributes from objects as well as localizing them. Currently it is developed to recognize attributes from pedestrians only, using the Richly Annotated Pedestrian (RAP) database or PETA database.

Installation

  1. Clone this repository

    # Make sure to clone with --recursive
    git clone --recursive https://github.com/kyu-sz/Weakly-supervised-Pedestrian-Attribute-Localization-Network.git
  2. Build Caffe and pycaffe

    This project use python layers for input, etc. When building Caffe, set the WITH_PYTHON_LAYER option to true.

    WITH_PYTHON_LAYER=1 make all pycaffe -j 8
  3. Download the RAP database

    To get the Richly Annotated Pedestrian (RAP) database, please visit rap.idealtest.org to learn about how to download a copy of it.

    It should have two zip files.

    $RAP/RAP_annotation.zip
    $RAP/RAP_dataset.zip
    
  4. Unzip them both to the directory.

    cd $RAP
    unzip RAP_annotation.zip
    unzip RAP_dataset.zip
  5. Create symlinks for the RAP database

    cd $WPAL_NET_ROOT/data/dataset/
    ln -s $RAP RAP

Usage

To train the model, first fetch a pretrained VGG_CNN_S model by:

./data/scripts/fetch_pretrained_vgg_cnn_s_model.sh

Then run experiment script for training:

./experiments/example/VGG_CNN_S/train_vgg_s_rap_0.sh

Experiment script for testing is also available:

./experiments/examples/VGG_CNN_S/test_vgg_s_rap.sh

Acknowledgements

The project layout and some codes are derived from Mr. Ross Girshick's py-faster-rcnn.

We use VGG_CNN_S as pretrained model. Information can be found on Mr. K. Simonyan's Gist. It is from the BMVC-2014 paper "Return of the Devil in the Details: Delving Deep into Convolutional Nets":

Return of the Devil in the Details: Delving Deep into Convolutional Nets
K. Chatfield, K. Simonyan, A. Vedaldi, A. Zisserman
British Machine Vision Conference, 2014 (arXiv ref. cs1405.3531)

wpal-network's People

Contributors

kyu-sz avatar

Watchers

John D. Pope avatar  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.