我正在尝试提取图层激活以将其作为要素保存在本地。我还是CNN的新手,所以我想展示我的所作所为,我想知道我所做的事情是否正确:
公共静态……
对于提取的功能,Nd4j有一个可以使用的PCA: https://github.com/deeplearning4j/nd4j/blob/master/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/dimensionalityreduction/PCA.java
但这有点多余。我会考虑直接使用转移学习。您不必手动执行任何操作。 有关更多示例,请参阅此处的文档: https://github.com/deeplearning4j/dl4j-examples/blob/master/dl4j-spark-examples/dl4j-spark/src/main/java/org/deeplearning4j/transferlearning/vgg16/TransferLearning.md
转移学习api将为您提供使用预训练模型所需的内容,同时仅修改它们以使用不同的输出层。我们甚至在上面的例子中用VGG16覆盖了它。