search

johnolafenwa / DeepStack

The World's Leading Cross Platform AI Engine for Edge Devices
Apache License 2.0
678 stars 107 forks source link

Docs and tooling for custom models #13

Closed robmarkcole closed 3 years ago

robmarkcole commented 4 years ago

I am currently researching the landscape of tools for creating datasets for custom object detection models. Is there any docs on this for generating models compatible with deepstack? If not can you provide some guidance or recommendation on how to get started, and I will do so :)

johnolafenwa commented 4 years ago

Hey @robmarkcole , Going forward, deepstack custom models will be based on, https://github.com/facebookresearch/detr.

With dataset annotated with https://github.com/tzutalin/labelImg

Based on the DETR above, we shall be creating a library for training object detection models to be used with deepstack. We shall work on this together. The library can be in this repo as well. I will setup the directory this weekend.

Majorly, it will involve modifying the data loader to read csv rather than coco.

robmarkcole commented 4 years ago

Thanks for ths info. Btw I found this (incomplete) reference https://github.com/lessw2020/training-detr I like the idea of leveraging google colab for free compute

robmarkcole commented 3 years ago

done in docs repo