Refer to [urgent] How to save the bbox confidence probability (i.e. prediction score) during faster-rcnn inference? - #5 by Morganh