ZFTurbo / volumentations

Library for 3D augmentations

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How can I do if I want to use two kinds of mask at the same time?

yjsb opened this issue · comments

commented

I have two kinds of mask, the tumor and the boundary of tumor, and they are two different array.
If I want to argument the image, I have to do same operation in this two masks.
I saw the operation is only for dictionary:

    targets=[['image'],['mask']]

how can I do? I try

    data = {'image': ct_array, 'mask1': seg_array, 'mask':boundary_array}
    aug_data = self.get_augmentation(**data)
    ct_array, seg_array, boundary_array = aug_data['image'], aug_data['mask1'], aug_data['mask']

but it does not work.