rikai.contrib.tfhub.tensorflow package¶

Submodules¶

SSD: Single Shot MultiBox Detector https://arxiv.org/abs/1512.02325

class rikai.contrib.tfhub.tensorflow.ssd.SSDModelType¶

predict(images, *args, **kwargs) → Any¶: Run model inference and convert return types into Rikai-compatible types.

schema() → str¶

Return the string value of model schema.

Examples

>>> model_type.schema()
... "array<struct<box:box2d, score:float, label_id:int>>"

transform() → Callable¶

A callable to pre-process the data before calling inference.

It will be feed into torch.data.DataLoader or tensorflow.data.Dataset.map().