filter other labels

Signed-off-by: Can-Zhao <[email protected]>
Project-MONAI · Feb 19, 2025 · ce3532d · ce3532d
1 parent f62ddd6
commit ce3532d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/maisi_ct_generative/scripts/sample.py b/models/maisi_ct_generative/scripts/sample.py
@@ -729,7 +729,7 @@ def sample_multiple_images(self, num_img):
                     self.output_dir, "sample_" + output_postfix + "_image" + self.image_output_ext
                 )
                 # filter out the organs that are not in anatomy_list
-                # synthetic_labels = filter_mask_with_organs(synthetic_labels, self.anatomy_list)
+                synthetic_labels = filter_mask_with_organs(synthetic_labels, self.anatomy_list)
                 label_saver = SaveImage(
                     output_dir=self.output_dir,
                     output_postfix=output_postfix + "_label",