v1.0.4

Author: qchapp

pyproject.toml

@@ -11,7 +11,7 @@ line_length = 79
 
 [project]
 name = "unet_lungs_segmentation"
-version = "1.0.3"
+version = "1.0.4"
 description = "A U-Net model for lung segmentation in mice CT scans."
 authors = [
     { name = "Chappuis Quentin", email = "quentin.chappuis@epfl.ch" }

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = unet-lungs-segmentation
-version = 1.0.1
+version = 1.0.4
 description = 3D U-Net model for the segmentation of the lungs in mice CT scans.
 long_description = file: README.md
 long_description_content_type = text/markdown

src/unet_lungs_segmentation/cli.py

@@ -9,7 +9,7 @@ def process_input_file_predict(input_image_file, threshold, lungs_predict):
 
     if threshold is None:
         threshold = 0.5
-    pred = lungs_predict.segment_lungs(pred, threshold)
+    pred = lungs_predict.segment_lungs(image, threshold)
 
     pt = Path(input_image_file)
     out_file_name = pt.parent / f"{pt.stem}_mask.tif"