How can we use mask rcnn weights in keras?

ibm-aur-nlp / PubLayNet

Other

900 stars 165 forks source link

How can we use mask rcnn weights in keras? #14

Closed hrithikpuri5 closed 4 years ago

hrithikpuri5 commented 4 years ago

Thank you for sharing the dataset! I want to use the maskrcnn with keras. How can we use the config and weights for the same?

zhxgj commented 4 years ago

@hrithikpuri5 I do not think you can, since the pre-trained model uses Detectron and Caffe2 (pytorch) framework. If you want to use keras with tensorflow backend, you will need to train the model on PubLayNet.