Skip to content

PadListDataCollate to support inverse MetaTensor #4641

Closed
@wyli

Description

@wyli

PadListDataCollate.inverse should support MetaTensor (as it's used in test time augmentation)

for key in d:
transform_key = InvertibleTransform.trace_key(key)
if transform_key in d:
transform = d[transform_key][-1]
if not isinstance(transform, Dict):

also the test cases are incomplete:

loader = DataLoader(dataset, batch_size=10, collate_fn=collate_method)
# check collation in forward direction
for data in loader:
if t_type == dict:
decollated_data = decollate_batch(data)
for d in decollated_data:
PadListDataCollate.inverse(d)
if __name__ == "__main__":

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions