Change main() to test(), and Path to str

Author: MMelQin

monai/deploy/operators/dicom_data_loader_operator.py

@@ -244,7 +244,7 @@ def populate_series_attributes(self, series, sop_instance):
             pass
 
 
-def main():
+def test():
     data_path = "../../../examples/ai_spleen_seg_data/dcm"
 
     loader = DICOMDataLoaderOperator()
@@ -258,4 +258,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    test()

monai/deploy/operators/dicom_seg_writer_operator.py

@@ -121,7 +121,7 @@ def compute(self, op_input: InputContext, op_output: OutputContext, context: Exe
         output_dir = op_output.get().path
         output_dir.mkdir(parents=True, exist_ok=True)
 
-        self.process_images(seg_image_numpy, dicom_series, output_path)
+        self.process_images(seg_image, study_selected_series_list, output_dir)
 
     def process_images(
         self, image: Union[Image, Path], study_selected_series_list: StudySelectedSeries, output_dir: Path
@@ -146,7 +146,6 @@ def process_images(
             DICOMSegmentationWriterOperator.DCM_EXTENSION,
         )
         output_path = output_dir / output_filename
-
         # Pick DICOM Series that was used as input for getting the seg image.
         # For now, first one in the list.
         for study_selected_series in study_selected_series_list:
@@ -164,16 +163,16 @@ def create_dicom_seg(self, image: Image, dicom_series: DICOMSeries, file_path: P
         # DICOM Seg creation
         self._seg_writer = DICOMSegWriter()
         try:
-            self._seg_writer.write(image, dicom_dataset_list, file_path.name, self._seg_labels)
+            self._seg_writer.write(image, dicom_dataset_list, str(file_path), self._seg_labels)
             # TODO: get a class to encapsulate the seg label information.
 
             # Test reading back
-            _ = self._read_from_dcm(file_path.name)
+            _ = self._read_from_dcm(str(file_path))
         except Exception as ex:
             print("DICOMSeg creation failed. Error:\n{}".format(ex))
             raise
 
-    def _read_from_dcm(self, file_path):
+    def _read_from_dcm(self, file_path: str):
         """Read dcm file into pydicom Dataset
 
         Args:

monai/deploy/operators/dicom_series_selector_operator.py

@@ -279,7 +279,7 @@ def _print_instance_properties(obj: object, pre_fix: str = None, print_val=True)
         print(f"{pre_fix}  {attribute}: {type(attr_val)} {attr_val if print_val else ''}")
 
 
-def main():
+def test():
     from pathlib import Path
 
     from monai.deploy.operators.dicom_data_loader_operator import DICOMDataLoaderOperator
@@ -349,4 +349,4 @@ def main():
 """
 
 if __name__ == "__main__":
-    main()
+    test()

monai/deploy/operators/dicom_series_to_volume_operator.py

@@ -337,7 +337,7 @@ def _get_instance_properties(obj: object, not_none: bool = True) -> Dict:
         return prop_dict
 
 
-def main():
+def test():
 
     from pathlib import Path
 
@@ -359,4 +359,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    test()