search

eto-ai / rikai

Parquet-based ML data format optimized for working with unstructured data
https://rikai.readthedocs.io/en/latest/
Apache License 2.0
137 stars 19 forks source link

TFHub Registry #538

Closed da-tubi closed 2 years ago

da-tubi commented 2 years ago

close #381

Demo sql using Tensorflow Hub

CREATE MODEL tfssd
PREPROCESSOR 'rikai.contrib.tensorflow.models.ssd.pre_processing'
POSTPROCESSOR 'rikai.contrib.tensorflow.models.ssd.post_processing'
OPTIONS (device="cpu", batch_size=32)
RETURNS {OUTPUT_SCHEMA}
USING "tfhub:///tensorflow/ssd_mobilenet_v2/2";

select ML_PREDICT(tfssd, to_image('{image_path}')) as preds