mosamatic2 2.0.9__tar.gz → 2.0.10__tar.gz

{mosamatic2-2.0.9 → mosamatic2-2.0.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mosamatic2
-Version: 2.0.9
+Version: 2.0.10
 Summary: 
 Author: Ralph Brecheisen
 Author-email: r.brecheisen@maastrichtuniversity.nl
@@ -29,8 +29,8 @@ Requires-Dist: tensorflow-intel (==2.15.0) ; platform_system == "Windows"
 Requires-Dist: tensorflow-io-gcs-filesystem (==0.31.0) ; platform_system == "Windows"
 Requires-Dist: tensorflow-io-gcs-filesystem (>=0.31.0) ; platform_system == "Darwin" and platform_machine == "arm64"
 Requires-Dist: tensorflow-macos (==2.15.0) ; platform_system == "Darwin" and platform_machine == "arm64"
-Requires-Dist: torch (>=2.8.0)
-Requires-Dist: torchvision (>=0.23.0)
+Requires-Dist: torch (==2.5.1)
+Requires-Dist: torchvision (>=0.20.1)
 Requires-Dist: totalsegmentator (>=2.11.0)
 Requires-Dist: vtk (>=9.5.1)
 Description-Content-Type: text/markdown

{mosamatic2-2.0.9 → mosamatic2-2.0.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mosamatic2"
-version = "2.0.9"
+version = "2.0.10"
 description = ""
 authors = [
     {name = "Ralph Brecheisen", email = "r.brecheisen@maastrichtuniversity.nl"}
@@ -19,8 +19,8 @@ dependencies = [
     "pillow>=11.3.0",
     "python-gdcm>=3.0.26",
     "scipy>=1.15.3",
-    "torch>=2.8.0",
-    "torchvision>=0.23.0",
+    "torch==2.5.1",
+    "torchvision>=0.20.1",
     "tensorflow==2.15.*; platform_system == 'Linux'",
     "tensorflow-intel==2.15.0; platform_system == 'Windows'",
     "tensorflow-macos==2.15.0; platform_system == 'Darwin' and platform_machine == 'arm64'",

{mosamatic2-2.0.9 → mosamatic2-2.0.10}/src/mosamatic2/cli.py

@@ -7,6 +7,7 @@ from mosamatic2.commands import (
     dicom2nifti,
     selectslicefromscans,
     createdicomsummary,
+    defaultpipeline,
 )
 from mosamatic2.core.utils import show_doc_command
 
@@ -35,4 +36,5 @@ main.add_command(createpngsfromsegmentations.createpngsfromsegmentations)
 main.add_command(dicom2nifti.dicom2nifti)
 main.add_command(selectslicefromscans.selectslicefromscans)
 main.add_command(createdicomsummary.createdicomsummary)
+main.add_command(defaultpipeline.defaultpipeline)
 main.add_command(show_doc_command(main)) # Special command to show long description for command

mosamatic2-2.0.10/src/mosamatic2/commands/defaultpipeline.py

@@ -0,0 +1,70 @@
+import click
+
+from mosamatic2.core.pipelines import DefaultPipeline
+
+
+@click.command(help='Runs default L3 analysis pipeline')
+@click.option(
+    '--images', 
+    required=True, 
+    type=click.Path(exists=True), 
+    help='Directory with L3 images',
+)
+@click.option(
+    '--model_files', 
+    required=True, 
+    type=click.Path(), 
+    help='Input directory with AI model files'
+)
+@click.option(
+    '--output', 
+    required=True, 
+    type=click.Path(), 
+    help='Output directory'
+)
+@click.option(
+    '--overwrite', 
+    type=click.BOOL, 
+    default=False, 
+    help='Overwrite [true|false]'
+)
+def defaultpipeline(images, model_files, output, overwrite):
+    """
+    Runs default L3 analysis pipeline
+    
+    Parameters
+    ----------
+    --images : str
+        Directory with L3 images
+
+    --model_files : str
+        Directory with AI model files. This directory can ONLY contain the following
+        files:
+
+        (1) contour_model-1.0.zip
+        (2) model-1.0.zip
+        (3) params-1.0.json
+
+        These files can be downloaded from:
+        https://mosamatic.rbeesoft.nl/wordpress/mosamatic/installation/
+
+    --output : str
+        Path to output directory
+
+    --overwrite : bool
+        Overwrite contents output directory [true|false]
+    """
+    pipeline = DefaultPipeline(
+        inputs={'images': images, 'model_files': model_files},
+        params={
+            'target_size': 512,
+            'model_type': 'tensorflow',
+            'model_version': 1.0,
+            'file_type': 'npy',
+            'fig_width': 10,
+            'fig_height': 10,
+        },
+        output=output,
+        overwrite=overwrite,
+    )
+    pipeline.run()

{mosamatic2-2.0.9 → mosamatic2-2.0.10}/src/mosamatic2/core/tasks/selectslicefromscanstask/selectslicefromscanstask.py

@@ -97,7 +97,12 @@ class SelectSliceFromScansTask(Task):
         for step in range(nr_steps):
             scan_dir = scan_dirs[step]
             scan_name = os.path.split(scan_dir)[1]
-            self.extract_masks(scan_dir)
+            try:
+                self.extract_masks(scan_dir)
+            except Exception as e:
+                LOG.warning(f'Could not extract masks from {scan_dir} [{str(e)}]')
+                self.set_progress(step, nr_steps)
+                continue
             file_path = self.find_slice(scan_dir, vertebra)
             if file_path is not None:
                 extension = '' if file_path.endswith('.dcm') else '.dcm'

mosamatic2-2.0.10/src/mosamatic2/ui/resources/VERSION

@@ -0,0 +1 @@
+2.0.10 

mosamatic2-2.0.9/src/mosamatic2/ui/resources/VERSION

@@ -1 +0,0 @@
-2.0.9