Why do you have output_category_mask=False
and are expecting 2 outputs ? You are specifically asking the model to only return 1 output.
Please check the documentation and source code.
output_confidence_masks:
Whether to output confidence.
output_category_mask:
Whether to output category mask.