Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

https://github.com/dgriffiths3/ml_segmentation

Machine learning semantic segmentation - Random Forest, SVM, GBC
https://github.com/dgriffiths3/ml_segmentation

image-segmentation machine-learning python random-forest segmentation support-vector-machine

Last synced: about 2 months ago
JSON representation

Machine learning semantic segmentation - Random Forest, SVM, GBC

Lists

README

        

# Machine Learning - Image Segmentation

Per pixel image segmentation using machine learning algorithms. Programmed using the following libraries: Scikit-Learn, Scikit-Image OpenCV, and Mahotas and ProgressBar. Compatible with Python 2.7+ and 3.X.

### Feature vector

Spectral:

* Red
* Green
* Blue

Texture:

* Local binary pattern

Haralick (Co-occurance matrix) features (Also texture):

* Angular second moment
* Contrast
* Correlation
* Sum of Square: variance
* Inverse difference moment
* Sum average
* Sum variance
* Sum entropy
* Entropy

### Supported Learners

* Support Vector Machine
* Random Forest
* Gradient Boosting Classifier

### Example Usage

python train.py -i -l -c -o

python inference.py -i -m -o

python evaluation.py -i -g [-m]

### Example Output

![Example Output](pots/image_small.png)