jhultman / vision3d

Research platform for 3D object detection in PyTorch.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Recall of RPN stage

cqsnove opened this issue · comments

Hi,I wrote the eval_rpn.py based on your code, and the results of operation are as follows:
total bbox recall(thresh=0.100): 12516 / 19060 = 0.656663
total bbox recall(thresh=0.300): 12242 / 19060 = 0.642288
total bbox recall(thresh=0.500): 11379 / 19060 = 0.597009
total bbox recall(thresh=0.700): 8201 / 19060 = 0.430273
total bbox recall(thresh=0.900): 653 / 19060 = 0.034260

The recall with a threshold of 0.7 is quite different from that in the paper. Is there something wrong with the code or some evaluation details I didn't pay attention to?

eval_rpn.zip

Hi, this library is still a work in progress. I have made several updates in a private branch which may bring results closer to those claimed in paper. (Among others, the smooth L1 loss is missing a scaling factor, the focal loss needs to be averaged by batch size, etc.) I will push those changes here soon in the next major release.