SeetaSeg

SeetaSeg is a platform implementing popular segmentation algorithms.

This repository is based on seeta-dragon, while the style of codes is torch.

Installation

Build From Source

If you prefer to develop modules as well as running experiments, following commands will build but not install to site-packages:

cd seetaseg && python setup.py build

Install From Source

Clone this repository to local disk and install:

cd seetaseg && python setup.py install

Install From Git

You can also install it from remote repository:

pip install git+https://gitlab.seetatech.com/seetaresearch/seetaseg.git

Quick Start

Train a segmentation models

cd tools
python train.py --cfg <MODEL_YAML>

We have provided the default YAML examples into configs.

Test a segmentations model

cd tools
python test.py --cfg <MODEL_YAML> --exp_dir <EXP_DIR> --iter <ITERATION>

Benchmark and Model Zoo

Results and models are available in the Model Zoo.

License

BSD 2-Clause license