Git Product home page Git Product logo

chelovekhe / boundary-detection-via-convolution-deconvolution-neural-network-with-bma Goto Github PK

View Code? Open in Web Editor NEW

This project forked from wuga214/boundary-detection-via-convolution-deconvolution-neural-network-with-bma

0.0 1.0 0.0 164.91 MB

Boundary-Detection-via-Convolution-Deconvolution-Neural-Network

Home Page: http://wuga214.github.io/Boundary-Detection-via-Convolution-Deconvolution-Neural-Network-with-BMA/

License: MIT License

Python 99.59% MATLAB 0.41%

boundary-detection-via-convolution-deconvolution-neural-network-with-bma's Introduction

Boundary-Detection-via-Convolution-Deconvolution-Neural-Network

Code Description

This code was generated to implement deconvolution neural network under Theano background. We found that there was no existing code to actually implement deconvolution neural network under Theano and its extension codes, even through many papers claims that they are using convolution-deconvolution structure. The only one workable code in python is under caffe environment to solve segmentation problem, in which, they actually modified caffe foundamentally. We know that most of the computer vision research is leaded in Matlab, but, for fast training using servers, python is still the optimal choice.

Boundary detection

We used our network to do boundary detection task. The network does not apply optimal configuration in this experiment, and we only want to show the code is correct and workable. We highly suggest you modify the code to achieve your own task.

We also introduced a novel structure that can combine multiple network configurations, please see our paper in this positroy. Paper

This is one comparision of our network(second from left) with benchmark algorithm(right most).

png

In relatively noisy background, our implementation captures too much noise. But it is due to our network doesn't using optimal configuration. You can make it better. png

System Requirement

You need to install Theano and Cuda, that is all

Contact

Ga Wu, email, Kelley Engineering Center, Oregon State University

Xu Xu, email, Kelley Engineering Center, Oregon State University

Code Updating

Data Operation Code Updated[03-05-2016]

  1. Python can load images into memory and create list with element size : 481x321x3, This format can be directly used in Keras

  2. Label data was in the form of *.mat. We extracted single ground truth from it into form of csv txt

Matlab Script 3. CSV label file loading in python is now functional. 4. Label reshape function works, now the data & label can be directly used by keras to do prediction!

Layer Code Updated[03-07-2016]

  1. Max Pooling layer redefined under theano environment. Now the function output maxed matrix and mask!

  2. Reversed Max Pooling layer is functional, it can recover the 'original' shape of matrix through max pooling mask saved

  3. Convolution layer is functional, just warp of con2d in theano

Network Class Updated[03-08-2016]

  1. Convolution task can be worked on our code now

  2. Added Keras Backend. Now the convolution layer can accept padding!!!!

Accelerated version of Code[03-09-2016]

  1. The code is able to run on GPU now! Max pooling and reverse pooling can run fast enough when training!

Complete training and evaluating[03-10-2016]

  1. Forward Propagation works

  2. Loss function works(binary cross entropy)

  3. Data now in cPickle format(easier to load in correct shape)

  4. Backpropagation works, but the error function may has some glitches... Always show 100% error

  5. Code can run on Pelican server now

Bug Free Code!!!Yeah!![03-10-2016]

  1. The code is fully functional and without any bug. Training and testing result seems encouraging! png

boundary-detection-via-convolution-deconvolution-neural-network-with-bma's People

Contributors

wuga214 avatar xuxucmkox avatar

Watchers

 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.