careamics 0.0.7__py3-none-any.whl → 0.0.9__py3-none-any.whl

careamics/lvae_training/eval_utils.py

@@ -107,6 +107,15 @@ def get_first_index(bin_count, quantile):
     return None
 
 
+def get_device():
+    if torch.cuda.is_available():
+        return "cuda"
+    elif torch.backends.mps.is_available():
+        return "mps"
+    else:
+        return "cpu"
+
+
 def show_for_one(
     idx,
     val_dset,
@@ -470,6 +479,7 @@ def get_predictions(
     dset: Dataset,
     batch_size: int,
     tile_size: Optional[tuple[int, int]] = None,
+    grid_size: Optional[int] = None,
     mmse_count: int = 1,
     num_workers: int = 4,
 ) -> tuple[dict, dict, dict]:
@@ -510,11 +520,12 @@ def get_predictions(
                 dset=d,
                 batch_size=batch_size,
                 tile_size=tile_size,
+                grid_size=grid_size,
                 mmse_count=mmse_count,
                 num_workers=num_workers,
             )
             # get filename without extension and path
-            filename = str(d._fpath).split("/")[-1].split(".")[0]
+            filename = d._fpath.name
             multifile_stitched_predictions[filename] = stitched_predictions
             multifile_stitched_stds[filename] = stitched_stds
         return (
@@ -527,11 +538,12 @@ def get_predictions(
             dset=dset,
             batch_size=batch_size,
             tile_size=tile_size,
+            grid_size=grid_size,
             mmse_count=mmse_count,
             num_workers=num_workers,
         )
         # get filename without extension and path
-        filename = str(dset._fpath).split("/")[-1].split(".")[0]
+        filename = dset._fpath.name
         return (
             {filename: stitched_predictions},
             {filename: stitched_stds},
@@ -550,6 +562,8 @@ def get_single_file_predictions(
     if tile_size and grid_size:
         dset.set_img_sz(tile_size, grid_size)
 
+    device = get_device()
+
     dloader = DataLoader(
         dset,
         pin_memory=False,
@@ -558,14 +572,14 @@ def get_single_file_predictions(
         batch_size=batch_size,
     )
     model.eval()
-    model.cuda()
+    model.to(device)
     tiles = []
     logvar_arr = []
     with torch.no_grad():
         for batch in tqdm(dloader, desc="Predicting tiles"):
             inp, tar = batch
-            inp = inp.cuda()
-            tar = tar.cuda()
+            inp = inp.to(device)
+            tar = tar.to(device)
 
             # get model output
             rec, _ = model(inp)
@@ -589,10 +603,13 @@ def get_single_file_mmse(
     dset: Dataset,
     batch_size: int,
     tile_size: Optional[tuple[int, int]] = None,
+    grid_size: Optional[int] = None,
     mmse_count: int = 1,
     num_workers: int = 4,
 ) -> tuple[np.ndarray, np.ndarray]:
     """Get patch-wise predictions from a model for a single file dataset."""
+    device = get_device()
+
     dloader = DataLoader(
         dset,
         pin_memory=False,
@@ -600,18 +617,19 @@ def get_single_file_mmse(
         shuffle=False,
         batch_size=batch_size,
     )
-    if tile_size:
-        dset.set_img_sz(tile_size, tile_size[-1] // 2)
+    if tile_size and grid_size:
+        dset.set_img_sz(tile_size, grid_size)
+
     model.eval()
-    model.cuda()
+    model.to(device)
     tile_mmse = []
     tile_stds = []
     logvar_arr = []
     with torch.no_grad():
         for batch in tqdm(dloader, desc="Predicting tiles"):
             inp, tar = batch
-            inp = inp.cuda()
-            tar = tar.cuda()
+            inp = inp.to(device)
+            tar = tar.to(device)
 
             rec_img_list = []
             for _ in range(mmse_count):

{careamics-0.0.7.dist-info → careamics-0.0.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: careamics
-Version: 0.0.7
+Version: 0.0.9
 Summary: Toolbox for running N2V and friends.
 Project-URL: homepage, https://careamics.github.io/
 Project-URL: repository, https://github.com/CAREamics/careamics

{careamics-0.0.7.dist-info → careamics-0.0.9.dist-info}/RECORD

@@ -106,7 +106,7 @@ careamics/losses/lvae/loss_utils.py,sha256=QxzA2N1TglR4H0X0uyTWWytDagE1lA9IB_TK1
 careamics/losses/lvae/losses.py,sha256=wHT1dx04BZ_OI-_S7cFQ5hFmMetm6FSnuZfwZBBtIpY,17977
 careamics/lvae_training/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 careamics/lvae_training/calibration.py,sha256=xHbiLcY2csYos3s7rRSqp7P7G-9wzULcSo1JfVzfIjE,7239
-careamics/lvae_training/eval_utils.py,sha256=7N1thslU4IU1lM1tGg3-wa8AFf5_R2lOSQ7ZZ91AUII,30030
+careamics/lvae_training/eval_utils.py,sha256=FxZmmT6vMRluLYnnCEQtLcz5Q45OAqmxXbQo6KPbQEk,30372
 careamics/lvae_training/get_config.py,sha256=dwVfaQS7nzjQss0E1gGLUpQpjPcOWwLgIhbu3Z0I1rg,3068
 careamics/lvae_training/lightning_module.py,sha256=ryr7iHqCMzCl5esi6_gEcnKFDQkMrw0EXK9Zfgv1Nek,27186
 careamics/lvae_training/metrics.py,sha256=KTDAKhe3vh-YxzGibjtkIG2nnUyujbnwqX4xGwaRXwE,6718
@@ -171,8 +171,8 @@ careamics/utils/ram.py,sha256=tksyn8dVX_iJXmrDZDGub32hFZWIaNxnMheO5G1p43I,244
 careamics/utils/receptive_field.py,sha256=Y2h4c8S6glX3qcx5KHDmO17Kkuyey9voxfoXyqcAfiM,3296
 careamics/utils/serializers.py,sha256=mILUhz75IMpGKnEzcYu9hlOPG8YIiIW09fk6eZM7Y8k,1427
 careamics/utils/torch_utils.py,sha256=_Cf3HdlIRl5hxfpUg9aofCSlcW7GSsIJxsbSORXko0U,3010
-careamics-0.0.7.dist-info/METADATA,sha256=K3w_i8E8INNeZwEQdVtGHDFS_C0031Bfqzn23e8SSO4,3967
-careamics-0.0.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-careamics-0.0.7.dist-info/entry_points.txt,sha256=2fSNVXJWDJgFLATVj7MkjFNvpl53amG8tUzC3jf7G1s,53
-careamics-0.0.7.dist-info/licenses/LICENSE,sha256=6zdNW-k_xHRKYWUf9tDI_ZplUciFHyj0g16DYuZ2udw,1509
-careamics-0.0.7.dist-info/RECORD,,
+careamics-0.0.9.dist-info/METADATA,sha256=WX-n7dq4Qr5djLTSxELRii5NKHf_XqNz_vuJcyNEE7Y,3967
+careamics-0.0.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+careamics-0.0.9.dist-info/entry_points.txt,sha256=2fSNVXJWDJgFLATVj7MkjFNvpl53amG8tUzC3jf7G1s,53
+careamics-0.0.9.dist-info/licenses/LICENSE,sha256=6zdNW-k_xHRKYWUf9tDI_ZplUciFHyj0g16DYuZ2udw,1509
+careamics-0.0.9.dist-info/RECORD,,