luxonis/depthai-model-zoo

DepthAI Model Zoo is a collection of open-source neural network models and datasets created and maintained by DepthAI developers and community

DepthAI - Model ZOO

DepthAI Model Zoo is a collection of open-source neural network models and datasets created and maintained by DepthAI developers and the community.

We try to provide already converted latest state-of-the-art models ready for use with DepthAI and our OAK cameras. Most models are accompanied by additional metadata (FPS, accuracy, number of parameters, and FLOPs) and links to the experiments that show the example usage of each model.

Models in DepthAI Model ZOO should be in BGR color order and CHW (planar) channel layout. You can change these values with Model Optimizer's --layout and reverse_input_channels arguments.

Usage

To use the models on camera, use blobconverter API:

1. install blobconverter using pip:

   pip install blobconverter
   

2. use the following snippet to download the YOLOP model and get its blob path:

   model_path = blobconverter.from_zoo(name="yolop_320x320",
                                       zoo_type="depthai",
                                       shaves=6)
   

In the example above we use the YOLOP model from the DepthAI Model ZOO. In case you wish to download the model from the Open Model ZOO, you can omit the zoo_type flag.

Additional information

For more information please visit:

• blobconverter repository for more information on how to use the API,
• blobconverter web interface for easy conversion of the models that are not in our model zoo,
• our documentation for more information on how to convert and deploy the models.

Contribute

We invite the community to help out by uploading their models via Pull Requests. Help with performance estimation of the models on our devices is also appreciated.

Thanks to everyone who has already contributed!