Skip to content

Some Example Neural Models that we've trained along with the training scripts

License

Notifications You must be signed in to change notification settings

jcbrtl/depthai-ml-training

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

58 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Start with the why!

We want you to be able to train your own neural models (object detectors) for DepthAI.

This repository contains some pre-trained models under model-zoo and the Jupyter Notebooks we used to produce these neural models in colab notebooks.

Use the model-zoo models if we happened to already train something you need (and see here for models that Intel already trained that run on DepthAI already) and use the colab notebooks if you'd like to train something new or improve upon the training we've already done.

What is DepthAI?

It's the platform for Embedded Spatial AI and Embedded AI. Four trillion operations on a tiny device which can ouput structured data of what is in the physical world - what it is, and where it is in meters - in real-time.

How do I train models? What is Google Colaboratory?

Google Colaboratory is a great tool to do so! It allows you to train neural models yourself using their fast GPU instances, and in some cases (depending on the dependencies of the training frameworks), even using the Google TPU - all for free!

It is exactly intended for this proof of concept and initial research. And if you hit the limits of the free account, you can upgtrade to a Pro version of Google Colab for only $10/month.

You can browse (and open/use) all the Jupyter Notebooks in this repo by clicking on Open In Colab.

About

Some Example Neural Models that we've trained along with the training scripts

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 100.0%