spacr 0.2.5__py3-none-any.whl → 0.2.8__py3-none-any.whl

spacr/utils.py

@@ -1,4 +1,4 @@
-import sys, os, re, sqlite3, torch, torchvision, random, string, shutil, cv2, tarfile, glob, psutil, platform
+import sys, os, re, sqlite3, torch, torchvision, random, string, shutil, cv2, tarfile, glob, psutil, platform, gzip, subprocess
 
 import numpy as np
 from cellpose import models as cp_models
@@ -87,6 +87,12 @@ from scipy.stats import f_oneway, kruskal
 from sklearn.cluster import KMeans
 from scipy import stats
 
+def save_settings(settings, name='settings'):
+    
+    settings_df = pd.DataFrame(list(settings.items()), columns=['Key', 'Value'])
+    settings_csv = os.path.join(settings['src'],'settings',f'{name}.csv')
+    os.makedirs(os.path.join(settings['src'],'settings'), exist_ok=True)
+    settings_df.to_csv(settings_csv, index=False)
 
 def print_progress(files_processed, files_to_process, n_jobs, time_ls=None, batch_size=None, operation_type=""):
     if isinstance(files_processed, list):
@@ -116,10 +122,9 @@ def print_progress(files_processed, files_to_process, n_jobs, time_ls=None, batc
         else:
             average_time_img = average_time / batch_size
             time_info = f'Time/batch: {average_time:.3f}sec, Time/image: {average_time_img:.3f}sec, Time_left: {time_left:.3f} min.'
-
-    print(f'Progress: {files_processed}/{files_to_process}, operation_type: {operation_type} {time_info}')
-
-
+    else:
+        time_info = None
+    print(f'Progress: {files_processed}/{files_to_process}, operation_type: {operation_type}, {time_info}')
 
 def reset_mp():
     current_method = get_start_method()
@@ -1649,7 +1654,7 @@ def split_my_dataset(dataset, split_ratio=0.1):
     val_dataset = Subset(dataset, val_indices)
     return train_dataset, val_dataset
 
-def classification_metrics(all_labels, prediction_pos_probs, loader_name, loss, epoch):
+def classification_metrics(all_labels, prediction_pos_probs, loss, epoch):
     """
     Calculate classification metrics for binary classification.
 
@@ -1698,11 +1703,9 @@ def classification_metrics(all_labels, prediction_pos_probs, loader_name, loss,
     else:
         acc_nc = np.nan
     data_dict = {'accuracy': acc_all, 'neg_accuracy': acc_nc, 'pos_accuracy': acc_pc, 'loss':loss.item(),'prauc':pr_auc, 'optimal_threshold':optimal_threshold}
-    data_df = pd.DataFrame(data_dict, index=[str(epoch)+'_'+loader_name]) 
+    data_df = pd.DataFrame(data_dict, index=[str(epoch)]) 
     return data_df
     
-
-
 def compute_irm_penalty(losses, dummy_w, device):
     """
     Computes the Invariant Risk Minimization (IRM) penalty.
@@ -1740,7 +1743,7 @@ def compute_irm_penalty(losses, dummy_w, device):
 #    summary(base_model, (channels, height, width))
 #    return
 
-def choose_model(model_type, device, init_weights=True, dropout_rate=0, use_checkpoint=False, channels=3, height=224, width=224, chan_dict=None, num_classes=2):
+def choose_model(model_type, device, init_weights=True, dropout_rate=0, use_checkpoint=False, channels=3, height=224, width=224, chan_dict=None, num_classes=2, verbose=False):
     """
     Choose a model for classification.
 
@@ -1772,7 +1775,7 @@ def choose_model(model_type, device, init_weights=True, dropout_rate=0, use_chec
         print(f'Invalid model_type: {model_type}. Compatible model_types: {model_types}')
         return
 
-    print(f'\rModel parameters: Architecture: {model_type} init_weights: {init_weights} dropout_rate: {dropout_rate} use_checkpoint: {use_checkpoint}', end='\r', flush=True)
+    print(f'Model parameters: Architecture: {model_type} init_weights: {init_weights} dropout_rate: {dropout_rate} use_checkpoint: {use_checkpoint}', end='\r', flush=True)
     
     if model_type == 'custom':
         
@@ -1783,8 +1786,8 @@ def choose_model(model_type, device, init_weights=True, dropout_rate=0, use_chec
     else:
         print(f'Compatible model_types: {model_types}')
         raise ValueError(f"Invalid model_type: {model_type}")
-
-    print(base_model)
+    if verbose:
+        print(base_model)
     
     return base_model
 
@@ -3449,7 +3452,7 @@ def setup_plot(figuresize, black_background):
     fig, ax = plt.subplots(1, 1, figsize=(figuresize, figuresize))
     return fig, ax
 
-def plot_clusters(ax, embedding, labels, colors, cluster_centers, plot_outlines, plot_points, smooth_lines, figuresize=50, dot_size=50, verbose=False):
+def plot_clusters(ax, embedding, labels, colors, cluster_centers, plot_outlines, plot_points, smooth_lines, figuresize=10, dot_size=50, verbose=False):
     unique_labels = np.unique(labels)
     for cluster_label, color, center in zip(unique_labels, colors, cluster_centers):
         cluster_data = embedding[labels == cluster_label]
@@ -4424,3 +4427,20 @@ def correct_masks(src):
     cell_path = os.path.join(src,'norm_channel_stack', 'cell_mask_stack')
     convert_and_relabel_masks(cell_path)
     _load_and_concatenate_arrays(src, [0,1,2,3], 1, 0, 2)
+
+def count_reads_in_fastq(fastq_file):
+    count = 0
+    with gzip.open(fastq_file, "rt") as f:
+        for _ in f:
+            count += 1
+    return count // 4
+
+
+# Function to determine the CUDA version
+def get_cuda_version():
+    try:
+        output = subprocess.check_output(['nvcc', '--version'], stderr=subprocess.STDOUT).decode('utf-8')
+        if 'release' in output:
+            return output.split('release ')[1].split(',')[0].replace('.', '')
+    except (subprocess.CalledProcessError, FileNotFoundError):
+        return None

{spacr-0.2.5.dist-info → spacr-0.2.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: spacr
-Version: 0.2.5
+Version: 0.2.8
 Summary: Spatial phenotype analysis of crisp screens (SpaCr)
 Home-page: https://github.com/EinarOlafsson/spacr
 Author: Einar Birnir Olafsson
@@ -9,47 +9,54 @@ Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 License-File: LICENSE
-Requires-Dist: torch <3.0,>=2.2.1
-Requires-Dist: torchvision <1.0,>=0.17.1
-Requires-Dist: torch-geometric <3.0,>=2.5.1
-Requires-Dist: numpy <2.0,>=1.26.4
-Requires-Dist: pandas <3.0,>=2.2.1
-Requires-Dist: statsmodels <1.0,>=0.14.1
-Requires-Dist: scikit-image <1.0,>=0.22.0
-Requires-Dist: scikit-learn <2.0,>=1.4.1
-Requires-Dist: seaborn <1.0,>=0.13.2
-Requires-Dist: matplotlib <4.0,>=3.8.3
-Requires-Dist: shap <1.0,>=0.45.0
-Requires-Dist: pillow <11.0,>=10.2.0
-Requires-Dist: imageio <3.0,>=2.34.0
-Requires-Dist: scipy <2.0,>=1.12.0
-Requires-Dist: ipywidgets <9.0,>=8.1.2
-Requires-Dist: mahotas <2.0,>=1.4.13
-Requires-Dist: btrack <1.0,>=0.6.5
-Requires-Dist: trackpy <1.0,>=0.6.2
-Requires-Dist: cellpose <4.0,>=3.0.6
-Requires-Dist: IPython <9.0,>=8.18.1
-Requires-Dist: opencv-python-headless <5.0,>=4.9.0.80
-Requires-Dist: umap-learn <1.0,>=0.5.6
-Requires-Dist: ttkthemes <4.0,>=3.2.2
-Requires-Dist: xgboost <3.0,>=2.0.3
-Requires-Dist: PyWavelets <2.0,>=1.6.0
-Requires-Dist: torchcam <1.0,>=0.4.0
-Requires-Dist: ttf-opensans >=2020.10.30
-Requires-Dist: customtkinter <6.0,>=5.2.2
-Requires-Dist: biopython <2.0,>=1.80
-Requires-Dist: lxml <6.0,>=5.1.0
-Requires-Dist: psutil <6.0,>=5.9.8
-Requires-Dist: gputil <2.0,>=1.4.0
-Requires-Dist: gpustat <2.0,>=1.1.1
-Requires-Dist: pyautogui <1.0,>=0.9.54
-Requires-Dist: huggingface-hub <0.25,>=0.24.0
+Requires-Dist: torch<3.0,>=2.0
+Requires-Dist: torchvision<1.0,>=0.1
+Requires-Dist: torch-geometric<3.0,>=2.5
+Requires-Dist: numpy<2.0,>=1.26.4
+Requires-Dist: bottleneck<2.0,>=1.3.6
+Requires-Dist: numexpr<3.0,>=2.8.4
+Requires-Dist: pandas<3.0,>=2.2.1
+Requires-Dist: statsmodels<1.0,>=0.14.1
+Requires-Dist: scikit-image<1.0,>=0.22.0
+Requires-Dist: scikit-learn<2.0,>=1.4.1
+Requires-Dist: seaborn<1.0,>=0.13.2
+Requires-Dist: matplotlib<4.0,>=3.8.3
+Requires-Dist: shap<1.0,>=0.45.0
+Requires-Dist: pillow<11.0,>=10.2.0
+Requires-Dist: imageio<3.0,>=2.34.0
+Requires-Dist: scipy<2.0,>=1.12.0
+Requires-Dist: ipywidgets<9.0,>=8.1.2
+Requires-Dist: mahotas<2.0,>=1.4.13
+Requires-Dist: btrack<1.0,>=0.6.5
+Requires-Dist: trackpy<1.0,>=0.6.2
+Requires-Dist: cellpose<4.0,>=3.0.6
+Requires-Dist: IPython<9.0,>=8.18.1
+Requires-Dist: opencv-python-headless<5.0,>=4.9.0.80
+Requires-Dist: umap-learn<1.0,>=0.5.6
+Requires-Dist: ttkthemes<4.0,>=3.2.2
+Requires-Dist: xgboost<3.0,>=2.0.3
+Requires-Dist: PyWavelets<2.0,>=1.6.0
+Requires-Dist: torchcam<1.0,>=0.4.0
+Requires-Dist: ttf-opensans>=2020.10.30
+Requires-Dist: customtkinter<6.0,>=5.2.2
+Requires-Dist: biopython<2.0,>=1.80
+Requires-Dist: lxml<6.0,>=5.1.0
+Requires-Dist: psutil<6.0,>=5.9.8
+Requires-Dist: gputil<2.0,>=1.4.0
+Requires-Dist: gpustat<2.0,>=1.1.1
+Requires-Dist: pyautogui<1.0,>=0.9.54
+Requires-Dist: tables<4.0,>=3.8.0
+Requires-Dist: rapidfuzz<4.0,>=3.9
+Requires-Dist: keyring<16.0,>=15.1
+Requires-Dist: screeninfo<1.0,>=0.8.1
+Requires-Dist: ipykernel
+Requires-Dist: huggingface-hub<0.25,>=0.24.0
 Provides-Extra: dev
-Requires-Dist: pytest <3.11,>=3.9 ; extra == 'dev'
+Requires-Dist: pytest<3.11,>=3.9; extra == "dev"
 Provides-Extra: full
-Requires-Dist: opencv-python ; extra == 'full'
+Requires-Dist: opencv-python; extra == "full"
 Provides-Extra: headless
-Requires-Dist: opencv-python-headless ; extra == 'headless'
+Requires-Dist: opencv-python-headless; extra == "headless"
 
 .. |Documentation Status| image:: https://readthedocs.org/projects/spacr/badge/?version=latest
    :target: https://spacr.readthedocs.io/en/latest/?badge=latest

{spacr-0.2.5.dist-info → spacr-0.2.8.dist-info}/RECORD

@@ -1,4 +1,4 @@
-spacr/__init__.py,sha256=pJ7Mm7Kb1DhHIdLmNgMILFVWJ9QAG47pT0M6wtiXl8E,1465
+spacr/__init__.py,sha256=8NZIlJOY2OzRCFjXvqusFL7BfyEJwNqB2lL8QNB-Kgo,1141
 spacr/__main__.py,sha256=bkAJJD2kjIqOP-u1kLvct9jQQCeUXzlEjdgitwi1Lm8,75
 spacr/app_annotate.py,sha256=nEIL7Fle9CDKGo3sucG_03DgjUQt5W1M1IHBIpVBr08,2171
 spacr/app_classify.py,sha256=urTP_wlZ58hSyM5a19slYlBxN0PdC-9-ga0hvq8CGWc,165
@@ -8,23 +8,23 @@ spacr/app_measure.py,sha256=_K7APYIeOKpV6e_LcqabBjvEi7mfq9Fch8175x1x0k8,162
 spacr/app_sequencing.py,sha256=DjG26jy4cpddnV8WOOAIiExtOe9MleVMY4MFa5uTo5w,157
 spacr/app_umap.py,sha256=ZWAmf_OsIKbYvolYuWPMYhdlVe-n2CADoJulAizMiEo,153
 spacr/chris.py,sha256=YlBjSgeZaY8HPy6jkrT_ISAnCMAKVfvCxF0I9eAZLFM,2418
-spacr/core.py,sha256=WBVZfuVJPqI7hXaZVLPWCbe-CCLDNjsCB3nHCF2Axmg,148064
-spacr/deep_spacr.py,sha256=ASBsN4JpHp_3S-91JUsB34IWTjTGPYI7jKV2qZnUR5M,37005
+spacr/core.py,sha256=B2KMyG6IrVtV9d4XQkXVblbvfFhgcMY2o-mtmqDVDic,146519
+spacr/deep_spacr.py,sha256=a2YewgkQvLV-95NYJAutnojvJmX4S8z_wv6Tb-XIgUI,34484
 spacr/graph_learning.py,sha256=1tR-ZxvXE3dBz1Saw7BeVFcrsUFu9OlUZeZVifih9eo,13070
-spacr/gui.py,sha256=NoqaHYjqvyiexfXMsFWrRYjcyqAOhEIYnR-82LkKzdk,7277
-spacr/gui_core.py,sha256=UqX0JE7mCi_4TlVCtEMMTHSUEC2BvZSVUmWzue2HYok,30756
-spacr/gui_elements.py,sha256=ISGt_gEozMx_BWWgXvcgEEKeqvCxfmoCDVHWlrJ1roQ,106811
-spacr/gui_utils.py,sha256=nRH1bEXjrTsfgp01H7ayIkimN0v7G163XjiruGwySP0,28050
-spacr/io.py,sha256=UYISLJgwpwyoTxKy1v1wzFQ6cLX77h2rHh4t5fF1_4w,115461
+spacr/gui.py,sha256=RMg0bgbUpO6JwaWuNVMwuVZ18j4WlER3nW0Eaa0YZ30,7883
+spacr/gui_core.py,sha256=gTgsBzhZ5Q4vl6fT8VwrHWZ6BXWW5FNli8CE3WNMfv0,39986
+spacr/gui_elements.py,sha256=OA514FUVRKAcdu9CFVOt7UEzn1vztakQ-rDyKqV0b9A,129771
+spacr/gui_utils.py,sha256=pq_bmZ527S1j2s6McvqMhHNI05hJycBhHM8GY_jH9Ng,30597
+spacr/io.py,sha256=ZtVNbEom8X8p_KfsuWw0glGwLg6S0CfwwevDPGdfiSc,117342
 spacr/logger.py,sha256=7Zqr3TuuOQLWT32gYr2q1qvv7x0a2JhLANmZcnBXAW8,670
-spacr/measure.py,sha256=A_n7y6cF4FL2VDggLaeeKdoWLx8UeHmUZL54NFM_jvI,54742
-spacr/plot.py,sha256=4o9X76ur2kBe6TtOrbIPfo04iC60OZ1rNJoegBxtLmk,72361
-spacr/sequencing.py,sha256=fHZRnoMSxmhMdadkei3lUeBdckqFyptWdQyWsDW3aaU,83304
-spacr/settings.py,sha256=QS8D7zrhxyArRsavfz2w2bh6VnYJk3rUBwoGdE5Qqfc,61254
+spacr/measure.py,sha256=4rmzH_a5Y0s1qALVi6YRut3xpnkJXs5vzeTPCEf3QS8,54871
+spacr/plot.py,sha256=xVnbML7WpAEzdJdrLeYRk6aPinZSiV2dLAeu4mh7n0k,73963
+spacr/sequencing.py,sha256=y7EB8226B0b0gnGXt6jqBaFVATrM1Y89v3rtHb8XR_k,75746
+spacr/settings.py,sha256=1PE6rQi0JoxbPSwEdEEWlJvrFW-dBkdY0WP461X-4FA,67839
 spacr/sim.py,sha256=FveaVgBi3eypO2oVB5Dx-v0CC1Ny7UPfXkJiiRRodAk,71212
 spacr/sim_app.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spacr/timelapse.py,sha256=KMYCgHzf9LTZe-lWl5mvH2EjbKRE6OhpwdY13wEumGc,39504
-spacr/utils.py,sha256=TT2gb2nmhNSwkHheaOfpDPXhpE90_6Er3nA77gAbO3U,188225
+spacr/utils.py,sha256=2v0N1GJPqGyhxLJGSl9cGaaLQr04ehzqcOxiD-zQ1WI,189142
 spacr/version.py,sha256=axH5tnGwtgSnJHb5IDhiu4Zjk5GhLyAEDRe-rnaoFOA,409
 spacr/resources/font/open_sans/OFL.txt,sha256=bGMoWBRrE2RcdzDiuYiB8A9OVFlJ0sA2imWwce2DAdo,4484
 "spacr/resources/font/open_sans/OpenSans-Italic-VariableFont_wdth,wght.ttf",sha256=QSoWv9h46CRX_fdlqFM3O2d3-PF3R1srnb4zUezcLm0,580280
@@ -92,9 +92,9 @@ spacr/resources/icons/umap.png,sha256=dOLF3DeLYy9k0nkUybiZMe1wzHQwLJFRmgccppw-8b
 spacr/resources/models/cp/toxo_plaque_cyto_e25000_X1120_Y1120.CP_model,sha256=z8BbHWZPRnE9D_BHO0fBREE85c1vkltDs-incs2ytXQ,26566572
 spacr/resources/models/cp/toxo_plaque_cyto_e25000_X1120_Y1120.CP_model_settings.csv,sha256=fBAGuL_B8ERVdVizO3BHozTDSbZUh1yFzsYK3wkQN68,420
 spacr/resources/models/cp/toxo_pv_lumen.CP_model,sha256=2y_CindYhmTvVwBH39SNILF3rI3x9SsRn6qrMxHy3l0,26562451
-spacr-0.2.5.dist-info/LICENSE,sha256=SR-2MeGc6SCM1UORJYyarSWY_A-JaOMFDj7ReSs9tRM,1083
-spacr-0.2.5.dist-info/METADATA,sha256=fiV7Ivmrb4xlR-6ACV2hT5in8bRgPkiqZrB3K4TP-Ds,5194
-spacr-0.2.5.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-spacr-0.2.5.dist-info/entry_points.txt,sha256=BMC0ql9aNNpv8lUZ8sgDLQMsqaVnX5L535gEhKUP5ho,296
-spacr-0.2.5.dist-info/top_level.txt,sha256=GJPU8FgwRXGzKeut6JopsSRY2R8T3i9lDgya42tLInY,6
-spacr-0.2.5.dist-info/RECORD,,
+spacr-0.2.8.dist-info/LICENSE,sha256=SR-2MeGc6SCM1UORJYyarSWY_A-JaOMFDj7ReSs9tRM,1083
+spacr-0.2.8.dist-info/METADATA,sha256=6FX9b07eeM3Ogd2aSrRfgA3qKVW3d321cvsSEonD2Yk,5388
+spacr-0.2.8.dist-info/WHEEL,sha256=HiCZjzuy6Dw0hdX5R3LCFPDmFS4BWl8H-8W39XfmgX4,91
+spacr-0.2.8.dist-info/entry_points.txt,sha256=BMC0ql9aNNpv8lUZ8sgDLQMsqaVnX5L535gEhKUP5ho,296
+spacr-0.2.8.dist-info/top_level.txt,sha256=GJPU8FgwRXGzKeut6JopsSRY2R8T3i9lDgya42tLInY,6
+spacr-0.2.8.dist-info/RECORD,,

{spacr-0.2.5.dist-info → spacr-0.2.8.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: setuptools (72.2.0)
 Root-Is-Purelib: true
 Tag: py3-none-any