def METHOD_NAME(self, batch):
    tensor_annotations = batch[1]
    label = []
    for annotation in tensor_annotations:
        if len(annotation["boxes"]):
            bbox = torch.stack(annotation["boxes"])
            bbox[:, 2:] = bbox[:, 2:] - bbox[:, :2]
            label.append(
                torch.concat([torch.stack(annotation["labels"]).reshape((-1, 1)), bbox], dim=1)
            )
        else:
            label.append(torch.tensor([]))
    return label