如何在java中使用预训练的cnn？

Question

我正在做一个我需要进行人脸识别的项目。我使用LENET架构创建了一个CNN。现在我想在我的项目中使用预训练的权重。我怎么能在JAVA中做到这一点。 请帮助!!

Answer 1

您可以使用DL4J（Deeplearning4j）。他们支持Convolutional Nets，你可以阅读this link。特别是this code可能接近您的需求。这是一个示例，说明如何使用DL4J配置ConvNet。