From c82fb8ce56413adc34bbd563725417ef58aa92be Mon Sep 17 00:00:00 2001 From: s214735 <s214735@dtu.dk> Date: Fri, 27 Dec 2024 12:41:16 +0100 Subject: [PATCH] More changes --- docs/models.md | 5 ++++- qim3d/viz/colormaps/_segmentation.py | 16 ++++++++-------- 2 files changed, 12 insertions(+), 9 deletions(-) diff --git a/docs/models.md b/docs/models.md index 5e058b8e..a3e0f840 100644 --- a/docs/models.md +++ b/docs/models.md @@ -2,4 +2,7 @@ The `qim3d` library aims to ease the creation of ML models for volumetric images -::: qim3d.ml.models._unet \ No newline at end of file +::: qim3d.ml.models + options: + members: + - UNet \ No newline at end of file diff --git a/qim3d/viz/colormaps/_segmentation.py b/qim3d/viz/colormaps/_segmentation.py index b6ee2f88..cf96e5b7 100644 --- a/qim3d/viz/colormaps/_segmentation.py +++ b/qim3d/viz/colormaps/_segmentation.py @@ -61,10 +61,10 @@ def segmentation( ```python import qim3d - cmap_bright = qim3d.viz.colormaps.objects(num_labels=100, style = 'bright', first_color_background=True, background_color="black", min_dist=0.7) - cmap_soft = qim3d.viz.colormaps.objects(num_labels=100, style = 'soft', first_color_background=True, background_color="black", min_dist=0.2) - cmap_earth = qim3d.viz.colormaps.objects(num_labels=100, style = 'earth', first_color_background=True, background_color="black", min_dist=0.8) - cmap_ocean = qim3d.viz.colormaps.objects(num_labels=100, style = 'ocean', first_color_background=True, background_color="black", min_dist=0.9) + cmap_bright = qim3d.viz.colormaps.segmentation(num_labels=100, style = 'bright', first_color_background=True, background_color="black", min_dist=0.7) + cmap_soft = qim3d.viz.colormaps.segmentation(num_labels=100, style = 'soft', first_color_background=True, background_color="black", min_dist=0.2) + cmap_earth = qim3d.viz.colormaps.segmentation(num_labels=100, style = 'earth', first_color_background=True, background_color="black", min_dist=0.8) + cmap_ocean = qim3d.viz.colormaps.segmentation(num_labels=100, style = 'ocean', first_color_background=True, background_color="black", min_dist=0.9) display(cmap_bright) display(cmap_soft) @@ -77,11 +77,11 @@ def segmentation( import qim3d vol = qim3d.examples.cement_128x128x128 - binary = qim3d.processing.filters.gaussian(vol, sigma = 2) < 60 - labeled_volume, num_labels = qim3d.processing.operations.watershed(binary) + binary = qim3d.filters.gaussian(vol, sigma = 2) < 60 + labeled_volume, num_labels = qim3d.segmentation.watershed(binary) - cmap = qim3d.viz.colormaps.objects(num_labels, style = 'bright') - qim3d.viz.slicer(labeled_volume, axis = 1, cmap=cmap) + cmap = qim3d.viz.colormaps.segmentation(num_labels, style = 'bright') + qim3d.viz.slicer(labeled_volume, slice_axis = 1, color_map=cmap) ```  -- GitLab