search

johnolafenwa / DeepStack

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

What format are custom models expected to be in? #79

Closed jasonmadigan closed 3 years ago

jasonmadigan commented 3 years ago

The documentation wasn't particularly clear, wondered what format are custom models expected to be in?

robmarkcole commented 3 years ago

The models are yolo5 format as described on https://docs.deepstack.cc/custom-models/training/index.html#important-parameters Is this what you want to know?

jasonmadigan commented 3 years ago

It does, thanks @robmarkcole!

I have a use-case that's looking at some smoke detection to one of my and actually came across your other repo, which I found very helpful: https://github.com/robmarkcole/fire-detection-from-images

Have been trying to find similar models (ideally, ones easily compatible with deepstack), but which support smoke moreso than fire.

robmarkcole commented 3 years ago

You can just create a model for smoke in exactly the same way I've created one for fire. Lets take the conversation to https://github.com/robmarkcole/fire-detection-from-images/discussions/21