Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
erogol committed Dec 24, 2016
1 parent c03bd72 commit 4cc2fee
Showing 1 changed file with 10 additions and 8 deletions.
18 changes: 10 additions & 8 deletions README.md
@@ -1,21 +1,23 @@
Seg-Torch for Image Segmentation with Torch
# 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;

https://github.com/e-lab/ENet-training
https://github.com/fedor-chervinskii/segnet-torch
(real project page) http://mi.eng.cam.ac.uk/projects/segnet/
* https://github.com/e-lab/ENet-training
* https://github.com/fedor-chervinskii/segnet-torch
* (real project page) http://mi.eng.cam.ac.uk/projects/segnet/

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
# 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.

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
# 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) :) .

Expand Down

0 comments on commit 4cc2fee

Please sign in to comment.