Skip to content

megvii-research/basedet

Repository files navigation

BaseDet

English | 中文

BaseDet is a fast and easy-to-use detection toolbox. Learn more at our documentation.

Features

  • This repo is powered by the MegEngine deep learning framework.
  • It provides serveral classic SOTA models and related components, which could be used as basic libraray.

Installation

See INSTALL.md.

Getting Started

  1. Make sure that BaseDet is installed in a right way.

    You could run the following command to check.

    python3 -c "import basedet; print(basedet.__version__)"
  2. Prepare dataset

  3. Train model

    BaseDet provides a simple command basedet_train to train model. Just simply run the follwing command to train network (config is used to describe the training process).

    basedet_train -f config.py
  4. Test model

    Like training, simply use the fellowing command.

    basedet_test -f config.py

Model Zoo and Baselines

We provide a large set of baseline results and trained models available for download in the Model Zoo.

License

BaseDet is released under the Apache 2.0 license.

About

An object detection codebase based on MegEngine.

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Contributors 2

  •  
  •  

Languages