def METHOD_NAME(
    batch: List[Any],
) -> Tuple[Tensor, Tuple[Tensor, Tensor, List[Tuple[Tensor, Tensor]]]]:
    """
    Collate function to be used for creating a DataLoader with values for Yolo model
    input.
    :param batch: a batch of data points and annotations transformed by
        bounding_box_and_labels_to_yolo_fmt
    :return: the batch stacked as tensors for all values except for the
        original annotations
    """
    images = []
    targets = []
    annotations = []
    for idx, (image, (target, annotation)) in enumerate(batch):
        images.append(image.unsqueeze(0))
        img_label = torch.ones(target.size(0), 1) * idx
        targets.append(torch.cat((img_label, target), 1))
        annotations.append(annotation)
    images = torch.cat(images, 0)
    targets = torch.cat(targets, 0)
    return images, (targets, annotations)