我在 Python 中使用 Keras 制作了一个图像分类模型,格式为“.h5”。我正在尝试使用 Deeplearning4j 在我的 Android 应用程序中使用它。
当我尝试通过使用NativeImageLoader构造函数加载Mat图像来进行图像分类时,我遇到了一个问题。代码如下:
NativeImageLoader nativeImageLoader = new NativeImageLoader(60, 60, 3);
INDArray image = nativeImageLoader.asMatrix(testImage); // testImage is of Mat format
// 0-255 to 0-1
DataNormalization scaler = new ImagePreProcessingScaler(0, 1);
scaler.transform(image);
// Pass through to neural Net
INDArray output = model.output(image);
INDArray labels = model.getLabels();
当应用程序构建时,它会在上面代码的第二行给出错误,即INDArray image = nativeImageLoader.asMatrix(testImage);
构建 apk 时出现以下错误:
Error:(1109, 51) error: cannot access BufferedImage
class file for java.awt.image.BufferedImage not found
我试图找到解决方案,但这表明Android 不支持 AWT 包。
请帮助我解决方案或解决方法。谢谢你。