seasonSH/DocFace

java error

zfs1993 opened this issue · 0 comments

i transfer it to pb,when i run it in java it encouter problems. i try the model facenet with pb file and success.so i am confused that weather docface model can run in java?
Exception in thread "main" java.lang.RuntimeException: Reshape cannot infer the missing input size for an empty tensor unless all specified input sizes are non-zero
[[{{node None/FaceResNet/NetA_1/conv2/SE/Flatten/flatten/Reshape}} = Reshape[T=DT_FLOAT, Tshape=DT_INT32, _device="/job:localhost/replica:0/task:0/device:CPU:0"](None/FaceResNet/NetA_1/conv2/SE/AvgPool2D/AvgPool, None/FaceResNet/NetA_1/conv2/SE/Flatten/flatten/Reshape/shape)]]
at com.umetrip.cv.service.TFTest.performInference(TFTest.java:64)
at com.umetrip.cv.service.TFTest.main(TFTest.java:26)
Caused by: java.lang.IllegalArgumentException: Reshape cannot infer the missing input size for an empty tensor unless all specified input sizes are non-zero
[[{{node None/FaceResNet/NetA_1/conv2/SE/Flatten/flatten/Reshape}} = Reshape[T=DT_FLOAT, Tshape=DT_INT32, _device="/job:localhost/replica:0/task:0/device:CPU:0"](None/FaceResNet/NetA_1/conv2/SE/AvgPool2D/AvgPool, None/FaceResNet/NetA_1/conv2/SE/Flatten/flatten/Reshape/shape)]]
at org.tensorflow.Session.run(Native Method)
at org.tensorflow.Session.access$100(Session.java:48)
at org.tensorflow.Session$Runner.runHelper(Session.java:314)
at org.tensorflow.Session$Runner.run(Session.java:264)
at com.umetrip.cv.service.TFTest.performInference(TFTest.java:58)
... 1 more