neuro-sam 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

neuro_sam/napari_utils/punet_widget.py

@@ -15,14 +15,8 @@ import napari
 from napari.qt.threading import thread_worker
 
 # Import the model class
-# Assuming running from root, so punet is a package
-try:
-    from neuro_sam.punet.punet_inference import run_inference_volume
-except ImportError:
-    # Fallback if running from a different context, try to append path
-    import sys
-    sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'punet'))
-    from neuro_sam.punet_inference import run_inference_volume
+# Assumes installed package structure
+from neuro_sam.punet.punet_inference import run_inference_volume
 
 
 

neuro_sam/punet/prob_unet_deepd3.py

@@ -10,9 +10,9 @@ import torch.nn.functional as F
 from torch.distributions import Normal, Independent, kl
 import numpy as np
 
-from deepd3_model import DeepD3Model
-from unet_blocks import *
-from utils import init_weights, init_weights_orthogonal_normal, l2_regularisation
+from .deepd3_model import DeepD3Model
+from .unet_blocks import *
+from .utils import init_weights, init_weights_orthogonal_normal, l2_regularisation
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 

neuro_sam/punet/prob_unet_with_tversky.py

@@ -12,8 +12,8 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 from torch.distributions import kl
-from deepd3_model import DeepD3Model
-from prob_unet_deepd3 import AxisAlignedConvGaussian, Fcomb
+from .deepd3_model import DeepD3Model
+from .prob_unet_deepd3 import AxisAlignedConvGaussian, Fcomb
 
 
 class TverskyLoss(nn.Module):

neuro_sam/punet/punet_inference.py

@@ -7,7 +7,7 @@ import torch
 import tifffile as tiff
 from tqdm import tqdm
 
-from prob_unet_with_tversky import ProbabilisticUnetDualLatent
+from .prob_unet_with_tversky import ProbabilisticUnetDualLatent
 
 
 def pad_to_multiple(img: np.ndarray, multiple: int = 32):

neuro_sam/punet/unet_blocks.py

@@ -2,7 +2,7 @@ import torch
 import torch.nn as nn
 from torch.autograd import Variable
 import numpy as np
-from utils import init_weights
+from .utils import init_weights
 
 class DownConvBlock(nn.Module):
     """

{neuro_sam-0.1.4.dist-info → neuro_sam-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: neuro-sam
-Version: 0.1.4
+Version: 0.1.6
 Summary: Neuro-SAM: Foundation Models for Dendrite and Dendritic Spine Segmentation
 Author-email: Nipun Arora <nipunarora8@yahoo.com>
 License: MIT License

{neuro_sam-0.1.4.dist-info → neuro_sam-0.1.6.dist-info}/RECORD

@@ -30,19 +30,19 @@ neuro_sam/napari_utils/color_utils.py,sha256=Hf5R8f0rh7b9CY1VT72o3tLGfGnnjRREkX8
 neuro_sam/napari_utils/contrasting_color_system.py,sha256=a-lt_3zJLDL9YyIdWJhFDGMYzBb6yH85cV7BNCabbdI,6771
 neuro_sam/napari_utils/main_widget.py,sha256=yahfPLwmhBt_hImpRykIObzfMwbVZvVJTEKKzMZ11bw,48588
 neuro_sam/napari_utils/path_tracing_module.py,sha256=0mMAtrMmtgK_ujMzaWzIguYVDPr8nfzalaTAwgF3NaQ,44062
-neuro_sam/napari_utils/punet_widget.py,sha256=GzY8fGzx5xvitkIiD24TjWktW1ex4F7ujYjCu3Rd6xA,17058
+neuro_sam/napari_utils/punet_widget.py,sha256=bhg2Iw8I46mCbJ_6ocCpWK4SRILStDFtrW2TtupRi4k,16790
 neuro_sam/napari_utils/segmentation_model.py,sha256=uAnNvMdZVrtlaST36TpCdyoYiaNtVsQLeaPcroD-aT0,33939
 neuro_sam/napari_utils/segmentation_module.py,sha256=TPn_x53uwf-WVC0vTleX2g9fCFXXWNAGtuB8Ht5vmAM,28909
 neuro_sam/napari_utils/visualization_module.py,sha256=JtZlBoKlfIwVLa2Sqg7b2KTr07fNlAcwR0M7fHsn2oM,24723
 neuro_sam/punet/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 neuro_sam/punet/deepd3_model.py,sha256=nGVEqzCPz_E4cFA6QmknW2CffDcjxH7VsdYAyTdAtY0,7509
-neuro_sam/punet/prob_unet_deepd3.py,sha256=qZcKo4dZeEemqD7PxLHZu5YB_h3Q0--W3vTRhXjYxS0,14695
-neuro_sam/punet/prob_unet_with_tversky.py,sha256=D-Ur5AfGsVl2dH35jbZL8BaHdJHzbTrHLsS-cagBzoQ,12843
-neuro_sam/punet/punet_inference.py,sha256=U3ne-sP1y732PISFYlV50cPbXN6eEREQue1Jf6w1Io8,8032
+neuro_sam/punet/prob_unet_deepd3.py,sha256=syXNleUVrfYtmVveN9G461oAhumxsijsavps8in4VRw,14698
+neuro_sam/punet/prob_unet_with_tversky.py,sha256=2dBbO_BEHbhYWBXW7rXQX6s2DnqoTgBKkgk6VkgN-Ds,12845
+neuro_sam/punet/punet_inference.py,sha256=v5ufB2Zz5WfgfFZ5-rDjBEobpr5gy-HKPPWZpCALV7A,8033
 neuro_sam/punet/run_inference.py,sha256=c9ATKWJvhOzNEaww_sUCI5fFS1q0bQ4GYUwNUqxWcwA,5312
-neuro_sam/punet/unet_blocks.py,sha256=xc-nZql_CSsyEeCxKIY-WIhiXG1KyRiCSIESztGJKEs,3170
+neuro_sam/punet/unet_blocks.py,sha256=ZRNKay9P3OnJ0PmtKXw_iSgUyRE1DkkGefGXwSbYZGY,3171
 neuro_sam/punet/utils.py,sha256=ibwcpkqqZ3_3Afz2VYxzplz8_8FWQ5qYQqjJiKS8hIo,1786
-neuro_sam-0.1.4.dist-info/licenses/LICENSE,sha256=akmTIN8IuZn3Y7UK_8qVQnyKDWSDcVUwB8RPGNXCojw,1068
+neuro_sam-0.1.6.dist-info/licenses/LICENSE,sha256=akmTIN8IuZn3Y7UK_8qVQnyKDWSDcVUwB8RPGNXCojw,1068
 sam2/__init__.py,sha256=uHyh6VzVS4F2box0rPDpN5UmOVKeQNK0CIaTKG9JQZ4,395
 sam2/automatic_mask_generator.py,sha256=Zt8mbb4UQSMFrjOY8OwbshswOpMhaxAtdn5sTuXUw9c,18461
 sam2/benchmark.py,sha256=m3o1BriIQuwJAx-3zQ_B0_7YLhN84G28oQSV5sGA3ak,2811
@@ -86,8 +86,8 @@ sam2/utils/__init__.py,sha256=NL2AacVHZOe41zp4kF2-ZGcUCi9zFwh1Eo9spNjN0Ko,197
 sam2/utils/amg.py,sha256=t7MwkOKvcuBNu4FcjzKv9BpO0av5Zo9itZ8b3WQMpdg,12842
 sam2/utils/misc.py,sha256=AWAMAcFhzQedcQb7HU2oRc-RqjGrK87K-MsVG21tIKI,13090
 sam2/utils/transforms.py,sha256=ujpk9GAMYvIJIGpt87QOP88TPtrjL61liDG7DCptEUY,4885
-neuro_sam-0.1.4.dist-info/METADATA,sha256=SCUeOtRyt-711lsALQT9aJCOsd8cK_xyYfIMhVzc6pE,9618
-neuro_sam-0.1.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-neuro_sam-0.1.4.dist-info/entry_points.txt,sha256=K0SaLior6N1REi_K0O4962Wn4DhTSos_Sx6Enx0mIm4,105
-neuro_sam-0.1.4.dist-info/top_level.txt,sha256=yPbWxFcw79sErTk8zohihUHMK9LL31i3bXir2MrS4OQ,15
-neuro_sam-0.1.4.dist-info/RECORD,,
+neuro_sam-0.1.6.dist-info/METADATA,sha256=P7v6cbsVv3H4usDScH0_8pVQsarkXpUXDO-Ephq6adg,9618
+neuro_sam-0.1.6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+neuro_sam-0.1.6.dist-info/entry_points.txt,sha256=K0SaLior6N1REi_K0O4962Wn4DhTSos_Sx6Enx0mIm4,105
+neuro_sam-0.1.6.dist-info/top_level.txt,sha256=yPbWxFcw79sErTk8zohihUHMK9LL31i3bXir2MrS4OQ,15
+neuro_sam-0.1.6.dist-info/RECORD,,