def segmentation_collate()

in scripts/dataset/dataset.py [0:0]

• 4 lines of code
• 3 McCabe index (conditional complexity)

def segmentation_collate(items):
    imgs = torch.stack([item["img"] for item in items])
    metas = [item["meta"] for item in items]

    return {"img": imgs, "meta": metas}