Detectron.pytorch
Detectron.pytorch copied to clipboard
A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
@roytseng-tw, hi, thanks for your excellent work, I just curious about the mask loss that in your implementation. In the paper, it says that they use the predicted label by...
### Expected results load checkpoints from Detectron R-50_FPN_2x  ### Actual results ``` --------------------------------------------------------------------------- KeyError Traceback (most recent call last) ~/work/Detectron.pytorch/tools/train_net_step.py in () 457 458 if __name__ == '__main__': -->...
Do you plan to support training Using Precomputed RPN Proposals?
related to [issue161](https://github.com/roytseng-tw/Detectron.pytorch/issues/161)
Hi @roytseng-tw , Thanks for writing this amazing repo. I have added (explicit) support for Pascal VOC 2007 and 2012 data set, along with a custom data set. I have...
Support single GPU and multi GPU testing on RPN proposals Do __not__ support RPN only training. Cases such as 1. training only the RPN branch 2. training on pre-computed proposals...
## PLEASE FOLLOW THESE INSTRUCTIONS BEFORE POSTING 1. **Read the README.md thoroughly ! README.md is not a decoration.** 2. Please search existing *open and closed* issues in case your issue...
have you anyone got such a problem? when run testing. Traceback (most recent call last): File "tools/test_net.py", line 14, in from core.test_engine import run_inference File "/home/wangshuainan/mask-rcnn.pytorch/lib/core/test_engine.py", line 39, in from...
If you try to use `RoICrop` as a `ROI_XFROM_METHOD` it fails on the following line https://github.com/roytseng-tw/Detectron.pytorch/blob/dddeb6f4ff9fe14e122c26ec0ab5c1928027d8dc/lib/modeling/model_builder.py#L283 as `grid_size` is not defined for `Generalized_RCNN`. It appears that original Detectron doesn't support...
Hi, Thanks for the code. It's very helpful! Any advice on how to train RPN_ONLY? I guess the current code doesn't support RPN_ONLY training? Thanks.