mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge branch 'limengzhang/fix_SegDataSample' into 'refactor_dev' 

[Fix] Fix wrong example in SegDataSample

See merge request openmmlab-enterprise/openmmlab-ce/mmsegmentation!28
pull/1801/head
zhengmiao 2022-06-07 13:39:02 +00:00
parent 4ce9c01e33 dbb1316c47
commit 49ada036d8
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mmseg/core/data_structures/seg_data_sample.py
View File

@ -22,10 +22,10 @@ class SegDataSample(BaseDataElement):
>>> img_meta = dict(img_shape=(4, 4, 3), >>> img_meta = dict(img_shape=(4, 4, 3),
... pad_shape=(4, 4, 3)) ... pad_shape=(4, 4, 3))
>>> gt_segmentations = PixelData(metainfo=img_meta) >>> gt_segmentations = PixelData(metainfo=img_meta)
>>> gt_segmentations.gt_sem_seg = torch.randint(0, 2, (1, 4, 4)) >>> gt_segmentations.data = torch.randint(0, 2, (1, 4, 4))
>>> data_sample.gt_segmentations = gt_segmentations >>> data_sample.gt_sem_seg = gt_segmentations
>>> assert 'img_shape' in data_sample.gt_segmentations.metainfo_keys() >>> assert 'img_shape' in data_sample.gt_sem_seg.metainfo_keys()
>>> data_sample.gt_segmentations >>> data_sample.gt_sem_seg.shape
(4, 4) (4, 4)
>>> print(data_sample) >>> print(data_sample)
<SegDataSample( <SegDataSample(
@ -33,14 +33,14 @@ class SegDataSample(BaseDataElement):
META INFORMATION META INFORMATION
DATA FIELDS DATA FIELDS
gt_segmentations: <PixelData( gt_sem_seg: <PixelData(
META INFORMATION META INFORMATION
img_shape: (4, 4, 3) img_shape: (4, 4, 3)
pad_shape: (4, 4, 3) pad_shape: (4, 4, 3)
DATA FIELDS DATA FIELDS
gt_sem_seg: tensor([[[1, 1, 1, 0], data: tensor([[[1, 1, 1, 0],
[1, 0, 1, 1], [1, 0, 1, 1],
[1, 1, 1, 1], [1, 1, 1, 1],
[0, 1, 0, 1]]]) [0, 1, 0, 1]]])