Skip to content

Instantly share code, notes, and snippets.

@frankfliu
Created December 27, 2021 17:37
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 frankfliu/550f284c0e990e123ba95915219a56ac to your computer and use it in GitHub Desktop.
Save frankfliu/550f284c0e990e123ba95915219a56ac to your computer and use it in GitHub Desktop.
DJL blockrunner example
package ai.djl.examples;
import ai.djl.inference.Predictor;
import ai.djl.modality.Classifications;
import ai.djl.modality.cv.*;
import ai.djl.modality.cv.transform.*;
import ai.djl.modality.cv.translator.*;
import ai.djl.repository.zoo.*;
public class Example {
public static void main(String[] args) throws Exception {
String modelUrl = "djl://ai.djl.mxnet/resnet/0.0.1/resnet18_v1";
// String modelUrl = "djl://ai.djl.pytorch/resnet/0.0.1/traced_resnet18";
// String modelUrl = "djl://ai.djl.tensorflow/resnet/0.0.1/resnet50";
Criteria<Image, Classifications> criteria = Criteria.builder()
.setTypes(Image.class, Classifications.class)
.optModelUrls(modelUrl)
.optTranslator(ImageClassificationTranslator.builder()
.addTransform(new Resize(224, 224))
.addTransform(new ToTensor())
.optApplySoftmax(true).build())
.build();
ZooModel<Image, Classifications> model = criteria.loadModel();
Predictor<Image, Classifications> predictor = model.newPredictor();
String imageURL =
"https://raw.githubusercontent.com/awslabs/djl/master/examples/src/test/resources/kitten.jpg";
Image image = ImageFactory.getInstance().fromUrl(imageURL);
Classifications output = predictor.predict(image);
System.out.println(output);
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment