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

iou(pred_box, truth_box) or iou(prior_box, truth_box) should be calculated here? #306

Open
WormCoder opened this issue Sep 11, 2018 · 1 comment
Labels
question Further information is requested

Comments

@WormCoder
Copy link

batch_ious = self._batch_iou(box_preds, gt_boxes) # (B, N, M)

In the yolov3's paper, ignore_threshold if for the iou between prior(or anchor) box and truth box, i.e. the iou(prior_box, truth_box), so why do you use iou(pred_box, truth_box) here?

@WormCoder WormCoder changed the title iou(pred_box, truth_box) of iou(prior_box, truth_box) should be calculated here? iou(pred_box, truth_box) or iou(prior_box, truth_box) should be calculated here? Sep 11, 2018
@zhreshold
Copy link
Member

The order is swapped by transpose. The reason to use it in reverse order is because anchors are of batch size 1, and by doing this we can save batch_size times the computation.

@zhreshold zhreshold added the question Further information is requested label Sep 14, 2018
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