Conditions | 2 |
Total Lines | 10 |
Code Lines | 10 |
Lines | 0 |
Ratio | 0 % |
Changes | 0 |
1 | import SimpleITK as sitk |
||
23 | def apply_transform(self, subject): |
||
24 | for image in self.get_images(subject): |
||
25 | assert image.data.ndim == 4 and image.data.shape[0] == 1 |
||
26 | sitk_image = image.as_sitk() |
||
27 | connected_components = sitk.ConnectedComponent(sitk_image) |
||
28 | labeled_cc = sitk.RelabelComponent(connected_components) |
||
29 | largest_cc = labeled_cc == 1 |
||
30 | tensor, _ = self.sitk_to_nib(largest_cc) |
||
31 | image.set_data(tensor) |
||
32 | return subject |
||
33 |