Подготовка изображения средствами nd4j

Есть код на питоне для подготовки изображения к классификации:

test_image = image.load_img(image_path, target_size=(64, 64))
test_image = image.img_to_array(test_image)
test_image = np.expand_dims(test_image, axis=0)
test_image = test_image.astype('float') / 255.0
classifier = classifier_initialization()
classifier.load_weights('network.h5')
return classifier.predict(test_image)

Нужно переписать его на джаву, самые близкие к оригиналу результаты получаются с :

NativeImageLoader loader = new NativeImageLoader(64, 64);
INDArray image = loader.asMatrix(file);
DataNormalization scalar = new ImagePreProcessingScaler(0,1);
scalar.transform(image);
INDArray output = model.output(image);

Вопрос: что ещё нужно сделать для получения одинаковых результатов?