Name Last Update
configs Loading commit data...
csrc Loading commit data...
scripts Loading commit data...
seetadet 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...

SeetaDet

SeetaDet is a platform implementing popular object detection algorithms.

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

The torch-style codes help us to simplify the hierarchical pipeline of modern detection.

Requirements

seeta-dragon >= 0.3.0.dev20201014

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

Install From Source

Clone this repository to local disk and install:

cd seetadet && python setup.py install

Install From Git

You can also install it from remote repository:

pip install git+https://gitlab.seetatech.com/seetaresearch/seetadet.git@master

Quick Start

Train a detection model

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

We have provided the default YAML examples into configs.

Test a detection model

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

Or

cd tools
python test_all.py --cfg <MODEL_YAML> --exp_dir <EXP_DIR> --last 1

Export a detection model to ONNX

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

Benchmark and Model Zoo

Results and models are available in the Model Zoo.

Supported Backbones

Supported Algorithms

License

BSD 2-Clause license