search

JerryJiaGit / facenet_trt

NVIDIA TensorRT implementation for facenet with pre-trained Inception-ResNet v1 SavedModel/Ckpt and MTCNN Networks
MIT License
104 stars 28 forks source link

Dimension issue #11

Open jhanvi55 opened 3 years ago

jhanvi55 commented 3 years ago

I'm trying to load classifier.py and getting the following errorL

/usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/importer.py", line 497, in _import_graph_def_internal graph._c_graph, serialized, options) # pylint: disable=protected-access tensorflow.python.framework.errors_impl.InvalidArgumentError: Node 'gradients/InceptionResnetV1/Bottleneck/BatchNorm/cond/FusedBatchNorm_1_grad/FusedBatchNormGrad' has an _output_shapes attribute inconsistent with the GraphDef for output #3: Dimension 0 in both shapes must be equal, but are 0 and 512. Shapes are [0] and [512]

ValueError: Node 'gradients/InceptionResnetV1/Bottleneck/BatchNorm/cond/FusedBatchNorm_1_grad/FusedBatchNormGrad' has an _output_shapes attribute inconsistent with the GraphDef for output #3: Dimension 0 in both shapes must be equal, but are 0 and 512. Shapes are [0] and [512].

Please help.

JerryJiaGit commented 3 years ago

Could you have a try with this model? https://1drv.ms/u/s!Avn1tfibQ0-ab8z0TkPsnW0Uneo?e=WRwgyH