search

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

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

How to fit in my own dataset #11

Open momo1986 opened 5 years ago

momo1986 commented 5 years ago

Hello Cheispolo.

I have modified your script on my own dataset. Generally, the configuration is like your previous Keras file. Thanks for sharing! The keypoint number is 11: However, I met some problems: image The loss of both training dataset and vlidation daset cannot be reduced.

How can I resolve the problem with hyper-parameter adjusting?

My input data is 256*256 pixel.

fastovetsilya commented 3 years ago

Try making changes in compile() function in MaskRCNN class in model.py. You can try using Adam optimizer and see if it works.