mtcnn-java
mtcnn-java copied to clipboard
Problem in try to use GraphRunner for FaceNet
I came to have the same problem maybe you've solved.
I use the pre-trained FaceNet Model. the same as you are:
/===== Input Map =====/ Map<String, INDArray> feedMap = Maps.newHashMap(); feedMap.put("input:0", imgArr); /===== Set Phase Input =====/ INDArray phaseInput = ???; feedMap.put("phase_train:0", phaseInput); /===== Run Graph =====/ Map<String, INDArray> resultMap = this.m_faceNetGraphRunner.run(feedMap); /===== Output Map =====/ INDArray embeddings = resultMap.get("embeddings:0");
the GraphRunner cannot accept Boolean feed.
Have you solved it or found some walkaround ?
My appreciate