for key, value in box_predictions.items():
        box_predictions[key] = tf.concat(value, axis=1)