How can I do if I want to use two kinds of mask at the same time?
yjsb opened this issue · comments
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.