Есть много разных моделей для разных задач. Вы можете попробовать использовать UNet, если вам нужно сегментировать один объект (например, баллон / без балуна).
Если вам нужно обнаружить несколько объектов, вы можете попробовать использовать маску R-CNN.
Существует руководство для UNet
PS для задач обнаружения объектов, которые я рекомендую использовать PyTorch, а не TensorFlow. Это много готового кода PyTorch для задачи сегментации, например. Вот учебник для PyTorch Mask R-CNN