Segmentation with deep learning
Branch: master
Clone or download
erogol Merge pull request #4 from eriche2016/master
Update camvid.lua bug for file path reading
Latest commit adc170e Dec 30, 2016
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data
exp_model
models
train
train_scripts
.gitignore
README.md
opts.lua
run.lua
test.lua
train.lua
training_for_new_dataset.txt

README.md

Seg-Torch for Image Segmentation with Torch

This work was sparked by my personal research on simple segmentation methods based on deep learning. It is the harvest of two great predecessors;

However this code includes radical differences (such as data loading, augmentation, memory optimization) and it has more generic type of implementation suitable for use in any custom project. You only need to modify data-loader files data/custom-gen.lua and data/custom.lua.

Be warned this is susceptible to bugs. Any pull request is appreciated.

Check train_scripts/ for example execution.

Models

  • SegNet: Very simple encoder-decoder network, segmenting end2end
  • EroNet: Very similar but it chops Batch-Normalization and uses ELU activation. It is lower in accuracy but faster in training.

Example Results

exp_model/ includes a proof of concept on CamVid dataset. If you compare the results with the real-project this implementation has higher values interestingly (at least for me) :) .

Model will be shared on Dropbox, as soon as I find some time to do so.