Skip to content
Discussion options

You must be logged in to vote

@kliebhart

Here is Image classification example for TFLite: https://github.com/deepjavalibrary/djl/blob/master/engines/tflite/tflite-engine/src/test/java/ai/djl/tflite/engine/TfLiteTest.java#L41-L52

You can see it's almost exactly the same as using other engine.

In your case, you need create your own Translator<byte[], Classifications> class and hand pre/post processing to match your model.

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by MisterK91
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants