Skip to content

Commit

Permalink
Created DLICV pipelien file for better import | arguments fixed and w…
Browse files Browse the repository at this point in the history
…orking with just the 3 basic(i, o, d)
  • Loading branch information
spirosmaggioros committed Dec 11, 2024
1 parent ed2f01a commit 871d48a
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions DLICV/dlicv_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,20 @@ def run_pipeline(
nps: int = 2,
prev_stage_predictions: Optional[str] = None,
) -> None:
"""
Run dlicv pipeline function
:param in_dir: The input directory
:type in_dir: str
:param out_dir: The output directory
:type out_dir: str
:param device: cpu/cuda/mps
:type device: str
Any other argument is not needed for 99% of you.
Devs should see the code
:rtype: None
"""
f = [0]
if clear_cache:
shutil.rmtree(os.path.join(Path(__file__).parent, "nnunet_results"))
Expand Down

0 comments on commit 871d48a

Please sign in to comment.