fastMONAI 0.3.2__py3-none-any.whl → 0.3.3__py3-none-any.whl

fastMONAI/__init__.py

@@ -1 +1 @@
-__version__ = "0.3.2"
+__version__ = "0.3.3"

fastMONAI/external_data.py

@@ -257,6 +257,7 @@ def download_medmnist3d_dataset(study: str, path: (str, Path) = '../data',
     dataset_file_path = path / f'{study}.npz'
 
     try:
+        #todo: check if dataset is downloaded
         download_url(url=MURLs.MEDMNIST_DICT[study], filepath=dataset_file_path)
     except:
         raise ValueError(f"Dataset '{study}' does not exist.")

fastMONAI/vision_data.py

@@ -110,19 +110,19 @@ class MedImageDataLoaders(DataLoaders):
 # %% ../nbs/02_vision_data.ipynb 16
 @typedispatch
 def show_batch(x: MedImage, y, samples, ctxs=None, max_n=6, nrows=None, 
-               ncols=None, figsize=None, channel: int = 0, indices=None, 
-               anatomical_plane: int = 0, **kwargs):
-    """Showing a batch of samples for classification and regression tasks."""
-    
+               ncols=None, figsize=None, channel=0, indices=None,
+               anatomical_plane=0, **kwargs):
+    '''Showing a batch of samples for classification and regression tasks.'''
+
     if ctxs is None: 
         ctxs = get_grid(min(len(samples), max_n), nrows=nrows, ncols=ncols, figsize=figsize)
-    
+        
     n = 1 if y is None else 2
     
     for i in range(n):
         ctxs = [
-            b.show(ctx=c, channel=channel, indices=indices, anatomical_plane=anatomical_plane, **kwargs) 
-            for b, c, _ in zip(samples.itemgot(i), ctxs, range(max_n))
+            b.show(ctx=c, channel=channel, indices=indices, anatomical_plane=anatomical_plane, **kwargs)
+            for b,c,_ in zip(samples.itemgot(i),ctxs,range(max_n))
         ]
 
     plt.tight_layout()
@@ -131,9 +131,9 @@ def show_batch(x: MedImage, y, samples, ctxs=None, max_n=6, nrows=None,
 
 # %% ../nbs/02_vision_data.ipynb 17
 @typedispatch
-def show_batch(x: MedImage, y: MedMask, samples, ctxs=None, max_n=6, nrows: int = None,
-               ncols: int = None, figsize=None, channel: int = 0, indices: int = None,
-               anatomical_plane: int = 0, **kwargs):
+def show_batch(x: MedImage, y: MedMask, samples, ctxs=None, max_n=6, nrows=None,
+               ncols=None, figsize=None, channel=0, indices=None,
+               anatomical_plane=0, **kwargs):
     """Showing a batch of decoded segmentation samples."""
 
     nrows, ncols = min(len(samples), max_n), x.shape[1] + 1
@@ -226,7 +226,7 @@ def show_results(x: MedImage, y: MedMask, samples, outs, ctxs=None, max_n: int =
 
 # %% ../nbs/02_vision_data.ipynb 23
 @typedispatch
-def plot_top_losses(x: MedImage, y, samples, outs, raws, losses, nrows: int = None, 
+def plot_top_losses(x: MedImage, y: TensorCategory, samples, outs, raws, losses, nrows: int = None, 
                     ncols: int = None, figsize=None, channel: int = 0, indices: int = None, 
                     anatomical_plane: int = 0, **kwargs):
     """Show images in top_losses along with their prediction, actual, loss, and probability of actual class."""

fastMONAI/vision_inference.py

@@ -4,15 +4,16 @@
 __all__ = ['inference', 'refine_binary_pred_mask']
 
 # %% ../nbs/06_vision_inference.ipynb 1
-import numpy as np
+from copy import copy
 from pathlib import Path
-from torchio import Resize
+import torch
+import numpy as np
 from scipy.ndimage import label
-from .vision_core import *
-from .vision_augmentation import do_pad_or_crop
 from skimage.morphology import remove_small_objects
 from SimpleITK import DICOMOrient, GetArrayFromImage
-from copy import copy
+from torchio import Resize
+from .vision_core import *
+from .vision_augmentation import do_pad_or_crop
 
 # %% ../nbs/06_vision_inference.ipynb 3
 def _to_original_orientation(input_img, org_orientation):
@@ -74,7 +75,7 @@ def inference(learn_inf, reorder, resample, fn: (str, Path) = '',
 def refine_binary_pred_mask(pred_mask, 
                             remove_size: (int, float) = None,
                             percentage: float = 0.2,
-                            verbose: bool = False) -> np.ndarray:
+                            verbose: bool = False) -> torch.Tensor:
     """Removes small objects from the predicted binary mask.
 
     Args:
@@ -102,4 +103,4 @@ def refine_binary_pred_mask(pred_mask,
     processed_mask = remove_small_objects(
         labeled_mask, min_size=small_objects_threshold)
 
-    return np.where(processed_mask > 0, 1., 0.)
+    return torch.Tensor(processed_mask > 0).float()                          

{fastMONAI-0.3.2.dist-info → fastMONAI-0.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fastMONAI
-Version: 0.3.2
+Version: 0.3.3
 Summary: fastMONAI library
 Home-page: https://github.com/MMIV-ML/fastMONAI
 Author: Satheshkumar Kaliyugarasan

{fastMONAI-0.3.2.dist-info → fastMONAI-0.3.3.dist-info}/RECORD

@@ -1,20 +1,20 @@
-fastMONAI/__init__.py,sha256=vNiWJ14r_cw5t_7UDqDQIVZvladKFGyHH2avsLpN7Vg,22
+fastMONAI/__init__.py,sha256=8KcCYTXH99C2-gCLuPILJvtT9YftRWJsartIx6TQ2ZY,22
 fastMONAI/_modidx.py,sha256=AGOW75o_d2Vd7CBku-4zz53OzUqkp4QqTEqsucyU7jg,27510
 fastMONAI/dataset_info.py,sha256=w5LKGmEzFtlqhxqzhFLWUfyVkUL0EYypZQCH9Ay6jgg,4948
-fastMONAI/external_data.py,sha256=RWk_MVy4x7MuKz2A6cRAThz2diV0546gYZchLQtU3wg,10970
+fastMONAI/external_data.py,sha256=Hq4GPEWsBi9-fyk9r3WlBdUZjfNzeaKe5uqTwFD8qfU,11016
 fastMONAI/research_utils.py,sha256=LZu62g8BQAVYS4dD7qDsKHJXZnDd1uLkJ6LoaMDhUhk,590
 fastMONAI/utils.py,sha256=9I5nl6Sb0NbTxhr6FDnW4dapDhzPtmxGcJeXkkY4v3E,1406
 fastMONAI/vision_all.py,sha256=qAsncBCDTx3Ae8Hw44TK4E5wflO-8JbRWBu0CRLLGnY,360
 fastMONAI/vision_augmentation.py,sha256=lAlrLm8jbXRmk9a6e8_o_CNTS6Pyp-KKNXwjpelUUJc,9070
 fastMONAI/vision_core.py,sha256=KDoLS9UYNsh5nPIYc2JiDjWBLy1IRSpjH9cqEHjk4Vw,7428
-fastMONAI/vision_data.py,sha256=uFs4SGPeFHwNL9tLhuz4yBYG-s4KBEh_YC8SBhVCj5I,10530
-fastMONAI/vision_inference.py,sha256=wd-zJbL28B3N5T_WQZXhjI8atNyAEVQUY9PGhTU7F9s,3699
+fastMONAI/vision_data.py,sha256=QZqHK35L9sU9BiQ2nIqGzOEBAGHfuke7YloyZrPkYto,10491
+fastMONAI/vision_inference.py,sha256=k-rGKhVtdrZT9GvQEglEa0sZ283ue16mEPTZ-ZSKc5Q,3744
 fastMONAI/vision_loss.py,sha256=NrHnk1yD4EBKsp6aippppXU4l-mwmsZOqE_bsZP3ZNI,3591
 fastMONAI/vision_metrics.py,sha256=CVxdOBPaMJT6Mo5jF3WoQj6a3C-_FsnBicMAU_ZrFS8,3549
 fastMONAI/vision_plot.py,sha256=S2_yE0VcAwECgvXHb1YMJbTZq4iREh-VIgPShLcUma0,3095
-fastMONAI-0.3.2.dist-info/LICENSE,sha256=xV8xoN4VOL0uw9X8RSs2IMuD_Ss_a9yAbtGNeBWZwnw,11337
-fastMONAI-0.3.2.dist-info/METADATA,sha256=dEvtIMtl-0ppdOzC_jjZpPKo33jPmfW_wxjWiqN8C5s,5770
-fastMONAI-0.3.2.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-fastMONAI-0.3.2.dist-info/entry_points.txt,sha256=mVBsykSXMairzzk3hJaQ8c-UiwUZqGnn4aFZ24CpsBM,40
-fastMONAI-0.3.2.dist-info/top_level.txt,sha256=o8y7SWF9odtnIT3jvYtUn9okbJRlaAMCy7oPFCeQvQ8,10
-fastMONAI-0.3.2.dist-info/RECORD,,
+fastMONAI-0.3.3.dist-info/LICENSE,sha256=xV8xoN4VOL0uw9X8RSs2IMuD_Ss_a9yAbtGNeBWZwnw,11337
+fastMONAI-0.3.3.dist-info/METADATA,sha256=xVkZtZdie9sgwLhpKNl8yZaA7YKFDDFLYB-PDo5yOeA,5770
+fastMONAI-0.3.3.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+fastMONAI-0.3.3.dist-info/entry_points.txt,sha256=mVBsykSXMairzzk3hJaQ8c-UiwUZqGnn4aFZ24CpsBM,40
+fastMONAI-0.3.3.dist-info/top_level.txt,sha256=o8y7SWF9odtnIT3jvYtUn9okbJRlaAMCy7oPFCeQvQ8,10
+fastMONAI-0.3.3.dist-info/RECORD,,