Skip to content

Instantly share code, notes, and snippets.

@filipsedivy
Created March 21, 2022 13:42
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save filipsedivy/e8ca4ef2422e4dc73b1c225f9b2a30c4 to your computer and use it in GitHub Desktop.
Save filipsedivy/e8ca4ef2422e4dc73b1c225f9b2a30c4 to your computer and use it in GitHub Desktop.
import ai.djl.modality.cv.transform.Resize;
import ai.djl.modality.cv.transform.ToTensor;
import ai.djl.modality.cv.translator.ImageClassificationTranslator;
ImageClassificationTranslator.Builder myTranslator = ImageClassificationTranslator.builder()
.addTransform(new Resize(224, 224))
.addTransform(new ToTensor());
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment