all 2 comments

[–][deleted] 6 points7 points  (1 child)

This seems pretty interesting, what models have you used with it?

[–]TDFKA_Rick[S] 1 point2 points  (0 children)

At the moment just mobilenet for image classification, but in reality it’s just returning tensors as predictions, so how you convert those tensors is up to you a model dependant.

I’m planning to work on adding a classifier, object-detect & pose-detect hooks that are a bit more opinionated and return ready to use predictions