reformat

Signed-off-by: Can-Zhao <[email protected]>

Author: Can-Zhao

generation/maisi/scripts/diff_model_create_training_data.py

@@ -17,12 +17,11 @@
 import os
 from pathlib import Path
 
+import monai
 import nibabel as nib
 import numpy as np
 import torch
 import torch.distributed as dist
-
-import monai
 from monai.transforms import Compose
 from monai.utils import set_determinism
 

generation/maisi/scripts/diff_model_infer.py

@@ -21,12 +21,11 @@
 import numpy as np
 import torch
 import torch.distributed as dist
-from tqdm import tqdm
-
 from monai.inferers import sliding_window_inference
-from monai.utils import set_determinism
-from monai.networks.schedulers import RFlowScheduler
 from monai.inferers.inferer import SlidingWindowInferer
+from monai.networks.schedulers import RFlowScheduler
+from monai.utils import set_determinism
+from tqdm import tqdm
 
 from .diff_model_setting import initialize_distributed, load_config, setup_logging
 from .sample import ReconModel, check_input
@@ -179,7 +178,7 @@ def run_inference(
                 image, _ = noise_scheduler.step(model_output, t, image, next_t)  # type: ignore
 
         inferer = SlidingWindowInferer(
-            roi_size=[80,80,80],
+            roi_size=[80, 80, 80],
             sw_batch_size=1,
             progress=True,
             mode="gaussian",

generation/maisi/scripts/diff_model_setting.py

@@ -17,7 +17,6 @@
 
 import torch
 import torch.distributed as dist
-
 from monai.utils import RankFilter
 
 

generation/maisi/scripts/diff_model_train.py

@@ -18,17 +18,16 @@
 from datetime import datetime
 from pathlib import Path
 
+import monai
 import torch
 import torch.distributed as dist
-from torch.amp import GradScaler, autocast
-from torch.nn.parallel import DistributedDataParallel
-
-import monai
 from monai.data import DataLoader, partition_dataset
-from monai.transforms import Compose
-from monai.utils import first
 from monai.networks.schedulers import RFlowScheduler
 from monai.networks.schedulers.ddpm import DDPMPredictionType
+from monai.transforms import Compose
+from monai.utils import first
+from torch.amp import GradScaler, autocast
+from torch.nn.parallel import DistributedDataParallel
 
 from .diff_model_setting import initialize_distributed, load_config, setup_logging
 from .utils import define_instance

generation/maisi/scripts/infer_controlnet.py

@@ -18,7 +18,7 @@
 
 import torch
 import torch.distributed as dist
-from monai.data import decollate_batch, MetaTensor
+from monai.data import MetaTensor, decollate_batch
 from monai.networks.utils import copy_model_state
 from monai.transforms import SaveImage
 from monai.utils import RankFilter

generation/maisi/scripts/inference.py

@@ -14,15 +14,16 @@
 import json
 import logging
 import os
-import tempfile
 import sys
+import tempfile
 
 import monai
 import torch
 from monai.apps import download_url
 from monai.config import print_config
 from monai.transforms import LoadImage, Orientation
 from monai.utils import set_determinism
+
 from scripts.sample import LDMSampler, check_input
 from scripts.utils import define_instance
 from scripts.utils_plot import find_label_center_loc, get_xyz_plot, show_image

generation/maisi/scripts/sample.py

@@ -19,17 +19,17 @@
 
 import monai
 import torch
-from monai.inferers.inferer import DiffusionInferer
 from monai.data import MetaTensor
 from monai.inferers import sliding_window_inference
+from monai.inferers.inferer import DiffusionInferer
 from monai.transforms import Compose, SaveImage
 from monai.utils import set_determinism
 from tqdm import tqdm
 
 from .augmentation import augmentation
 from .find_masks import find_masks
-from .utils import binarize_labels, general_mask_generation_post_process, get_body_region_index_from_mask, remap_labels
 from .quality_check import is_outlier
+from .utils import binarize_labels, general_mask_generation_post_process, get_body_region_index_from_mask, remap_labels
 
 
 class ReconModel(torch.nn.Module):

generation/maisi/scripts/utils.py

@@ -11,9 +11,9 @@
 
 import copy
 import json
+import logging
 import math
 import os
-import logging
 from argparse import Namespace
 from datetime import timedelta
 from typing import Any, Sequence
@@ -22,11 +22,11 @@
 import skimage
 import torch
 import torch.distributed as dist
-from monai.transforms.utils_morphological_ops import dilate, erode
 from monai.bundle import ConfigParser
 from monai.config import DtypeLike, NdarrayOrTensor
 from monai.data import CacheDataset, DataLoader, partition_dataset
 from monai.transforms import Compose, EnsureTyped, Lambdad, LoadImaged, Orientationd
+from monai.transforms.utils_morphological_ops import dilate, erode
 from monai.utils import TransformBackends, convert_data_type, convert_to_dst_type, get_equivalent_dtype
 from scipy import stats
 from torch import Tensor