konfai 1.1.8__py3-none-any.whl → 1.2.0__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 konfai might be problematic. Click here for more details.

Files changed (36) hide show
  1. konfai/__init__.py +59 -14
  2. konfai/data/augmentation.py +457 -286
  3. konfai/data/data_manager.py +533 -316
  4. konfai/data/patching.py +300 -183
  5. konfai/data/transform.py +408 -275
  6. konfai/evaluator.py +325 -68
  7. konfai/main.py +71 -22
  8. konfai/metric/measure.py +360 -244
  9. konfai/metric/schedulers.py +24 -13
  10. konfai/models/classification/convNeXt.py +187 -81
  11. konfai/models/classification/resnet.py +272 -58
  12. konfai/models/generation/cStyleGan.py +233 -59
  13. konfai/models/generation/ddpm.py +348 -121
  14. konfai/models/generation/diffusionGan.py +757 -358
  15. konfai/models/generation/gan.py +177 -53
  16. konfai/models/generation/vae.py +140 -40
  17. konfai/models/registration/registration.py +135 -52
  18. konfai/models/representation/representation.py +57 -23
  19. konfai/models/segmentation/NestedUNet.py +339 -68
  20. konfai/models/segmentation/UNet.py +140 -30
  21. konfai/network/blocks.py +331 -187
  22. konfai/network/network.py +795 -427
  23. konfai/predictor.py +644 -238
  24. konfai/trainer.py +509 -222
  25. konfai/utils/ITK.py +191 -106
  26. konfai/utils/config.py +152 -95
  27. konfai/utils/dataset.py +326 -455
  28. konfai/utils/utils.py +497 -249
  29. {konfai-1.1.8.dist-info → konfai-1.2.0.dist-info}/METADATA +1 -3
  30. konfai-1.2.0.dist-info/RECORD +38 -0
  31. konfai/utils/registration.py +0 -199
  32. konfai-1.1.8.dist-info/RECORD +0 -39
  33. {konfai-1.1.8.dist-info → konfai-1.2.0.dist-info}/WHEEL +0 -0
  34. {konfai-1.1.8.dist-info → konfai-1.2.0.dist-info}/entry_points.txt +0 -0
  35. {konfai-1.1.8.dist-info → konfai-1.2.0.dist-info}/licenses/LICENSE +0 -0
  36. {konfai-1.1.8.dist-info → konfai-1.2.0.dist-info}/top_level.txt +0 -0
konfai/__init__.py CHANGED
@@ -1,16 +1,61 @@
1
- import os
2
1
  import datetime
2
+ import os
3
+
4
+
5
+ def models_directory() -> str:
6
+ return _get_env("KONFAI_MODELS_DIRECTORY")
7
+
8
+
9
+ def checkpoints_directory() -> str:
10
+ return _get_env("KONFAI_CHECKPOINTS_DIRECTORY")
11
+
12
+
13
+ def path_to_models() -> str:
14
+ return _get_env("KONFAI_MODEL")
15
+
16
+
17
+ def predictions_directory() -> str:
18
+ return _get_env("KONFAI_PREDICTIONS_DIRECTORY")
19
+
20
+
21
+ def evaluations_directory() -> str:
22
+ return _get_env("KONFAI_EVALUATIONS_DIRECTORY")
23
+
24
+
25
+ def statistics_directory() -> str:
26
+ return _get_env("KONFAI_STATISTICS_DIRECTORY")
27
+
28
+
29
+ def setups_directory() -> str:
30
+ return _get_env("KONFAI_SETUPS_DIRECTORY")
31
+
32
+
33
+ def config_file() -> str:
34
+ return _get_env("KONFAI_config_file")
35
+
36
+
37
+ def konfai_state() -> str:
38
+ return _get_env("KONFAI_STATE")
39
+
40
+
41
+ def konfai_root() -> str:
42
+ return _get_env("konfai_root")
43
+
44
+
45
+ def cuda_visible_devices() -> str:
46
+ return _get_env("CUDA_VISIBLE_DEVICES")
47
+
48
+
49
+ def konfai_nb_cores() -> str:
50
+ return _get_env("KONFAI_NB_CORES")
51
+
52
+
53
+ def current_date() -> str:
54
+ return datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
55
+
3
56
 
4
- MODELS_DIRECTORY = lambda : os.environ["KONFAI_MODELS_DIRECTORY"]
5
- CHECKPOINTS_DIRECTORY =lambda : os.environ["KONFAI_CHECKPOINTS_DIRECTORY"]
6
- MODEL = lambda : os.environ["KONFAI_MODEL"]
7
- PREDICTIONS_DIRECTORY =lambda : os.environ["KONFAI_PREDICTIONS_DIRECTORY"]
8
- EVALUATIONS_DIRECTORY =lambda : os.environ["KONFAI_EVALUATIONS_DIRECTORY"]
9
- STATISTICS_DIRECTORY = lambda : os.environ["KONFAI_STATISTICS_DIRECTORY"]
10
- SETUPS_DIRECTORY = lambda : os.environ["KONFAI_SETUPS_DIRECTORY"]
11
- CONFIG_FILE = lambda : os.environ["KONFAI_CONFIG_FILE"]
12
- KONFAI_STATE = lambda : os.environ["KONFAI_STATE"]
13
- KONFAI_ROOT = lambda : os.environ["KONFAI_ROOT"]
14
- CUDA_VISIBLE_DEVICES = lambda : os.environ["CUDA_VISIBLE_DEVICES"]
15
- KONFAI_NB_CORES = lambda : os.environ["KONFAI_NB_CORES"]
16
- DATE = lambda : datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
57
+ def _get_env(var: str) -> str:
58
+ value = os.environ.get(var)
59
+ if value is None:
60
+ raise RuntimeError(f"Environment variable '{var}' is not set.")
61
+ return value