Skip to content

Commit a34a656

Browse files
slicepasteeinsyang723IamTingTing
committed
fix constraint for pixdim in track_transform_meta() function
Signed-off-by: Wei_Chuan, Chiang <[email protected]> Co-authored-by: einsyang723 <[email protected]> Co-authored-by: IamTingTing <[email protected]>
1 parent b6dd460 commit a34a656

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

monai/transforms/inverse.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -197,8 +197,9 @@ def track_transform_meta(
197197
else:
198198
raise
199199
out_obj.meta[MetaKeys.AFFINE] = convert_to_tensor(affine, device=torch.device("cpu"), dtype=torch.float64)
200-
spacing = affine_to_spacing(out_obj.meta[MetaKeys.AFFINE])
201-
out_obj.meta[MetaKeys.PIXDIM][1 : 1 + len(spacing)] = spacing
200+
if MetaKeys.PIXDIM in out_obj.meta:
201+
spacing = affine_to_spacing(out_obj.meta[MetaKeys.AFFINE])
202+
out_obj.meta[MetaKeys.PIXDIM][1 : 1 + len(spacing)] = spacing
202203

203204
if not (get_track_meta() and transform_info and transform_info.get(TraceKeys.TRACING)):
204205
if isinstance(data, Mapping):

0 commit comments

Comments
 (0)