Giter Club home page Giter Club logo

goturn_training's Introduction

GOTURN Training Toolkit

This is the code for training of GOTURN tracker implemented inside OpenCV.

Original GOTURN paper:

Learning to Track at 100 FPS with Deep Regression Networks,
David Held, Sebastian Thrun, Silvio Savarese,
European Conference on Computer Vision (ECCV), 2016 (In press)

Installation

Install dependencies:

sudo apt-get install libopencv-dev

Building HDF5 dataset

For GOTURN training, first HDF5 dataset shold be generated. It can be done by simple function call:

buildDB()

By default it generates 10 HDF5 datasets every with 500x10 samples (10 crop samples per image as proposed by authors). As alternative more low-level function can be used:

buildH5Datasets(datasetName, numberOfSamples)

Training network

GOTURN training requires a "bvlc_reference_caffenet.caffemodel" for GOTURN network weights inialization:

http://dl.caffe.berkeleyvision.org/bvlc_reference_caffenet.caffemodel

Training is launched by next line:

trainNet()

All hyperparameters are configured in goturnSolver.prototxt, for more details refer Caffe documentation.

Evaluate the tracker

In order to visualize results, GOTURN tracker can be tested on test dataset:

buildH5Datasets("D:/ALOV300++/trainDataset.h5", 10);
testNet("goturn_iter_30000.caffemodel");

First command generate a new small test dataset, and second launching a test procedure with visualization.

Pretrained GOTURN model

Also there is pretrained GOTURN model is available in OpenCV_extra repository

https://github.com/opencv/opencv_extra

goturn_training's People

Contributors

auron-x avatar

Watchers

 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.