comments updated

models/bamf_pet_ct_lung_tumor/utils/LungPostProcessor.py

@@ -1,3 +1,14 @@
+"""
+---------------------------------------------------------
+Post processing Module on segmentation output
+---------------------------------------------------------
+
+-------------------------------------------------
+Author: Jithendra Kumar
+Email:  [email protected]
+-------------------------------------------------
+
+"""
 import os
 import shutil
 import SimpleITK as sitk

models/bamf_pet_ct_lung_tumor/utils/NNUnetPETCTRunner.py

@@ -1,8 +1,8 @@
 """
 -------------------------------------------------
-MHub - NNU-Net Runner
+MHub - NNU-Net MultiModality Runner
        This is a base runner for pre-trained 
-       nnunet models.
+       nnunet models
 -------------------------------------------------
 
 -------------------------------------------------
@@ -129,7 +129,7 @@ def export_prob_mask(self, nnunet_out_dir: str, ref_file: InstanceData, output_d
 
     @IO.Instance()
     @IO.Input('in_ct_data', the="input ct data to run nnunet on")
-    @IO.Input('in_pt_data', the="input pt resampled data to run nnunet on")
+    @IO.Input('in_pt_data', the="input pt data to run nnunet on")
     @IO.Output("out_data", 'VOLUME_001.nii.gz', 'nifti:mod=seg:model=nnunet', the="output data from nnunet")
     def task(self, instance: Instance, in_ct_data: InstanceData,in_pt_data: InstanceData, out_data: InstanceData) -> None:
 

models/bamf_pet_ct_lung_tumor/utils/Registration.py

@@ -1,3 +1,14 @@
+"""
+---------------------------------------------------------
+Registration Module using SimpleITK
+---------------------------------------------------------
+
+-------------------------------------------------
+Author: Jithendra Kumar
+Email:  [email protected]
+-------------------------------------------------
+
+"""
 import os
 import shutil
 import SimpleITK as sitk

models/bamf_pet_ct_lung_tumor/utils/SitkNiftiConverter.py

@@ -13,16 +13,16 @@
 from enum import Enum
 from typing import List, Dict, Any
 
-from mhubio.core import Module, Instance, InstanceDataCollection, InstanceData, DataType, FileType
-from mhubio.core.IO import IO
-
 import pydicom
 import shutil
 import os, subprocess
 import pyplastimatch as pypla # type: ignore
 import SimpleITK as sitk
 from pathlib import Path
 
+from mhubio.core import Module, Instance, InstanceDataCollection, InstanceData, DataType, FileType
+from mhubio.core.IO import IO
+
 
 @IO.ConfigInput('in_datas', 'dicom:mod=ct|pt', the="target data that will be converted to nifti")
 @IO.Config('allow_multi_input', bool, False, the='allow multiple input files')