def get_data_shapes(masks):
    """Generating data shapes for tensorflow datasets depending on if we need to load the masks or not
    outputs:
        data shapes with masks = output data shapes for (images, ground truth boxes, ground truth labels, masks, mask id labels)
        data shapes without masks = output data shapes for (images, ground truth boxes, ground truth labels)
    """
    if masks:
        return ([None, None, None], [None, ], [None, None], [None, ], [None, None, None], [None, ])
    else:
        return ([None, None, None], [None, ], [None, None], [None, ])