careamics 0.1.0rc4__py3-none-any.whl → 0.1.0rc5__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of careamics might be problematic. Click here for more details.

Files changed (62) hide show
  1. careamics/careamist.py +12 -11
  2. careamics/config/__init__.py +0 -1
  3. careamics/config/architectures/unet_model.py +1 -0
  4. careamics/config/callback_model.py +1 -0
  5. careamics/config/configuration_example.py +0 -2
  6. careamics/config/configuration_factory.py +112 -42
  7. careamics/config/configuration_model.py +14 -16
  8. careamics/config/data_model.py +59 -157
  9. careamics/config/inference_model.py +19 -20
  10. careamics/config/references/algorithm_descriptions.py +1 -0
  11. careamics/config/references/references.py +1 -0
  12. careamics/config/support/supported_extraction_strategies.py +1 -0
  13. careamics/config/training_model.py +1 -0
  14. careamics/config/transformations/n2v_manipulate_model.py +1 -0
  15. careamics/config/transformations/nd_flip_model.py +6 -11
  16. careamics/config/transformations/normalize_model.py +1 -0
  17. careamics/config/transformations/transform_model.py +1 -0
  18. careamics/config/transformations/xy_random_rotate90_model.py +6 -8
  19. careamics/config/validators/validator_utils.py +1 -0
  20. careamics/conftest.py +1 -0
  21. careamics/dataset/dataset_utils/__init__.py +0 -1
  22. careamics/dataset/dataset_utils/dataset_utils.py +1 -0
  23. careamics/dataset/in_memory_dataset.py +14 -45
  24. careamics/dataset/iterable_dataset.py +13 -68
  25. careamics/dataset/patching/__init__.py +0 -7
  26. careamics/dataset/patching/patching.py +1 -0
  27. careamics/dataset/patching/sequential_patching.py +6 -6
  28. careamics/dataset/patching/tiled_patching.py +10 -6
  29. careamics/lightning_datamodule.py +20 -24
  30. careamics/lightning_module.py +1 -1
  31. careamics/lightning_prediction_datamodule.py +15 -10
  32. careamics/losses/__init__.py +0 -1
  33. careamics/losses/loss_factory.py +1 -0
  34. careamics/model_io/__init__.py +0 -1
  35. careamics/model_io/bioimage/_readme_factory.py +2 -1
  36. careamics/model_io/bioimage/bioimage_utils.py +1 -0
  37. careamics/model_io/bioimage/model_description.py +1 -0
  38. careamics/model_io/bmz_io.py +2 -1
  39. careamics/models/layers.py +1 -0
  40. careamics/models/model_factory.py +1 -0
  41. careamics/models/unet.py +91 -17
  42. careamics/prediction/stitch_prediction.py +1 -0
  43. careamics/transforms/__init__.py +2 -23
  44. careamics/transforms/compose.py +98 -0
  45. careamics/transforms/n2v_manipulate.py +18 -23
  46. careamics/transforms/nd_flip.py +38 -64
  47. careamics/transforms/normalize.py +45 -34
  48. careamics/transforms/pixel_manipulation.py +2 -2
  49. careamics/transforms/transform.py +33 -0
  50. careamics/transforms/tta.py +2 -2
  51. careamics/transforms/xy_random_rotate90.py +41 -68
  52. careamics/utils/__init__.py +0 -1
  53. careamics/utils/context.py +1 -0
  54. careamics/utils/logging.py +1 -0
  55. careamics/utils/metrics.py +1 -0
  56. careamics/utils/torch_utils.py +1 -0
  57. {careamics-0.1.0rc4.dist-info → careamics-0.1.0rc5.dist-info}/METADATA +16 -61
  58. careamics-0.1.0rc5.dist-info/RECORD +111 -0
  59. careamics/dataset/patching/patch_transform.py +0 -44
  60. careamics-0.1.0rc4.dist-info/RECORD +0 -110
  61. {careamics-0.1.0rc4.dist-info → careamics-0.1.0rc5.dist-info}/WHEEL +0 -0
  62. {careamics-0.1.0rc4.dist-info → careamics-0.1.0rc5.dist-info}/licenses/LICENSE +0 -0
@@ -1,44 +0,0 @@
1
- from typing import List, Union
2
-
3
- import albumentations as Aug
4
-
5
- from careamics.config.data_model import TRANSFORMS_UNION
6
- from careamics.transforms import get_all_transforms
7
-
8
-
9
- # TODO add some explanations on how the additional_targets is used
10
- def get_patch_transform(
11
- patch_transforms: Union[List[TRANSFORMS_UNION], Aug.Compose],
12
- with_target: bool,
13
- normalize_mask: bool = True,
14
- ) -> Aug.Compose:
15
- """Return a pixel manipulation function."""
16
- # if we passed a Compose, we just return it
17
- if isinstance(patch_transforms, Aug.Compose):
18
- return patch_transforms
19
-
20
- # empty list of transforms is a NoOp
21
- elif len(patch_transforms) == 0:
22
- return Aug.Compose(
23
- [Aug.NoOp()],
24
- additional_targets={}, # TODO this part need be checked (wrt segmentation)
25
- )
26
-
27
- # else we have a list of transforms
28
- else:
29
- # retrieve all transforms
30
- all_transforms = get_all_transforms()
31
-
32
- # instantiate all transforms
33
- transforms = [
34
- all_transforms[transform.name](**transform.model_dump())
35
- for transform in patch_transforms
36
- ]
37
-
38
- return Aug.Compose(
39
- transforms,
40
- # apply image aug to "target"
41
- additional_targets={"target": "image"}
42
- if (with_target and normalize_mask) # TODO check this
43
- else {},
44
- )
@@ -1,110 +0,0 @@
1
- careamics/__init__.py,sha256=DkMGt4t9ua0gCgvZFEtb6eydvoxG976T0KUro8KnDNA,760
2
- careamics/careamist.py,sha256=Exr8-AV0KSqZUEUdpw8dSRa_jcrq7p97qon6HqXnRS8,28409
3
- careamics/conftest.py,sha256=WR_iw4Eebfq6r_sY1No4GfB-_XG5JMkpxXpoorWy8Eg,658
4
- careamics/lightning_datamodule.py,sha256=84dsyl4rv3nhr3T6yGbOQYkNT-joWJJRDtKHoC3GThY,31915
5
- careamics/lightning_module.py,sha256=FuqX21dNIMMqXsJTThv2Xkj3XV0gOimJipvJPQN823A,10174
6
- careamics/lightning_prediction_datamodule.py,sha256=7R8Qd677uSHnIg8TWlIPibKGHuU0tbgODDuMchIZL5I,15364
7
- careamics/lightning_prediction_loop.py,sha256=EcUSgM2BHkWhjR4zN0NNf2M8xQF_I3aQ9qAFk0slYYw,4483
8
- careamics/py.typed,sha256=esB4cHc6c07uVkGtqf8at7ttEnprwRxwk8obY8Qumq4,187
9
- careamics/callbacks/__init__.py,sha256=spxJlDByD-6QtMl9vcIty8Wb0tyHaSTKTItozHenI44,204
10
- careamics/callbacks/hyperparameters_callback.py,sha256=CdOtxazXi6YXH3Fq4AahqBQ3Ai64YpAMkd8gFnAnkjc,1258
11
- careamics/callbacks/progress_bar_callback.py,sha256=ojeat3wwaqq7uEAVZBren71CZtNiEch1FR7o8FiaEPg,1868
12
- careamics/config/__init__.py,sha256=peYqbpE68rBgVp6kMRcfRkrcRhyeoNh6NEzJ-G-Ot4E,1020
13
- careamics/config/algorithm_model.py,sha256=7Y7-kr9Bc-cCOiavJ6vzahYTdlLZcLZo7FvKOPFZoWg,5732
14
- careamics/config/callback_model.py,sha256=h4q5tbjpauL5gQ1joeRvJw8Bes1s7Lp2dvcoHnFozqY,3225
15
- careamics/config/configuration_example.py,sha256=XmdMsl0Auq4G1tIm_XL9_gG-LWudVbXCFZAoy_RicqE,2605
16
- careamics/config/configuration_factory.py,sha256=4lafi6H5YLQhG-ByyONM06lDSO2OV34uhvwrDWW3eg0,18793
17
- careamics/config/configuration_model.py,sha256=dO32a2v7YXXBmqSUpUylb-_FQoHqWsXRS7zj19iwhMI,18754
18
- careamics/config/data_model.py,sha256=0Qn1PvNcVIevT2voolSULkvkPoQFeiBVu9mFUK9vE1c,16809
19
- careamics/config/inference_model.py,sha256=UI5UsE-2UUmhs5b0im9koztJkXulnXYbVFwOqt-tybY,8127
20
- careamics/config/noise_models.py,sha256=1P6dXqqBhbVHT7EZ2hluO2aHKIywkAyl4BndrvtKRKU,4338
21
- careamics/config/optimizer_models.py,sha256=Umu6O2qKv1o4ReWBuldAHmvF0u_DKS5eHCa7YdxvOaw,5266
22
- careamics/config/tile_information.py,sha256=2bVLTvZfdjPOZVq7jYXVuAFLlPFmJ2DBiwtwm5n7K-U,2939
23
- careamics/config/training_model.py,sha256=nAXZtqQsTmMIgQIPmGPpGQ_gYUatY_j-MK9B83joGAE,1558
24
- careamics/config/architectures/__init__.py,sha256=CdnViydyTdQixus3uWHBIgbgxmu9t1_ADehqpjN_57U,444
25
- careamics/config/architectures/architecture_model.py,sha256=EomNAhjuhu3uqEjzG6uvRclSQNVuGkcH_mt4R47yIHg,681
26
- careamics/config/architectures/custom_model.py,sha256=2tMvr6XqFY8hY8Tfljba-kp1c61fDKaV4vylbg4AyUg,4473
27
- careamics/config/architectures/register_model.py,sha256=sW6fvj54F6G1rg3fmXTwCAu_YcKs3xZTsdYnDGFRahY,2389
28
- careamics/config/architectures/unet_model.py,sha256=8zV9W1x9lvSCWNgtXmnPMBqt7s_-AzsUex9sjjNTe9c,2788
29
- careamics/config/architectures/vae_model.py,sha256=n1pvVmpelNqQXsXtxXPdmMBBkCLyMIlTQJbMqB-amCc,899
30
- careamics/config/references/__init__.py,sha256=rZAQzmrciX5cNICcXaBH6sbE6N6L7_qYQUkasNy9y-c,763
31
- careamics/config/references/algorithm_descriptions.py,sha256=LzY66zAtKj9705omjfhV0Ziq7vhWSZ4Tq_GPDwE2P-E,3540
32
- careamics/config/references/references.py,sha256=vcbt9qfsdzOmzqizu8COjtIXlrAyH6UT6bgoJy8D5oo,1572
33
- careamics/config/support/__init__.py,sha256=XMnPEPHVy63DDxAfczIFqFjuYrtDRrbzA5b0Ff1qmyU,1149
34
- careamics/config/support/supported_activations.py,sha256=-KaTxdm8IJ9c6xkCVMw3SWjbw6wmvV-1WuoeSbyCdH4,476
35
- careamics/config/support/supported_algorithms.py,sha256=wc3pTYhgoZ0W_O6jcJY0ZsVErYMSTmi24ptw3ezAuUw,316
36
- careamics/config/support/supported_architectures.py,sha256=EDrh8eghqxWcwjXVhhJpfymV6dsmyaUxk2E397VXXbA,495
37
- careamics/config/support/supported_data.py,sha256=AvdoWVLLpX5jzozJGUqeo5EJVkb6Z_OcQbaS8AkyHSg,2048
38
- careamics/config/support/supported_extraction_strategies.py,sha256=oVSVbFBrVMvvtHGtK5_e306ef3splJZHialoSQM_0zs,578
39
- careamics/config/support/supported_loggers.py,sha256=slbpUZSMdRMCKNAQfnCxh7n1Sql2NFoH79cIoiuavcU,159
40
- careamics/config/support/supported_losses.py,sha256=fu0SPPhubMVq3zXJk8G8Av9ojWNw62EKm7WnUmapuRs,502
41
- careamics/config/support/supported_optimizers.py,sha256=Y_Ns3DijRU9xUoY54AMm0VEeIx0ntS_4UkpyBKaSSl4,1337
42
- careamics/config/support/supported_pixel_manipulations.py,sha256=0lZfCwL9bwPK5E8qzi7l9UGOq41U2fCH9KpxnDEpznc,360
43
- careamics/config/support/supported_struct_axis.py,sha256=VzyKEVwNARtF9HviAmsHqAjyenHCRFR5UObks6s6R9Y,378
44
- careamics/config/support/supported_transforms.py,sha256=7lF2ZdWC7Fa7ExemQ6JzM9s9RXM0_tsGWilmmq-MoMw,889
45
- careamics/config/transformations/__init__.py,sha256=FnLBxNyJ0cTPf8e1dhtkhks5vu_z8Wi48aBjvAPDErk,357
46
- careamics/config/transformations/n2v_manipulate_model.py,sha256=qSY166kTIykLIfrTUBYYG8fbx3QOufzL7IXPAw85fx4,1853
47
- careamics/config/transformations/nd_flip_model.py,sha256=5ANbM4z-JDbc-Re2BtedzQFmS8Mc_iQkN-Dq2qxsK9A,856
48
- careamics/config/transformations/normalize_model.py,sha256=kIPFBM55-IePAkIQkSihz96gcbb1Yel6hCHcepCapFI,803
49
- careamics/config/transformations/transform_model.py,sha256=HqCnZLr-K9d54CxK8M0UaXdPqnPYso5Wg1i2dK5dmBI,999
50
- careamics/config/transformations/xy_random_rotate90_model.py,sha256=8sBGwMJzdLGmZrqC15ObZZNGpT9NpctzScV9LJ7zyuk,796
51
- careamics/config/validators/__init__.py,sha256=iv0nVI0W7j9DxFPwh0DjRCzM9P8oLQn4Gwi5rfuFrrI,180
52
- careamics/config/validators/validator_utils.py,sha256=81t79useG0ZfwZdodLfSl_799hBMriiFrohStRHl9nc,2623
53
- careamics/dataset/__init__.py,sha256=cUcqy1Nxa5WhDQim6948r3i1kGQ-HijUqAACyyM7cuU,174
54
- careamics/dataset/in_memory_dataset.py,sha256=_rRvDvtL_Jg4W8sx7plYArr1lBQFFFUoR-WBqmaRGwQ,12213
55
- careamics/dataset/iterable_dataset.py,sha256=xqlt2M6k3UUO2VX37Tn4L4TAniZfgNhE2tnM8_S_ENo,15035
56
- careamics/dataset/zarr_dataset.py,sha256=cpouzhaWUOkqAjjliJhrtfqVvZEMaq1d9V3LDoEvJQA,5644
57
- careamics/dataset/dataset_utils/__init__.py,sha256=nt3IzCFa8JstXbheNrA5D_wlJe0PNFpABmBA3f_Jsj0,447
58
- careamics/dataset/dataset_utils/dataset_utils.py,sha256=bKAUvLrnS-zxCpJ7H56clJ2pGB3jYYh7F20ZkG7bDIQ,2666
59
- careamics/dataset/dataset_utils/file_utils.py,sha256=VHK7__34keJfkYWfyrOApalFA0wyINLK7CB79T_Sqm4,4030
60
- careamics/dataset/dataset_utils/read_tiff.py,sha256=KcO5pZXaXEGXlJoCLnFriAHmHACCPI5kRWZK-oovfBo,1648
61
- careamics/dataset/dataset_utils/read_utils.py,sha256=hwb2pFr-ueusVf8hoTtDsD71V5HDVU-6eQ14cpPp6d4,547
62
- careamics/dataset/dataset_utils/read_zarr.py,sha256=B75gaOsLctkonYqwouCDWBu-AVwy697mJbEXDC0ft3I,1635
63
- careamics/dataset/patching/__init__.py,sha256=UJBR4X1H5B6BIcayesPGpWjxTkub1alf9Uzj_5mdnFE,130
64
- careamics/dataset/patching/patch_transform.py,sha256=kgUNUuk5QbJ8sZ3CFmBq06kO0FuclICX97Sj4PqoxyQ,1375
65
- careamics/dataset/patching/patching.py,sha256=aQbjJLa56x22WALV31K0SFvxFNtTZ-FX9EP1RiqTt0k,6258
66
- careamics/dataset/patching/random_patching.py,sha256=93R5a56O4k6CxrWvRpTwkFDnw37tpDTaajcPAQbc4Nw,6025
67
- careamics/dataset/patching/sequential_patching.py,sha256=Vm6em2ZPVUzqmrwLWJwD4jhI8hcSMjZ6YzpX7_nLUB8,5688
68
- careamics/dataset/patching/tiled_patching.py,sha256=Kq0CENa8JNINMMTGySmCdYjBPieFQKgxa14WbntPbtk,5702
69
- careamics/dataset/patching/validate_patch_dimension.py,sha256=cHiD2XNX44eArlRR7ZK7_4Mihv3DzjncthgN-nXqR4s,2001
70
- careamics/losses/__init__.py,sha256=tpiq1pGabqGm8U0AxfMM5bUkgXVCwG_8vzyVf_XeEVU,216
71
- careamics/losses/loss_factory.py,sha256=2gx5QkwLT79FlKpIj_-IZMoiH6VnzJ2KTo2Iy_cBwgI,991
72
- careamics/losses/losses.py,sha256=a-8NLw0R86r2XxcHh97D7Mnhtbn0c5VFjZLPfttKCCM,2304
73
- careamics/losses/noise_model_factory.py,sha256=zbRxlU3hTLJR9fM-PkeTthQ8idrwJD7-AelvXxaoHEQ,988
74
- careamics/losses/noise_models.py,sha256=fRr4_mxP_jirC-UkQDVF09RcwqvZJES9JVCc8tccCFc,17801
75
- careamics/model_io/__init__.py,sha256=x_wtgBZWoLWFLyrKmYFfc6rP-nLiRsBTqPBed3sT1Vo,156
76
- careamics/model_io/bmz_io.py,sha256=i2lfU4nngtBau1HAKlTFhtbmLlhfHKISBsw4MO5F6qY,6955
77
- careamics/model_io/model_io_utils.py,sha256=x5u1HszZgRfvuku4eKQhmInyxRaEzkgNqAhGSyLS6e4,2116
78
- careamics/model_io/bioimage/__init__.py,sha256=r94nu8WDAvj0Fbu4C-iJXdOhfSQXeZBvN3UKsLG0RNI,298
79
- careamics/model_io/bioimage/_readme_factory.py,sha256=cXNu2ZZdlSM7sfYo16wKSfS8CDoBsvNBj96k0HSk8vw,3498
80
- careamics/model_io/bioimage/bioimage_utils.py,sha256=YOaiaxFOvJYtrUZt9m447wgQr4nwZTcayVrqPbEopYg,1084
81
- careamics/model_io/bioimage/model_description.py,sha256=pOs3x5NgYe_I4GDzd2Wf6DBLCGQyd-rn7O89J65162I,9207
82
- careamics/models/__init__.py,sha256=Wty5hwQb_As33pQOZqY5j-DpDOdh5ArBH4BhQDSuXTQ,133
83
- careamics/models/activation.py,sha256=21g0isrf_SpY4MUQoxbcsc1hAP5luSmWgtzGOvXKzTw,936
84
- careamics/models/layers.py,sha256=3KBGhOgiZH-vz-fzvmHcFjCSUypi8ttNh2KDIWmx3Tg,11557
85
- careamics/models/model_factory.py,sha256=8jJHfY5tCkosmlfIaYy7DtMNW8wRJ0Ur2g54N1vjitY,1358
86
- careamics/models/unet.py,sha256=E5zRIRpVv7CeMcMddmHGL5UCBm_DCnVR323UXNkjZT4,10925
87
- careamics/prediction/__init__.py,sha256=-Bfc7UqPSqpGx0NGvHMkE-bHOkZYMn7EaxQ9tO6A3uU,118
88
- careamics/prediction/stitch_prediction.py,sha256=IPTTJ8FA0j_vmUOfqiiLe3JF6v5NlSOD3fBL1ZTQkq0,2215
89
- careamics/transforms/__init__.py,sha256=SRquA3XotroxiUF-AbUUkSeGVuX_BM8cauZplCwQqg4,1093
90
- careamics/transforms/n2v_manipulate.py,sha256=c16grua6w33ZRVVfW7wAOJ6EyvgPt0cXprTgBLSpKOg,4191
91
- careamics/transforms/nd_flip.py,sha256=NY0y2lrkF98XUE_twnPQ9z2xSxMvV2bYeVrJKKemjUU,2827
92
- careamics/transforms/normalize.py,sha256=khmL2xKNWHEWVx2oJWH-_JCDv9d5krFJgBQtuufaYT4,2727
93
- careamics/transforms/pixel_manipulation.py,sha256=FfTM7V2ql0vqRSFUsebBuIIPjcLo_rW8j74Lo2cUVaA,12776
94
- careamics/transforms/struct_mask_parameters.py,sha256=tcxs_MHxZTD6PZE2q5gsGJYIMDF368X9Ggw8O9SkfP0,360
95
- careamics/transforms/tta.py,sha256=4RqHwlOPpBDlHnv68XX_55ggxc2FHiepxRXIH1BfX30,1982
96
- careamics/transforms/xy_random_rotate90.py,sha256=hvXrpgK-rb4l7quRX6kXeostWr6cS-l0CicRzX2xcac,2710
97
- careamics/utils/__init__.py,sha256=vvrTq-DSqVwqrVlEDEgKygeN90ztn0aBKOcF88_t_Lo,335
98
- careamics/utils/base_enum.py,sha256=eoMYdKpcEojJWXrIxzchY1U0W_AubLijMt27WMoSXoc,753
99
- careamics/utils/context.py,sha256=S_c7PTNohilmnUVgWcKO1afVArHByKvJ2xTpAcXSUvs,1408
100
- careamics/utils/logging.py,sha256=VP6QK4GqVGZiVCOHcFwrfOY6qrvMvqP6_Wj2sZpbvfc,10321
101
- careamics/utils/metrics.py,sha256=nlZsewTQLTmmhvIHyio8lCDjXLkXSnWTEgzvqsBb64Q,2381
102
- careamics/utils/path_utils.py,sha256=mYxqHxHNKLKPJrs4w7ncyvUsoDVRRZZCVl_SJLW0tRs,518
103
- careamics/utils/ram.py,sha256=kI637_OtrOF0SoHfp_EcyszhlCwtufv1q4M-p0XUqgA,198
104
- careamics/utils/receptive_field.py,sha256=uALb2oCh4C_xmejdOU2aqyo5fOO9W2713Xb8QGQZvaw,3027
105
- careamics/utils/running_stats.py,sha256=GIPMPuH9EOUKD_cYBkJFPggXRKnQEiOXx68Pq9UCCVI,1384
106
- careamics/utils/torch_utils.py,sha256=gPZvuiWDoq8DwQZ3sddM9td8-WAlbOOm8dwspqHe27I,3013
107
- careamics-0.1.0rc4.dist-info/METADATA,sha256=oiCCO7lylOGG_qoLODCWAAx0NaYK7onLI9rltw1mPuA,4631
108
- careamics-0.1.0rc4.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
109
- careamics-0.1.0rc4.dist-info/licenses/LICENSE,sha256=6zdNW-k_xHRKYWUf9tDI_ZplUciFHyj0g16DYuZ2udw,1509
110
- careamics-0.1.0rc4.dist-info/RECORD,,