search

ge-xing / SegMamba

SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentation
https://arxiv.org/abs/2401.13560
366 stars 20 forks source link

KeyError: 'seg' #39

Open AndrewForresterGit opened 4 months ago

AndrewForresterGit commented 4 months ago
Screenshot 2024-06-18 at 12 15 11

While trying to predict, I the code doesn't seem to find the segmentations. What could be the issue?

ge-xing commented 4 months ago

You can delete this line to avoid this error. The "label" variable is not necessary during inference.