constructor({ iou_scores, pred_masks }) {
        super();
        this.iou_scores = iou_scores;
        this.pred_masks = pred_masks;
    }
@param {Object} output The output of the model.
@param {Tensor} output.iou_scores The output logits of the model.
@param {Tensor} output.pred_masks Predicted boxes.