如何将Mat（OpenCV）转换为INDArray（DL4J）？

Question

我有一个采用INDArray的keras网络模型，但我不知道如何将Mat转换为INDArray。我知道这是与How to convert Mat (opencv) to INDArray (DL4J)?相同的问题，但并没有帮助我。是否有执行此任务的API？谢谢

Answer 1

您可以使用NativeImageLoader进行转换。

import org.datavec.image.loader.NativeImageLoader;



(...)
        Mat cvImage();

        // Fill in your Mat with something

        NativeImageLoader nil = new NativeImageLoader();
        INDArray image = nil.asMatrix(cvImage).div;

请确保pom.xml中具有对datavec的依赖性。您遇到什么错误？