Superlee506 / Mask_RCNN_Humanpose

Mask R-CNN for Human Pose Estimation on Keras and TensorFlow.
Other
189 stars 36 forks source link

How to train a new model without the pre-trained COCOweight? #15

Closed chrispolo closed 6 years ago

chrispolo commented 6 years ago

Hello @Superlee506 , i came back again, Your train_human_pose.ipynb shows how to train a model about the humanpose,but i want to train a new model about humanpose without the pre_trained"mask_rcnn_coco.h5", How should i change? and if i trained with pretrained weight,the model i got will be based on the pre-trained weight? Thx a lot,hope you give me a answer in your convenience.

Superlee506 commented 6 years ago

@chrispolo If you don't want to initial the model with pre-trained model, you just need to uncomment the "model.load_weights". If you trained with pretrained weight, the model just initial the weight using this pretrained model.