box_prediction_head = keras_box_head.MaskRCNNBoxHead(