Import pretrained Keras model in Julia for inference

Specific Domains Machine Learning
6

To conclude : after trying ONNX and PyCall I am quite happy with PyCall. Here is an example:

using PyCall: @pyimport
import Images

@pyimport tensorflow as tf
infer(model, x) = model(tf.constant(x)).numpy()

img = Images.imresize(Images.load("data/elephant.jpg"), (224, 224))
rgb = [Float32(getproperty(color, channel)) for z = 1:50, color in img, channel in (:r, :g, :b)]
ResNet50 = tf.keras.applications.resnet50.ResNet50
model = ResNet50(weights = "imagenet")
@time scores = infer(model, rgb);

Note that ‘data/elephant.jpg’ is some downloaded picture file.

2 Likes