S

SeetaBase

A platform implementing popular base vision models.

06e7ac2e Initial repository · by Ting PAN
Name Last Update
configs/resnet Loading commit data...
csrc Loading commit data...
data/datasets Loading commit data...
seetabase Loading commit data...
tools Loading commit data...
.clang-format Loading commit data...
.flake8 Loading commit data...
.gitignore Loading commit data...
LICENSE Loading commit data...
MODEL_ZOO.md Loading commit data...
README.md Loading commit data...
setup.py Loading commit data...
version.txt Loading commit data...

SeetaBase

SeetaDet is a platform implementing popular base vision models.

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 seetabase && python setup.py build

Install From Source

Clone this repository to local disk and install:

cd seetabase && python setup.py install

Install From Git

You can also install it from remote repository:

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

Quick Start

Train a model

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

We have provided the default YAML examples into configs.

Test a 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