-
Notifications
You must be signed in to change notification settings - Fork 164
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Is there a plan to open source the trained weights of MaskRCNN? #8
Comments
Hi @hpanwar08 Thanks for your interests. Yes we do. We are going to migrate the data to another platform, which also supports sharing models. We are under internal legal assess now. Hopefully we can get approved soon. Please stay tuned :) |
Looking forward to meeting with faster data-getting method and shared-model :) |
@zhxgj HI,is the model ready to publish? How much time did you spend on training this model (what's the setting)? Is there any accelerating skills? Now I try to train the model, but find it may take me many days with only 4GPUs. Thanks. |
I am training on half of the training data on matterport maskrcnn on 1 GPU, it has been running for days. 1 epoch on half of the training data takes 42 hrs. |
It took me about a week to train the models on 8 K80 GPUs. The models are still under legal review and unfortunately I do not have timeline right now. But I think it should be soon. I will follow up with our legal team to accelerate things. |
Waiting for the pretrained model weights . Hope we will get it soon :) |
Thanks @monuminu I will keep following up with our legal team |
I have trained Mask RCNN using detectron2 on half of the training data (143k images) on a single GPU with resnet101 backbone initialized with coco weights. My results are close to the ones mentioned in the paper. |
Hi @hpanwar08, I got similar results on small objects. Below are my more detailed validation results.
|
Pre-trained Faster-RCNN model and Mask-RCNN model are released. |
Thanks @zhxgj |
Any plan to release the weights for detectron2? Thank you |
We plan to re-train the models on PubLayNet using Detectron2. The NVIDIA driver on our GPU cluster is out-dated and does not support Detectron2. We do not have the permission to upgrade the driver, but we are actively seeking a solution at the moment. To use Detectron model in Detectron2, I saw someone posted a solution here. I have not tested it, but you can have a try. |
Thank you this large dataset. Is there a plan to open source the trained weights and code in keras or pytorch?
The text was updated successfully, but these errors were encountered: