careamics 0.0.2__py3-none-any.whl → 0.0.3__py3-none-any.whl

careamics/utils/metrics.py

@@ -4,7 +4,7 @@ Metrics submodule.
 This module contains various metrics and a metrics tracking class.
 """
 
-from typing import Union
+from typing import Optional, Union
 
 import numpy as np
 import torch
@@ -113,3 +113,76 @@ def scale_invariant_psnr(
     range_parameter = (np.max(gt) - np.min(gt)) / np.std(gt)
     gt_ = _zero_mean(gt) / np.std(gt)
     return psnr(_zero_mean(gt_), _fix(gt_, pred), range_parameter)
+
+
+class RunningPSNR:
+    """Compute the running PSNR during validation step in training.
+
+    This class allows to compute the PSNR on the entire validation set
+    one batch at the time.
+
+    Attributes
+    ----------
+    N : int
+        Number of elements seen so far during the epoch.
+    mse_sum : float
+        Running sum of the MSE over the N elements seen so far.
+    max : float
+        Running max value of the N target images seen so far.
+    min : float
+        Running min value of the N target images seen so far.
+    """
+
+    def __init__(self):
+        """Constructor."""
+        self.N = None
+        self.mse_sum = None
+        self.max = self.min = None
+        self.reset()
+
+    def reset(self):
+        """Reset the running PSNR computation.
+
+        Usually called at the end of each epoch.
+        """
+        self.mse_sum = 0
+        self.N = 0
+        self.max = self.min = None
+
+    def update(self, rec: torch.Tensor, tar: torch.Tensor) -> None:
+        """Update the running PSNR statistics given a new batch.
+
+        Parameters
+        ----------
+        rec : torch.Tensor
+            Reconstructed batch.
+        tar : torch.Tensor
+            Target batch.
+        """
+        ins_max = torch.max(tar).item()
+        ins_min = torch.min(tar).item()
+        if self.max is None:
+            assert self.min is None
+            self.max = ins_max
+            self.min = ins_min
+        else:
+            self.max = max(self.max, ins_max)
+            self.min = min(self.min, ins_min)
+
+        mse = (rec - tar) ** 2
+        elementwise_mse = torch.mean(mse.view(len(mse), -1), dim=1)
+        self.mse_sum += torch.nansum(elementwise_mse)
+        self.N += len(elementwise_mse) - torch.sum(torch.isnan(elementwise_mse))
+
+    def get(self) -> Optional[torch.Tensor]:
+        """Get the actual PSNR value given the running statistics.
+
+        Returns
+        -------
+        Optional[torch.Tensor]
+            PSNR value.
+        """
+        if self.N == 0 or self.N is None:
+            return None
+        rmse = torch.sqrt(self.mse_sum / self.N)
+        return 20 * torch.log10((self.max - self.min) / rmse)

{careamics-0.0.2.dist-info → careamics-0.0.3.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.3
 Name: careamics
-Version: 0.0.2
+Version: 0.0.3
 Summary: Toolbox for running N2V and friends.
 Project-URL: homepage, https://careamics.github.io/
 Project-URL: repository, https://github.com/CAREamics/careamics
-Author-email: Melisande Croft <melisande.croft@fht.org>, Joran Deschamps <joran.deschamps@fht.org>, Igor Zubarev <igor.zubarev@fht.org>
+Author-email: CAREamics team <rse@fht.org>, Ashesh <ashesh.ashesh@fht.org>, Federico Carrara <federico.carrara@fht.org>, Melisande Croft <melisande.croft@fht.org>, Joran Deschamps <joran.deschamps@fht.org>, Vera Galinova <vera.galinova@fht.org>, Igor Zubarev <igor.zubarev@fht.org>
 License: BSD-3-Clause
 License-File: LICENSE
 Classifier: Development Status :: 3 - Alpha

{careamics-0.0.2.dist-info → careamics-0.0.3.dist-info}/RECORD

@@ -1,39 +1,42 @@
 careamics/__init__.py,sha256=xBCerWN66hv3T7dRGiUYLflmbJtJt1HqbSg9JCWp8pY,391
-careamics/careamist.py,sha256=Mi6a4aUAUPH0YCQm2kObQepG5OFpjlHT0qm6GxAiq58,27496
+careamics/careamist.py,sha256=wZxIf5E2Sebbcsd1IbTi6-VrPkI71M-kI9-bFHGmh-M,27867
 careamics/conftest.py,sha256=Od4WcaaP0UP-XUMrFr_oo4e6c2hi_RvNbuaRTopwlmI,911
 careamics/py.typed,sha256=esB4cHc6c07uVkGtqf8at7ttEnprwRxwk8obY8Qumq4,187
-careamics/config/__init__.py,sha256=qaR98bVRFnEHYGG5EgGBIa9P9AtMRmKzpXuc1exdteo,913
-careamics/config/algorithm_model.py,sha256=-Nx4E6M2EyBDozSTkXeYaj9b0KcIli0Gy6DE3P1WVfE,5443
+careamics/config/__init__.py,sha256=M_Y0sGIVr1lQF4IpDeVkWkspbUNlbdH-iDjapMSGOoc,1127
 careamics/config/callback_model.py,sha256=EeYHqpMIPQwyNxLRzzX32Uncl5mZuB1bJO76RHpNymg,4555
-careamics/config/configuration_factory.py,sha256=Ckkv4ber2WSaa-BaoDNuyvpVMd02Q-9T27yziXJ-1Ak,18538
-careamics/config/configuration_model.py,sha256=4LbAt3zUtx05mfTRrXBqD57iQn2s0Y93f81M8Gce4zo,18698
-careamics/config/data_model.py,sha256=dpRthXU8lINT3laJygqka-fmUDT7U34QUHZYBX95oY8,15070
+careamics/config/configuration_factory.py,sha256=TcB15HD81P-e-A2w8EfCdJLoYw3537tQ6NAAwKrSU6g,18799
+careamics/config/configuration_model.py,sha256=efUHSlYonO_m6eUPQLRb4NlsTADCL5lGCrwXlOGIHFU,18893
+careamics/config/data_model.py,sha256=zcVx8wN8IW-hMLXZ2jB7AWSQ1C2gSt87UtkoEtsClJA,15776
+careamics/config/fcn_algorithm_model.py,sha256=rwABUX1XwMMenv5Ff4z-ZTneFIhW_qIjMAbb57RV150,5082
 careamics/config/inference_model.py,sha256=UE_-ZmCX6LFCbDBOwyGnvuAboF_JNX2m2LcF0WiwgCI,6961
+careamics/config/likelihood_model.py,sha256=f8EmLi5MtOWXO_ZBrjSa27b7vThy579pY-cecstJQEw,1413
+careamics/config/nm_model.py,sha256=g1WR7uBtOcZuhfTWaO1Ma0-5Pu4YmRLchVj9eU1_GoY,3897
 careamics/config/optimizer_models.py,sha256=p6gDYtO-jFtL7zVX0-Id-rGJWkkyhbU3EBrWD_4TxZE,5726
 careamics/config/tile_information.py,sha256=c-_xrVPOgcnjiEzQ-9A_GhNPamObkMANbeHaRP29R-4,2059
 careamics/config/training_model.py,sha256=9dfCcEcdPsGDW1Q0OIR_PmKjkWbZjeOmGIKuIyRd-0E,1875
-careamics/config/architectures/__init__.py,sha256=CdnViydyTdQixus3uWHBIgbgxmu9t1_ADehqpjN_57U,444
-careamics/config/architectures/architecture_model.py,sha256=4WvQQJGz5DLFjOUryZx0fqPuEMlF2RhtlV5XudJTIbc,922
-careamics/config/architectures/custom_model.py,sha256=IRaequRi5BXMPL14gLr1B_27_XWPeWxQwgHF9_EJMaU,4664
+careamics/config/vae_algorithm_model.py,sha256=TkGyOPf_-vwfEFVRmst0q9Cnyz_Xa_U68Sy7VVKw4zE,6123
+careamics/config/architectures/__init__.py,sha256=VyuBIopbqpR0KPq6cAeX3qPCrlbY8ZB_rGXwa2ApNvc,447
+careamics/config/architectures/architecture_model.py,sha256=qqPpmkNLwTBC3nwOmXpl33UAvpDFsAxFRBw9MIsQgws,921
+careamics/config/architectures/custom_model.py,sha256=-ROcKn_Ai7SZqzQLoEUc2BLAalgjyvoOnyDKUiY_cp0,4876
+careamics/config/architectures/lvae_model.py,sha256=JqcRYH6809GsiGb1Mta_rx-Uc9MmWjnS7IIeplg_UFk,4710
 careamics/config/architectures/register_model.py,sha256=lHH0aUPmXtI3Bq_76zkhg07_Yb_nOJZkZJLCC_G-rZM,2434
 careamics/config/architectures/unet_model.py,sha256=8F2KosNkrXUP2bxlm-D1mowS9x3GOjyXjsEo1Kf-05k,3497
-careamics/config/architectures/vae_model.py,sha256=iLPwjI4B_Ivv_qQNUJc4-Gwm4z8UA3P5BsKQucRFEMI,962
 careamics/config/references/__init__.py,sha256=rZAQzmrciX5cNICcXaBH6sbE6N6L7_qYQUkasNy9y-c,763
 careamics/config/references/algorithm_descriptions.py,sha256=wR3hIoeg5eiUEPbwTxMpQYLTKQyRl_5naSDbBZOZESU,3541
 careamics/config/references/references.py,sha256=AXx08FJQxHb7SYOluCr_eQn_mbOris5dXqhKrCnhBTE,1573
 careamics/config/support/__init__.py,sha256=pKqk76kyBraiSC1SQos-cyiQwsfOLLkLuWj6Hw60LZ4,1041
-careamics/config/support/supported_activations.py,sha256=O27_dGDgw2P-DslKJsXGVAyS2NUQM6Ta4jeo2uTQlW0,519
-careamics/config/support/supported_algorithms.py,sha256=GCkauFDlmb2hJwFSdoIpGmpLjPeYFHOGy2NweKdw8T4,358
-careamics/config/support/supported_architectures.py,sha256=LLD6hyje9Q0BcvA7p2E8WW_cY5yEgMI_NAP4HBi27UU,540
+careamics/config/support/supported_activations.py,sha256=CqOWoziIK5jZZXJO7G7cGg3TTid1POqv8FXqxjXxyME,535
+careamics/config/support/supported_algorithms.py,sha256=Tt5PaV1mwSZdrc4XpW9M4-ofExoHnMgfebCSN6ns0AQ,953
+careamics/config/support/supported_architectures.py,sha256=f93cRyeBOu5rYzvYrvWPVpjyK54GhyVTzCT1GNSdhI8,494
 careamics/config/support/supported_data.py,sha256=T_mDiWLFMVji_EpjBABUObAJcnv-XBnqp9XUZP37Tdk,2902
 careamics/config/support/supported_loggers.py,sha256=ubSOkGoYabGbm_jmyc1R3eFcvcP-sHmuyiBi_d3_wLg,197
-careamics/config/support/supported_losses.py,sha256=TPsMCuDdgb64TRyDwonnwHb1R-rkn3OzhtHimyVtrOY,540
+careamics/config/support/supported_losses.py,sha256=2x5sZuxRbWJzodoL35I1mMYUUDMzk8UFiFdbyPwbJ4E,583
 careamics/config/support/supported_optimizers.py,sha256=xxbJsyohJTlHeUz2I4eRwcE3BeACs-6PH8cpX6w2wX8,1394
 careamics/config/support/supported_pixel_manipulations.py,sha256=rFiktUlvoFU7s1NAKEMqsXOzLw5eaw9GtCKUznvq6xc,432
 careamics/config/support/supported_struct_axis.py,sha256=alZMA5Y-BpDymLPUEd1zqVY0xMkgl9Rv1d4ujED6sco,424
 careamics/config/support/supported_transforms.py,sha256=4uob-bnZ5aqpN5aEI67-aa7bsmVCrKxEknzf2BAZ3W4,283
 careamics/config/transformations/__init__.py,sha256=oqwBAL2XXbPRZZ5iOzNqalX6SyJ1M-S0lkfbDGZOzyE,378
-careamics/config/transformations/n2v_manipulate_model.py,sha256=UTyfpm1mmMvYg_HoMzXilZhJGx_muiV-lLQ4UThCFJ0,1854
+careamics/config/transformations/n2v_manipulate_model.py,sha256=Mdxc4J3vxe_dM2CIhmTwwGOIirQvrQXLoa2vRsTzoYI,1855
 careamics/config/transformations/normalize_model.py,sha256=1Rkk6IkF-7ytGU6HSzP-TpOi4RRWiQJ6fOd8zammXcg,1936
 careamics/config/transformations/transform_model.py,sha256=i7KAtSv4nah2H7uyJFKqg7RdKF68OHIPMNNvDo0HxGY,1000
 careamics/config/transformations/xy_flip_model.py,sha256=zU-uZ1b1zNZWckbho3onN-B7BHKhN7jbgbNZyRQhv2s,1025
@@ -60,65 +63,76 @@ careamics/dataset/patching/sequential_patching.py,sha256=_l3Q2uYIhjMJMaxDdSbHC9_
 careamics/dataset/patching/validate_patch_dimension.py,sha256=sQQ0-4b4uu60MNKkoWv95KxQ80J7Ku0CEk0-kAXlKeI,2134
 careamics/dataset/tiling/__init__.py,sha256=XynyAz85hVfkLtrG0lrMr_aBQm_YEwfu5uFcXMGHlOA,190
 careamics/dataset/tiling/collate_tiles.py,sha256=OrPZ-n-V3uGOc_7CcPnyEJqdbEVDlTfJfWmZnyBZ-HA,978
+careamics/dataset/tiling/lvae_tiled_patching.py,sha256=MIdwy_tNOlkErHbiykHJNLTdDEc3sdjKX5DnMvgDILo,10157
 careamics/dataset/tiling/tiled_patching.py,sha256=ouxUWvttzmTY310CuiR25IGNVWgksVQAXoN4IXSi_G0,5942
 careamics/file_io/__init__.py,sha256=vdIx5JV3JwoyOeWnY_0tY6aekwPFy_8hBqe0Yj-aOH8,334
 careamics/file_io/read/__init__.py,sha256=I2Ios3fOoe_7f1nYT88qt2hcl0107aJCvA8yPfdpVIA,259
 careamics/file_io/read/get_func.py,sha256=yGXD0rTFD7u70FR0axrQtWies0aYW3iQ6f0Wfcd8z-8,1394
-careamics/file_io/read/tiff.py,sha256=_WVqUycI4NMk2GzDBEOWcGuSr1293673A1vs7WvZbS4,1358
+careamics/file_io/read/tiff.py,sha256=UMofW33rvByK9B1zYGhSrWAiAA3uQUV3OVK7cq9d0gQ,1359
 careamics/file_io/read/zarr.py,sha256=2jzREAnJDQSv0qmsL-v00BxmiZ_sp0ijq667LZSQ_hY,1685
 careamics/file_io/write/__init__.py,sha256=syy-e55OKPqa2Fn7G6szJrAmmJ4JUJyr4Y4ZSnKK0zg,283
 careamics/file_io/write/get_func.py,sha256=hyGHe1RX-lfa9QFAnwRCz_gS0NRiRnXEtg4Bdeh2Esc,1627
 careamics/file_io/write/tiff.py,sha256=tBGIgl-I1sMyBivgx-dOTBykXBODkgwPH8MT3_4KAE8,1050
-careamics/lightning/__init__.py,sha256=IAhfuveylgTdwIqynRqmGNOGrBlNmsXNgK3TuPDqU-o,572
-careamics/lightning/lightning_module.py,sha256=n8aQmHmjnztUTm5eob_MwcDw4W6pHMiUra1WNsvcVug,8951
-careamics/lightning/predict_data_module.py,sha256=rgHhS5fKoa5wscWyvmPP_FHHdVnDnLbF4sG7y_C7ZOM,12747
-careamics/lightning/train_data_module.py,sha256=XVLeFTfB5IQFdh9LnuvDNOMqqloRMH-miQxk8cVybjU,27897
+careamics/lightning/__init__.py,sha256=iS9dYpYrkjyeZfodcohc25rBTPzxj9l50iTb5Jv0j0o,588
+careamics/lightning/lightning_module.py,sha256=daUVIJei8bICfmZvshFBz0-6gS4m4MA2Xx9vVTbj0xI,21522
+careamics/lightning/predict_data_module.py,sha256=JNwujK6QwObSx6P25ghpGl2f2gGT3KVgYMTlonZzH20,12745
+careamics/lightning/train_data_module.py,sha256=rwhui0Oh2EyBMQsXADOrI-IXPTG_YePpM0KcDw15H8s,27895
 careamics/lightning/callbacks/__init__.py,sha256=neTWqF6EbBRVf0FqtBkPHeGIR3j1yTk0OASraEVN0Pg,312
-careamics/lightning/callbacks/hyperparameters_callback.py,sha256=ODJpwwdgc1-Py8yEUpXLar8_IOAcfR7lF3--6LfSiGc,1496
+careamics/lightning/callbacks/hyperparameters_callback.py,sha256=u45knOZHwoVHz6yYfrnERQuozT_SfZ1OrKP0QjeU4EM,1495
 careamics/lightning/callbacks/progress_bar_callback.py,sha256=8HvNSWZldixd6pjz0dLDo0apIbzTovv5smKmZ6tZQ8U,2444
 careamics/lightning/callbacks/prediction_writer_callback/__init__.py,sha256=tATV4kFZ8h4ZpN866URsf2vbEfY9HlHN4VALJcDySCY,548
 careamics/lightning/callbacks/prediction_writer_callback/file_path_utils.py,sha256=i4vGGiVLslafi-5iuvkAKzBgZ0BpwTTxSTo31oViFz4,1480
 careamics/lightning/callbacks/prediction_writer_callback/prediction_writer_callback.py,sha256=HGDyYLSak9puGEs42cI_08peAuOMaFyZaRkcpv_n0iY,8183
 careamics/lightning/callbacks/prediction_writer_callback/write_strategy.py,sha256=aqc6DDewD4n3tbLfqS4y-FfNKV9zy3-CZ0K2Mu16Mms,12567
-careamics/lightning/callbacks/prediction_writer_callback/write_strategy_factory.py,sha256=yki7H3_11nGOpCmjVqI5KY_kItHkQj3JQhLsnledT2A,7115
-careamics/losses/__init__.py,sha256=kVEwfZ2xXfd8x0n-VHGKm6qvzbto5pIIJYP_jN-bCtw,89
-careamics/losses/loss_factory.py,sha256=vaMlxH5oescWTKlK1adWwbeD9tW4Ti-p7qKmc1iHCi0,1005
-careamics/losses/losses.py,sha256=DKwHZ9ifVe6wMd3tBOiswLC-saU1bj1RCcXGOkREmKU,2328
+careamics/lightning/callbacks/prediction_writer_callback/write_strategy_factory.py,sha256=F1IpbNNgkv5eK8Xpqp7wqv2lsqEdP1wMRlBL7RBn93U,7114
+careamics/losses/__init__.py,sha256=4s73OlMCGJdNCVfOteJPyaPwflgIiu_fj3zpKLx7j6o,351
+careamics/losses/loss_factory.py,sha256=L68kC1Q13Yacr1PJ9IL6dhGjAsZSfhpfKeaptwLUpgE,4379
+careamics/losses/fcn/__init__.py,sha256=kf92MKFGHr6upiztZVgWwtGPf734DZyub92Rn8uEq8o,18
+careamics/losses/fcn/losses.py,sha256=NdOz29hzJ7D26p13q-g0NWoYwNauIWrP2xWww6YPbB8,2360
+careamics/losses/lvae/__init__.py,sha256=0FNtMLHrOMfagtWkaBdz1NTjyf2y0QLgysxJv5jq5uw,19
+careamics/losses/lvae/loss_utils.py,sha256=QxzA2N1TglR4H0X0uyTWWytDagE1lA9IB_TK1lms3ao,2720
+careamics/losses/lvae/losses.py,sha256=CotmiPlBBxWZwZ4reLG4SZBUKGu_jRfPukA4u1sn5Xw,16100
 careamics/lvae_training/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-careamics/lvae_training/data_modules.py,sha256=A5Uoo4qtPdX99QSi-Zl22LzO0I1DszJbQuXMGUXGQEE,46665
-careamics/lvae_training/data_utils.py,sha256=tRk0k0TkBLPocqlUlkwQN_dm5jzw5z74YNs2DsCuy9Y,21670
 careamics/lvae_training/eval_utils.py,sha256=_AlXNXk4uGS2AGsF4PHJZpJoWBgq32kvQLEh7awOIvc,32405
-careamics/lvae_training/get_config.py,sha256=-CWVxlPo71_huUSmXnmYvOmgvcvrZiv0wIpXnR32l6E,3054
+careamics/lvae_training/get_config.py,sha256=1PeqVo0iMDeKfGftUEypEBKGed9iepxgQmLkZQrWpyg,3062
 careamics/lvae_training/lightning_module.py,sha256=ryr7iHqCMzCl5esi6_gEcnKFDQkMrw0EXK9Zfgv1Nek,27186
 careamics/lvae_training/metrics.py,sha256=KTDAKhe3vh-YxzGibjtkIG2nnUyujbnwqX4xGwaRXwE,6718
-careamics/lvae_training/train_lvae.py,sha256=Eu--3-RHSfhQVsJ-CTDXhUeoM1fzf_H9IGtBaNPOsHI,11044
+careamics/lvae_training/train_lvae.py,sha256=uuLhBSbekLClg-I5hp7XTTlQhXO_j8hliIlpRvzKOI0,11074
 careamics/lvae_training/train_utils.py,sha256=e-d4QsF-li8MmAPkAmB1daHpkuU16nBTnQFZYqpTjn4,3567
+careamics/lvae_training/dataset/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+careamics/lvae_training/dataset/data_utils.py,sha256=LNKpPEnk97oELj1F9Yo8fO4zXGrJXBRkENYL50kAB8M,27117
+careamics/lvae_training/dataset/lc_dataset.py,sha256=EvCM2YeiXeSn5GXvqTvZPRriYzKf5Hz3JVmy5a6w44k,10460
+careamics/lvae_training/dataset/lc_dataset_config.py,sha256=vgsjNZaotH0MqrauIUZcwPQd9Fpfj4ZVASnhTTB2OKE,398
+careamics/lvae_training/dataset/vae_data_config.py,sha256=EzGrjqGNGrlBbYWEoIwgdoL7MNguLGVKn89yp1WIfv8,5981
+careamics/lvae_training/dataset/vae_dataset.py,sha256=wYT0s0LuECDQCXCrN91p2vUyo-tzagIFJyRD8FPTXa0,39947
 careamics/model_io/__init__.py,sha256=HITzjiuZQwo-rQ2_Ma3bz9l7PDANv1_S489E-tffV9s,155
-careamics/model_io/bmz_io.py,sha256=eKyqt-z6Y3Fv0F4EZGzeE1hUzxh9rQNcqkYmYopDoF4,7128
-careamics/model_io/model_io_utils.py,sha256=EebZL3t6oIHY0kuTKacmAEriTQ4B77KuAQ84UHG7XW4,2357
+careamics/model_io/bmz_io.py,sha256=YugFDqEHXqGnXrOkEBp_6utWUkTxJSNXUce7B5iCKOo,7575
+careamics/model_io/model_io_utils.py,sha256=Dc_0YmcUfS3HuEVgN7KFfbDH8SGywT_eUCIJDVpEUHc,2755
 careamics/model_io/bioimage/__init__.py,sha256=r94nu8WDAvj0Fbu4C-iJXdOhfSQXeZBvN3UKsLG0RNI,298
 careamics/model_io/bioimage/_readme_factory.py,sha256=LZAuEiWNBTPaD8KrLPMq16yJuOPKDZiGQuTMHKLvoT4,3514
-careamics/model_io/bioimage/bioimage_utils.py,sha256=xsrzTy40LVkMmLh2qbjvz7rcs6GnQWMMlCYgwcIlb1E,1157
-careamics/model_io/bioimage/model_description.py,sha256=3jw4wkJDefLEW-2BbEfAml3AwyteZszL-v8JYpJRcOo,9635
-careamics/models/__init__.py,sha256=Wty5hwQb_As33pQOZqY5j-DpDOdh5ArBH4BhQDSuXTQ,133
-careamics/models/activation.py,sha256=xdqz4-yKV7oElG_dDrYuibS8HOiYvKdV_r9FwWPvaDE,977
+careamics/model_io/bioimage/bioimage_utils.py,sha256=Md6y0-85q772A_G_lyhR8WKk17QTl1-0fqsEh5tAGyk,1157
+careamics/model_io/bioimage/model_description.py,sha256=au8NtLwYVtjRI0BW2hEAlDen5ZGt25KAcy7_ZnAoQO4,9640
+careamics/models/__init__.py,sha256=Xui2BLJd1I2r_E3Sj24fJALFTi2FGtfNscUWj_0c9Hk,93
+careamics/models/activation.py,sha256=nu3sDgd7Lsyw8rvmUwxNN-7SM09cEMfxZ9DRDzdSKns,1049
 careamics/models/layers.py,sha256=oWzpq8OdHFEJqPWC9X8IRPNe0XqAnesSqwoT6V3t1Mw,13712
-careamics/models/model_factory.py,sha256=5YRwRRUemxb-pTRL3VWn8N61tCGyhrurqPgcFaNETb0,1360
+careamics/models/model_factory.py,sha256=hqhV8sDq1JBLKt_7Vrw4wJSugKmBt2FIWyO7ePnJSTo,1759
 careamics/models/unet.py,sha256=3pXpiCIw7WUaDV0Jmczkxi99C5-Zu3NpQpWxgRkeGL8,14321
-careamics/models/lvae/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-careamics/models/lvae/layers.py,sha256=wFuQgmtJtB7YNuNi2dVoOEWq1ndR6ku4iGvC2u0TJlM,84991
-careamics/models/lvae/likelihoods.py,sha256=FRFTh34FaBLGxn9OXFzqFyHhhJMSKYhgqxwG65VbGh8,10489
-careamics/models/lvae/lvae.py,sha256=5RlK4-h55dGz9UMCh8JCbLsaaIQ5S2IKGeI9d4nD5dA,40167
-careamics/models/lvae/noise_models.py,sha256=yotY5gkPAowbI7esOmHlzBWcSsZlH2G3U7uYIWghGwY,15703
-careamics/models/lvae/utils.py,sha256=muy4nLHmnB3BPAI0tQbJK_vVtBZOLBvhrJigHIOx5V4,11542
+careamics/models/lvae/__init__.py,sha256=6dT6uqgT__V08EjoTGxXguTbTkySZmByS9J2Bj6WWLM,53
+careamics/models/lvae/layers.py,sha256=zRHlIHaFp4nMZwTKdzhKNKGPdarefaITTqNWO8YrkMM,85022
+careamics/models/lvae/likelihoods.py,sha256=aMzmofJCzrRPGKNhIxO_mZQGoSuXxeu3OKt-BKTWfdo,11625
+careamics/models/lvae/lvae.py,sha256=MVj8rMFMuY896H0UYR4f85qcmdI5sU01fmilGbazBP8,36391
+careamics/models/lvae/noise_models.py,sha256=ITapvSDjqfO91uZxrDDfnrbKpgS0yAk_Yo2ESoZDvb0,19707
+careamics/models/lvae/utils.py,sha256=4b6VKB8brg3yjv2mMb6I2KnT6HxnD5ZrfGcxGYKauHQ,11542
 careamics/prediction_utils/__init__.py,sha256=uYKzirlF-unFL9GbDPxFnYgOwSjGAtik9fonU7DfuEY,270
+careamics/prediction_utils/lvae_prediction.py,sha256=ZwPFCSeUGsULIMoMQWRYKHfLFaDm7UKyGaUMVfSUqfs,6210
+careamics/prediction_utils/lvae_tiling_manager.py,sha256=SI-JaJvLrKWBSHdm-FjcqWdbhlcflTRiKxYF7CSGzvA,13736
 careamics/prediction_utils/prediction_outputs.py,sha256=p3Nbw9wRLU_M5uixPbzj_DmfxcSL7Y8rAQ_aTx70KQI,4082
-careamics/prediction_utils/stitch_prediction.py,sha256=aA4rktZBgk776Iw10ABXmusuvm2vqvMwoeEPCCC0nw4,3150
+careamics/prediction_utils/stitch_prediction.py,sha256=HlfkJDirzbmil6Db-kFzlmhUREmGEZ2Ag8g-URbdIb0,3899
 careamics/transforms/__init__.py,sha256=VIHIsC8sMAh1TCm67ifB816Zp-LRo6rAONPuT2Qs3bs,483
 careamics/transforms/compose.py,sha256=dUHH-_ryZtnHuDZrbypf9B1oFsQv0_nYwS2ZbpGiaXw,2959
 careamics/transforms/n2v_manipulate.py,sha256=Gty7Jtu-RiFb1EnlrOi652qAOGKU5ZHvidRvykWqJxg,5438
 careamics/transforms/normalize.py,sha256=dfGWCGPyNwyEqg5wUCAA8cGdT1MvNkpKUEpw8Cw8DfA,7274
-careamics/transforms/pixel_manipulation.py,sha256=lNA19Vlo_3GHzRnT_4AFuv6eWQaxbie2PTYGalCY4YQ,13346
+careamics/transforms/pixel_manipulation.py,sha256=sMR3A1GrgL5AactzZYzvvtz7L13sA34ckVEoNTBmDqM,13362
 careamics/transforms/struct_mask_parameters.py,sha256=jE29Li9sx3olaRnqYfJsSlKi2t0WQzJmCm9aCbIQEsA,421
 careamics/transforms/transform.py,sha256=cEqc4ci8na70i-HIGYC7udRfVa8D_8OjdRVrr3txLvQ,464
 careamics/transforms/tta.py,sha256=78S7Df9rLHmEVSQSI1qDcRrRJGauyG3oaIrXkckCkmw,2335
@@ -129,12 +143,12 @@ careamics/utils/autocorrelation.py,sha256=M_WYzrEOQngc5iSXWar4S3-EOnK6DfYHPC2vVM
 careamics/utils/base_enum.py,sha256=bz1D8mDx5V5hdnJ3WAzJXWHJTbgwAky5FprUt9F5cMA,1387
 careamics/utils/context.py,sha256=Ljf70OR1FcYpsVpxb5Sr2fzmPVIZgDS1uZob_3BcELg,1409
 careamics/utils/logging.py,sha256=coIscjkDYpqcsGnsONuYOdIYd6_gHxdnYIZ-e9Y2Ybg,10322
-careamics/utils/metrics.py,sha256=9YQe5Aj2Pv2h9jnRFeRbDQ_3qXAW0QHpucSqiUtwDcA,2382
+careamics/utils/metrics.py,sha256=qx4QhOW9ACOgCjTUpYv631RrZ5YsdR0QQA1BY0YoRM8,4532
 careamics/utils/path_utils.py,sha256=8AugiG5DOmzgSnTCJI8vypXaPE0XhnR-9pzeiFUZ-0I,554
 careamics/utils/ram.py,sha256=tksyn8dVX_iJXmrDZDGub32hFZWIaNxnMheO5G1p43I,244
 careamics/utils/receptive_field.py,sha256=Y2h4c8S6glX3qcx5KHDmO17Kkuyey9voxfoXyqcAfiM,3296
 careamics/utils/torch_utils.py,sha256=g1zxdlM7_BA7mMLcCzmrxZX4LmH__KXlJibC95muVaA,3014
-careamics-0.0.2.dist-info/METADATA,sha256=Ss5EYkycsQMCKUrO2DxtmKWaCzsCU2RdSTWHZPOvAZg,3480
-careamics-0.0.2.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-careamics-0.0.2.dist-info/licenses/LICENSE,sha256=6zdNW-k_xHRKYWUf9tDI_ZplUciFHyj0g16DYuZ2udw,1509
-careamics-0.0.2.dist-info/RECORD,,
+careamics-0.0.3.dist-info/METADATA,sha256=MjZrNQfB-fS45U5SMjx2S6jOaFIbKPYSOBI30ZtqPmw,3626
+careamics-0.0.3.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+careamics-0.0.3.dist-info/licenses/LICENSE,sha256=6zdNW-k_xHRKYWUf9tDI_ZplUciFHyj0g16DYuZ2udw,1509
+careamics-0.0.3.dist-info/RECORD,,

careamics/config/architectures/vae_model.py

@@ -1,42 +0,0 @@
-"""VAE Pydantic model."""
-
-from typing import Literal
-
-from pydantic import (
-    ConfigDict,
-)
-
-from .architecture_model import ArchitectureModel
-
-
-class VAEModel(ArchitectureModel):
-    """VAE model placeholder."""
-
-    model_config = ConfigDict(
-        use_enum_values=True, protected_namespaces=(), validate_assignment=True
-    )
-
-    architecture: Literal["VAE"]
-    """Name of the architecture."""
-
-    def set_3D(self, is_3D: bool) -> None:
-        """
-        Set 3D model by setting the `conv_dims` parameters.
-
-        Parameters
-        ----------
-        is_3D : bool
-            Whether the algorithm is 3D or not.
-        """
-        raise NotImplementedError("VAE is not implemented yet.")
-
-    def is_3D(self) -> bool:
-        """
-        Return whether the model is 3D or not.
-
-        Returns
-        -------
-        bool
-            Whether the model is 3D or not.
-        """
-        raise NotImplementedError("VAE is not implemented yet.")