Open PkuRainBow opened 5 years ago
I am trying the solution by Ross. Solution
I will report the results latter.
@PkuRainBow hello, have you fixed the problem? btw, which context model are you using? Thank you.
@Redaimao I have solved this problem. I use the Res50-FPN as the backbone.
@PkuRainBow Would you mind give details on how to implement it? and also please share some details on context module. Thanks a lot!
@PkuRainBow hello, do you have implemented the performance of maskrnn with non local. (increase about 1 point )
I add a non-local block to the backbone and train the network following "e2e_mask_rcnn_R-50-RPN_1x.yaml".
However, I am not sure how to set the learning rate of the extra introduced non-local block. I just do not change the weight initializer and the learning rate setting related code.
Here I provide the change that I make,
First, I insert a non_local module within the function add_stage_w_non_local,
Then, I modify the backbone definition like below by replacing the add_stage with add_stage_w_non_local when we compute the self.res4.
Expected results
System information