| | |
| | from mmengine.structures import BaseDataElement, PixelData |
| |
|
| |
|
| | class SegDataSample(BaseDataElement): |
| | """A data structure interface of MMSegmentation. They are used as |
| | interfaces between different components. |
| | |
| | The attributes in ``SegDataSample`` are divided into several parts: |
| | |
| | - ``gt_sem_seg``(PixelData): Ground truth of semantic segmentation. |
| | - ``pred_sem_seg``(PixelData): Prediction of semantic segmentation. |
| | - ``seg_logits``(PixelData): Predicted logits of semantic segmentation. |
| | |
| | Examples: |
| | >>> import torch |
| | >>> import numpy as np |
| | >>> from mmengine.structures import PixelData |
| | >>> from mmseg.structures import SegDataSample |
| | |
| | >>> data_sample = SegDataSample() |
| | >>> img_meta = dict(img_shape=(4, 4, 3), |
| | ... pad_shape=(4, 4, 3)) |
| | >>> gt_segmentations = PixelData(metainfo=img_meta) |
| | >>> gt_segmentations.data = torch.randint(0, 2, (1, 4, 4)) |
| | >>> data_sample.gt_sem_seg = gt_segmentations |
| | >>> assert 'img_shape' in data_sample.gt_sem_seg.metainfo_keys() |
| | >>> data_sample.gt_sem_seg.shape |
| | (4, 4) |
| | >>> print(data_sample) |
| | <SegDataSample( |
| | |
| | META INFORMATION |
| | |
| | DATA FIELDS |
| | gt_sem_seg: <PixelData( |
| | |
| | META INFORMATION |
| | img_shape: (4, 4, 3) |
| | pad_shape: (4, 4, 3) |
| | |
| | DATA FIELDS |
| | data: tensor([[[1, 1, 1, 0], |
| | [1, 0, 1, 1], |
| | [1, 1, 1, 1], |
| | [0, 1, 0, 1]]]) |
| | ) at 0x1c2b4156460> |
| | ) at 0x1c2aae44d60> |
| | |
| | >>> data_sample = SegDataSample() |
| | >>> gt_sem_seg_data = dict(sem_seg=torch.rand(1, 4, 4)) |
| | >>> gt_sem_seg = PixelData(**gt_sem_seg_data) |
| | >>> data_sample.gt_sem_seg = gt_sem_seg |
| | >>> assert 'gt_sem_seg' in data_sample |
| | >>> assert 'sem_seg' in data_sample.gt_sem_seg |
| | """ |
| |
|
| | @property |
| | def gt_sem_seg(self) -> PixelData: |
| | return self._gt_sem_seg |
| |
|
| | @gt_sem_seg.setter |
| | def gt_sem_seg(self, value: PixelData) -> None: |
| | self.set_field(value, '_gt_sem_seg', dtype=PixelData) |
| |
|
| | @gt_sem_seg.deleter |
| | def gt_sem_seg(self) -> None: |
| | del self._gt_sem_seg |
| |
|
| | @property |
| | def pred_sem_seg(self) -> PixelData: |
| | return self._pred_sem_seg |
| |
|
| | @pred_sem_seg.setter |
| | def pred_sem_seg(self, value: PixelData) -> None: |
| | self.set_field(value, '_pred_sem_seg', dtype=PixelData) |
| |
|
| | @pred_sem_seg.deleter |
| | def pred_sem_seg(self) -> None: |
| | del self._pred_sem_seg |
| |
|
| | @property |
| | def seg_logits(self) -> PixelData: |
| | return self._seg_logits |
| |
|
| | @seg_logits.setter |
| | def seg_logits(self, value: PixelData) -> None: |
| | self.set_field(value, '_seg_logits', dtype=PixelData) |
| |
|
| | @seg_logits.deleter |
| | def seg_logits(self) -> None: |
| | del self._seg_logits |
| |
|
