search

chrispolo / Keypoints-of-humanpose-with-Mask-R-CNN

Use the Mask RCNN for the human pose estimation
263 stars 90 forks source link

Dimension error when running demo_human_pose.ipynb #14

Closed olivier-gillet closed 5 years ago

olivier-gillet commented 5 years ago

When I run demo_human_pose, I get this error : 

/usr/local/lib/python3.5/site-packages/tensorflow/python/framework/ops.py in _create_c_op(graph, node_def, inputs, control_inputs)
   1588   try:
-> 1589     c_op = c_api.TF_FinishOperation(op_desc)
   1590   except errors.InvalidArgumentError as e:

InvalidArgumentError: Dimension 1 in both shapes must be equal, but are 8 and 324. Shapes are [1024,8] and [1024,324]. for 'Assign_682' (op: 'Assign') with input shapes: [1024,8], [1024,324].

ValueError: Dimension 1 in both shapes must be equal, but are 8 and 324. Shapes are [1024,8] and [1024,324]. for 'Assign_682' (op: 'Assign') with input shapes: [1024,8], [1024,324].

Can you tell me what version of TF/Keras to use ? I use Tensorflow 1.9.0 + Keras 2.1.6

olivier-gillet commented 5 years ago

I solved this issue. It is because the model downloaded from this script is not the one trained with keypoints. To download the good model, refer to the baidu link in the README, or download it here : https://github.com/Superlee506/Mask_RCNN_Humanpose/releases/download/v0.9-alpha/mask_rcnn_coco_humanpose.h5