vacancy / PreciseRoIPooling

Precise RoI Pooling with coordinate gradient support, proposed in the paper "Acquisition of Localization Confidence for Accurate Object Detection" (https://arxiv.org/abs/1807.11590).
MIT License
772 stars 152 forks source link

Questions about IOU-Net #11

Closed SingL3 closed 5 years ago

SingL3 commented 5 years ago

Hello! I am very interesting in this work. After reading the paper, I have several questions. Firstly, the IOU head need label to regress, but the IOU and the final cls_score and bbox_reg are forwarded at the same time. So did you use the IOU between the RPN proposal and the groundtruth as the label of the IOU head? Secondly, at the refinement part, do you actually update the weight of the network or just try if the network will be better or tolerant after backward but didn't actually update it?

lycqu commented 5 years ago

I have the same questions with you ?May I comnunicate teh problems with you on wechat.MY wechat ID:18827384395. Three Question:(1)IOU loss (2)IOU label (3)the relationshape between the IOU loss and localization confidience.

SingL3 commented 5 years ago

I reread the training of IOU part. And here is my thinking about the first question I asked. Would you please see that whether I am right? The gt of the IOU branch is a sampling the augmenting of gt boxes. When training the whole network end-to-end, the proposal of RPN is pooled and fed into and then the cls bbox branch but not the IOU branch. And the augmentation of boxes is pooled and fed into both the cls bbox branch and IOU branch. I will appreciate it if you reply me. And I am still confused with the second question.

YuningJiang commented 5 years ago

To the first question, yes. To the second question, the weights of network will NOT be updated during the refinement step. You can put it in this way: IoU = f(bbox, w), where f(.) is the trained network to predict IoU, bbox is the box proposal and w are the weights of network. In the refinement step, we update bbox instead of w to maximize IoU.

SingL3 commented 5 years ago

@YuningJiang Thank you for your answer!

l2009312042 commented 5 years ago

good answer help me a lot