Allow ReadInput to return either a Tensor or a raw byte array

hotg-ai / rune

Rune provides containers to encapsulate and deploy edgeML pipelines and applications

Apache License 2.0

134 stars 15 forks source link

Allow ReadInput to return either a Tensor or a raw byte array #382

Closed Michael-F-Bryan closed 2 years ago

Michael-F-Bryan commented 2 years ago

After reading https://github.com/tensorflow/tfjs/issues/206, it doesn't seem like tf.js supports uint8 tensors. As a workaround, let's let people return either a tensor or a raw byte buffer that will be blindly copied across.