Skip to content
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

About the implement of mist layer #41

Open
anlimo1510 opened this issue Mar 1, 2021 · 0 comments
Open

About the implement of mist layer #41

anlimo1510 opened this issue Mar 1, 2021 · 0 comments
Labels
question Further information is requested

Comments

@anlimo1510
Copy link

anlimo1510 commented Mar 1, 2021

Thanks for your great work.

  1. According to your paper, is the process of selecting pseudo boxes in MIST the same as applying NMS with the threshold 0.2 to the top cfg.ROI_BOX_HEAD.OICR_P proposals?

  2. In your implement,

    _boxes = proposals.bbox[max_inds.t().contiguous().view(-1)].view(num_gt_cls.int(), -1, 4)
    _boxes = BatchBoxList(_boxes, proposals.size, mode=proposals.mode)
    ious = batch_boxlist_iou(_boxes, _boxes)
    k_ind = torch.zeros(num_gt_cls.int(), k, dtype=torch.bool, device=device)
    k_ind[:, 0] = 1 # always take the one with max score
    for ii in range(1, k):
    max_iou, _ = torch.max(ious[:,ii:ii+1, :ii], dim=2)
    k_ind[:, ii] = (max_iou < self.iou_th).byte().squeeze(-1)

    max_iou seems return the max iou among the ii proposal and all the proposals with higher score, but not the proposals in the pseudo boxes list. Is it the case?
    And then max_iou will determine whether the ii proposal be appended to the pseudo boxes list. If the max_iou is from the ii proposal and a proposal which is not selected (due to large overlap with existing proposals in pseudo boxes list), then ii proposal will not be appended to the pseudo boxes list. If so, is it slightly different from Algorithm 1 in your paper?

Thank you.

@jason718 jason718 added the question Further information is requested label Mar 21, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
question Further information is requested
Projects
None yet
Development

No branches or pull requests

2 participants