easy-faster-rcnn.pytorch
easy-faster-rcnn.pytorch copied to clipboard
About the NMS
Thanks for your clean and useful code implementation.I would like to consult you why 'class_bboxes' on the NMS are not arranged in descending order of 'class_probs' in 'model.py' when validating or testing. The code is as follows:
kept_indices = nms(class_bboxes, class_probs, threshold)
- First of all, the function
nmshas handled boxes sorting - So that in
generate_detectionsboxes doesn't require to sort beforenmsis called - You might be confused by similar procedure in
generate_proposalswhich has done sorting beforenms, it's because that we want to take top N boxes and then pass tonmsfor the efficiency (note that #boxes betweengenerate_detectionsandgenerate_proposalswas different, typically <1k and >10k)
Thank you for your reply. It is useful to me
I have another question: why is the threshold of detection_probs for the verification phase very low, so that there are enough boxes involved in the calculation of map? As in the code in evaluator.py:
kept_indices = (detection_probs > 0.05).nonzero().view(-1)
Right, and this threshold is determined by reference to other popular repos, such as maskrcnn-benchmark
Well, thanks a lot for your help.