I am trying to extend SSDMetaArch such that postprocess(...) returns detection_features (same as faster_rcnn_*). Is there a way to determine which box originates from which feature map, i.e. to map each box_encodings with feature_maps from raw prediction_dict?