keypoints detection on my own dataset

matterport / Mask_RCNN

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow

Other

24.53k stars 11.68k forks source link

keypoints detection on my own dataset #1076

Open ghost opened 5 years ago

ghost commented 5 years ago

Hey all, i am aiming to do keypoints detection on my own dataset which include only two keypoints and not 17 like human pose estimation.

I tried to use the pre trained weights of COCO for human pose estimation, but of course I get the following error "ValueError: Dimension 2 in both shapes must be equal, but are 2 and 17. Shapes are [2,2,2,512] and [2,2,17,512]. for 'Assign_724' (op: 'Assign') with input shapes: [2,2,2,512], [2,2,17,512]."

How can i solve this problem?

Thanks

ShibaPrasad commented 5 years ago

I am getting below error

ValueError Traceback (most recent call last) /miniconda/lib/python3.6/site-packages/tensorflow/python/framework/op_def_library.py in _apply_op_helper(self, op_type_name, name, keywords) 509 as_ref=input_arg.is_ref, --> 510 preferred_dtype=default_dtype) 511 except TypeError as err: ** would you please help me !!!!!