mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

Merge branch 'limengzhang/refactor_fix_to_tensor' into 'refactor_dev'

Browse Source 
[Fix] Fix omitting to_tensor in PackSegInputs

See merge request openmmlab-enterprise/openmmlab-ce/mmsegmentation!34
This commit is contained in:
zhengmiao 2022-06-09 03:19:56 +00:00
commit 5e7d7626a8
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmseg/datasets/pipelines/formatting.py
View File

@ -69,7 +69,8 @@ class PackSegInputs(BaseTransform):
data_sample = SegDataSample()
if 'gt_seg_map' in results:
gt_sem_seg_data = dict(
data=results['gt_seg_map'][None, ...].astype(np.int64))
data=to_tensor(results['gt_seg_map'][None,
...].astype(np.int64)))
data_sample.gt_sem_seg = PixelData(**gt_sem_seg_data)
img_meta = {}