Skip to content
Merged
3 changes: 2 additions & 1 deletion monai/data/image_reader.py
Original file line number Diff line number Diff line change
Expand Up @@ -766,7 +766,8 @@ def _get_seg_data(self, img):
all_segs = np.zeros([*spatial_shape, n_classes])

for i, (frames, description) in enumerate(self._get_frame_data(img)):
class_name = description.SegmentDescription
segment_label = getattr(description, "SegmentLabel", f"label_{i}")
class_name = getattr(description, "SegmentDescription", segment_label)
if class_name not in metadata["labels"].keys():
metadata["labels"][class_name] = i
class_num = metadata["labels"][class_name]
Expand Down