konfai 1.1.8__py3-none-any.whl → 1.2.0__py3-none-any.whl

{konfai-1.1.8.dist-info → konfai-1.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: konfai
-Version: 1.1.8
+Version: 1.2.0
 Summary: Modular and configurable Deep Learning framework with YAML and PyTorch
 Author-email: Valentin Boussot <boussot.v@gmail.com>
 License-Expression: Apache-2.0
@@ -27,8 +27,6 @@ Provides-Extra: lpips
 Requires-Dist: lpips; extra == "lpips"
 Provides-Extra: cluster
 Requires-Dist: submitit; extra == "cluster"
-Provides-Extra: plot
-Requires-Dist: matplotlib; extra == "plot"
 Dynamic: license-file
 
 [![License](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://github.com/vboussot/KonfAI/blob/main/LICENSE)

konfai-1.2.0.dist-info/RECORD

@@ -0,0 +1,38 @@
+konfai/__init__.py,sha256=qjE9Rqxo1sMrkqGS8I5xlGQMZnjIfU-CGgSI5Wmbmbs,1231
+konfai/evaluator.py,sha256=xAKWUDvdSxqYRUsKqH6ieQF06LWa785aE4zLv4I3_i4,17850
+konfai/main.py,sha256=Fc4HcJEhPmgunj_f-QYyvQNvjHrKHSUv27Okgu6V5_A,3842
+konfai/predictor.py,sha256=a8QbZPMKrf8bFz1XKZs7j6HT0RXC09HdiIQbvUBqKBM,34597
+konfai/trainer.py,sha256=4mc-8r-FxtX_EAn2su8qd-BLLQ0D0So8hh5rmuf6Hqs,27163
+konfai/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+konfai/data/augmentation.py,sha256=vcJE7mosvUkwbpbTN_lGP0S1uJrJYGjlLrt9VnDdJYY,27792
+konfai/data/data_manager.py,sha256=7-ruYS2HJMvLzTq5425loOf2OCpAm5MspdYoO74ICsw,31180
+konfai/data/patching.py,sha256=jS35OxnJagKNUnJu7TzuGZpVj9fP-6H4nc2OEYOGgt8,16494
+konfai/data/transform.py,sha256=YCldsqTTBFFCqc_VdvyuNVs2kmV56CxQBN5XhEoPxho,27745
+konfai/metric/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+konfai/metric/measure.py,sha256=0mOIZKTa2u0UECpoDSbdJUhttAw_e1BlsROQQpi1oBk,27804
+konfai/metric/schedulers.py,sha256=TpYMA24FMpxRnqfhMGb0i_Mm-bzT9kySbBgvkYk-6wM,1327
+konfai/models/classification/convNeXt.py,sha256=Ha9QYd1-JEYUwL7zgNNAWeuJLbUT7LCNUkLcsdySAHM,10060
+konfai/models/classification/resnet.py,sha256=4-l7HtpP_OqApDT8XjTH94fXIuiSzz833SUWXP2oFJo,10813
+konfai/models/generation/cStyleGan.py,sha256=jnpJQ6mtQS5n7CGJ0_mdvnM5lluuyMt1TzC7lzjU1A0,10557
+konfai/models/generation/ddpm.py,sha256=S2xceH_VHBuzrCwWMkCVW6Pt9sk_WxXg66I3KS3DxMU,16659
+konfai/models/generation/diffusionGan.py,sha256=wrp_gDQG-DWW0faK2MclISt7KYwBTHw7RGrD4X2hyRc,33399
+konfai/models/generation/gan.py,sha256=oYTrbyVkEaO91kIl-bnsJRraXbGur9eIQTbUvU7JmoE,9457
+konfai/models/generation/vae.py,sha256=riZwhkS9qSDH9h6ELa2kiLL3yh-ihoeBhfFPeCcssqQ,6255
+konfai/models/registration/registration.py,sha256=IlaeZgrO4Cqur-yzp5SN_Lkc7CroxRfXBXWk-FQ5zek,7147
+konfai/models/representation/representation.py,sha256=79ccCmpSTTaQzrEucN3FhK3qzzo86TGQ1jnNbvccn7I,2953
+konfai/models/segmentation/NestedUNet.py,sha256=W4uauwF0HY8ybi49iYiTlKLdJEyD7SaC967JHzHf6ZA,14885
+konfai/models/segmentation/UNet.py,sha256=Pu_LiQdO4Mrzyn0HRE6rwxUjHGH4OG-JpzWB_U1K46g,5602
+konfai/network/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+konfai/network/blocks.py,sha256=l70_oOcz5Hmol2xmxruG0kke_2SVgO3rXYXVTMSdAS8,15645
+konfai/network/network.py,sha256=PWI6W4sz7G5Pbb-79l7mL61AoyHWlQytRtBzvwh3Ro0,54800
+konfai/utils/ITK.py,sha256=HVed4Z96X1jTaWrrQNdoBMqOtVK9InAPlDBJu-5uv3g,15476
+konfai/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+konfai/utils/config.py,sha256=a7t44CYMUT5oCDdjL94IswhCVfFbQ5FCgDWZktDDkc4,14347
+konfai/utils/dataset.py,sha256=Au22fcADKyDJMfS8Z9q8kEXLtKkoufJsH7Pwly6pALo,28288
+konfai/utils/utils.py,sha256=jCj3tZ8agQYceSY_tlVYp88UFPE5oUn6tXrqnZGrKiI,28410
+konfai-1.2.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+konfai-1.2.0.dist-info/METADATA,sha256=0i5f81WQyblGv3hF9L4l-skjcPmx1NK7NwtPmcdAQIY,2451
+konfai-1.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+konfai-1.2.0.dist-info/entry_points.txt,sha256=fG82HRN5-g39ACSOCtij_I3N6EHxfYnMR0D7TI_8pW8,81
+konfai-1.2.0.dist-info/top_level.txt,sha256=xF470dkIlFoFqTZEOlRehKJr4WU_8OKGXrJqYm9vWKs,7
+konfai-1.2.0.dist-info/RECORD,,

konfai/utils/registration.py

@@ -1,199 +0,0 @@
-import SimpleITK as sitk
-from typing import Union
-import numpy as np
-import sys
-import scipy
-
-def parameterMap_to_transform(path_src: str) -> Union[sitk.Transform, list[sitk.Transform]]:
-    transform = sitk.ReadParameterFile(path_src)
-    format = lambda x: [float(i) for i in x]
-    dimension = int(transform["FixedImageDimension"][0])
-
-    if transform["Transform"][0] == "EulerTransform":
-        if dimension == 2:
-            result = sitk.Euler2DTransform()
-        else:
-            result = sitk.Euler3DTransform()
-        parameters = format(transform["TransformParameters"])
-        fixedParameters = format(transform["CenterOfRotationPoint"])+[0]
-    elif transform["Transform"][0] == "TranslationTransform":
-        result = sitk.TranslationTransform(dimension)
-        parameters = format(transform["TransformParameters"])
-        fixedParameters = []
-    elif transform["Transform"][0] == "AffineTransform":
-        result = sitk.AffineTransform(dimension)
-        parameters = format(transform["TransformParameters"])
-        fixedParameters = format(transform["CenterOfRotationPoint"])+[0]
-    elif transform["Transform"][0] == "BSplineStackTransform":
-        parameters = format(transform["TransformParameters"])
-        GridSize = format(transform["GridSize"])
-        GridOrigin = format(transform["GridOrigin"])
-        GridSpacing = format(transform["GridSpacing"])
-        GridDirection =  np.asarray(format(transform["GridDirection"])).reshape((dimension, dimension)).T.flatten() 
-        fixedParameters = np.concatenate([GridSize, GridOrigin, GridSpacing, GridDirection])
-
-        nb = int(format(transform["Size"])[-1])
-        sub = int(np.prod(GridSize))*dimension
-        results = []
-        for i in range(nb):
-            result = sitk.BSplineTransform(dimension)
-            sub_parameters = np.asarray(parameters[i*sub:(i+1)*sub])
-            result.SetFixedParameters(fixedParameters)
-            result.SetParameters(sub_parameters)
-            results.append(result)
-        return results
-    elif transform["Transform"][0] == "AffineLogStackTransform":
-        parameters = format(transform["TransformParameters"])
-        fixedParameters = format(transform["CenterOfRotationPoint"])+[0]
-
-        nb = int(transform["NumberOfSubTransforms"][0])
-        sub = dimension*4
-        results = []
-        for i in range(nb):
-            result = sitk.AffineTransform(dimension)
-            sub_parameters = np.asarray(parameters[i*sub:(i+1)*sub])
-
-            result.SetFixedParameters(fixedParameters)
-            result.SetParameters(np.concatenate([scipy.linalg.expm(sub_parameters[:dimension*dimension].reshape((dimension,dimension))).flatten(), sub_parameters[-dimension:]]))
-            results.append(result)
-        return results
-    else:
-        result = sitk.BSplineTransform(dimension)
-        
-        parameters = format(transform["TransformParameters"])
-        GridSize = format(transform["GridSize"])
-        GridOrigin = format(transform["GridOrigin"])
-        GridSpacing = format(transform["GridSpacing"])
-        GridDirection = np.array(format(transform["GridDirection"])).reshape((dimension,dimension)).T.flatten() 
-        fixedParameters = np.concatenate([GridSize, GridOrigin, GridSpacing, GridDirection])
-
-    result.SetFixedParameters(fixedParameters)
-    result.SetParameters(parameters)
-    return result
-
-if __name__ == "__main__":
-    out_path = sys.argv[1]
-    finename = sys.argv[2]
-    finename_dest = sys.argv[3]
-    transform = parameterMap_to_transform("{}/{}".format(out_path, finename))
-    sitk.WriteTransform(transform, "{}/{}".format(out_path, finename_dest))
-
-def getFlatLabel(mask: sitk.Image, labels: list[int]) -> sitk.Image:
-    data = sitk.GetArrayFromImage(mask)
-    result_data = np.zeros_like(data, np.uint8)
-
-    for label in labels:
-        result_data[data == label] = 1
-
-    result = sitk.GetImageFromArray(result_data)
-    result.CopyInformation(mask)        
-    return result
-
-def rampFilterHistogram(image: sitk.Image, rampStart: float, rampEnd: float) -> sitk.Image:
-    imageData = sitk.GetArrayFromImage(image)
-    filter = np.logical_and(imageData > rampStart, imageData < rampEnd)
-    rampWidth = rampEnd - rampStart
-    imageData[filter] = (1/rampWidth) * (imageData[filter] - rampStart) * imageData[filter]
-    filteredImage = sitk.GetImageFromArray(imageData)
-    filteredImage.CopyInformation(image)
-    return filteredImage
-
-def elastic_registration(   fixed_image : sitk.Image,
-                            moving_image : sitk.Image,
-                            fixed_mask : Union[sitk.Image, None],
-                            moving_mask : Union[sitk.Image, None],
-                            name_parameterMap : str,
-                            outputDir: str) -> sitk.Transform:
-        labels = np.unique(sitk.GetArrayFromImage(fixed_mask))
-        fixed_mask = getFlatLabel(fixed_mask, labels[1:])
-        moving_mask = getFlatLabel(moving_mask, labels[1:])
-
-        fixed_mask.CopyInformation(fixed_image)
-        moving_mask.CopyInformation(moving_image)
-
-        fixed_mask_dillated = sitk.BinaryDilate(fixed_mask, [5,5,5])
-        moving_mask_dillated = sitk.BinaryDilate(moving_mask, [5,5,5])
-        
-        fixed_image = sitk.Mask(fixed_image, fixed_mask)
-        moving_image = sitk.Mask(moving_image, moving_mask)
-
-
-        minGradientMagnitude = 50
-        fixed_image_gradient = rampFilterHistogram(sitk.VectorMagnitude(sitk.Gradient(fixed_image)), 0, minGradientMagnitude)
-        moving_image_gradient = rampFilterHistogram(sitk.VectorMagnitude(sitk.Gradient(moving_image)), 0, minGradientMagnitude)
-
-        elastixImageFilter = sitk.ElastixImageFilter()
-        elastixImageFilter.SetFixedImage(fixed_image_gradient)
-        elastixImageFilter.AddFixedImage(fixed_image)
-        elastixImageFilter.AddFixedImage(fixed_image)
-
-        if fixed_mask is not None:
-            elastixImageFilter.SetFixedMask(fixed_mask)
-            elastixImageFilter.AddFixedMask(fixed_mask_dillated)
-            elastixImageFilter.AddFixedMask(fixed_mask_dillated)
-
-
-        elastixImageFilter.SetMovingImage(moving_image_gradient)
-        elastixImageFilter.AddMovingImage(moving_image)
-        elastixImageFilter.AddMovingImage(moving_image)
-        
-        if moving_mask is not None:
-            elastixImageFilter.SetMovingMask(moving_mask)
-            elastixImageFilter.AddMovingMask(moving_mask_dillated)
-            elastixImageFilter.AddMovingMask(moving_mask_dillated)
-
-        elastixImageFilter.SetParameterMap(sitk.ReadParameterFile("{}.txt".format(name_parameterMap)))
-        elastixImageFilter.LogToConsoleOn()
-        elastixImageFilter.SetOutputDirectory(outputDir)
-        
-        elastixImageFilter.Execute()
-        
-        transform = parameterMap_to_transform("{}TransformParameters".format(outputDir))
-        
-        return transform
-
-def registration(   fixed_image : sitk.Image,
-                    moving_image : sitk.Image,
-                    fixed_mask : Union[sitk.Image, None],
-                    moving_mask : Union[sitk.Image, None],
-                    name_parameterMap : str,
-                    outputDir: str) -> sitk.Transform:
-        elastixImageFilter = sitk.ElastixImageFilter()
-        elastixImageFilter.SetFixedImage(fixed_image)
-        if fixed_mask is not None:
-            elastixImageFilter.SetFixedMask(fixed_mask)
-
-        elastixImageFilter.SetMovingImage(moving_image)
-        if moving_mask is not None:
-            elastixImageFilter.SetMovingMask(moving_mask)
-
-        elastixImageFilter.SetParameterMap(sitk.ReadParameterFile("{}.txt".format(name_parameterMap)))
-        elastixImageFilter.LogToConsoleOn()
-        elastixImageFilter.SetOutputDirectory(outputDir)
-        elastixImageFilter.Execute()
-        
-        transform = parameterMap_to_transform("{}TransformParameters".format(outputDir))
-        
-        return transform
-
-def registration_groupewise(images_1: sitk.Image, masks: sitk.Image, images_2: sitk.Image, name_parameterMap : str, output_dir: str):
-    elastixImageFilter = sitk.ElastixImageFilter()
-    elastixImageFilter.SetFixedImage(images_1)
-    elastixImageFilter.SetMovingImage(images_1)
-    elastixImageFilter.SetFixedMask(masks)
-    elastixImageFilter.SetMovingMask(masks)
-    
-    if images_2 is not None:
-        elastixImageFilter.AddFixedImage(images_2)
-        elastixImageFilter.AddMovingImage(images_2)
-        #elastixImageFilter.AddFixedImage(images_2)
-        #elastixImageFilter.AddMovingImage(images_2)
-    
-    elastixImageFilter.SetParameterMap(sitk.ReadParameterFile("{}.txt".format(name_parameterMap)))
-    elastixImageFilter.LogToConsoleOn()
-    elastixImageFilter.SetOutputDirectory(output_dir)
-    elastixImageFilter.LogToFileOn()
-    elastixImageFilter.Execute()
-
-    transforms = parameterMap_to_transform("{}TransformParameters".format(output_dir))
-    return transforms

konfai-1.1.8.dist-info/RECORD

@@ -1,39 +0,0 @@
-konfai/__init__.py,sha256=YXG-wpSEXWs6Jt3BDI77V4r89gEUNX-6lxW9btj5VYI,851
-konfai/evaluator.py,sha256=WM78NGydV0iqKElahr-WpOCZLEJXHjjPq-LS-gi23rk,8401
-konfai/main.py,sha256=kr7Iie_f67NF6G3dAAj9G6Z9dhn9RzbdLYpzy2WvIh8,2573
-konfai/predictor.py,sha256=ih0GVkYEhIxvolDdQ-Cr9mWTNvtIGyGadtrT_RAkzys,23052
-konfai/trainer.py,sha256=Edi8l8OOfCewYM-Cd5C5rCqCaprvlfxzohd4iLkK5u0,20632
-konfai/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-konfai/data/augmentation.py,sha256=mFVMpbJ8WBKGbMILdmTZYBA8k7kRDQVPOEy1A9t5QP4,32281
-konfai/data/data_manager.py,sha256=FHRGiSg5HcXmlMWntdNgv0_bty_gwaDUBWYt5LSw86k,29020
-konfai/data/patching.py,sha256=zAm6jjUW--lsqTBlDFICVlj7O_QOlSxStFAHS9S9H8I,14753
-konfai/data/transform.py,sha256=RSVCjHshw74yk4QV6d3lPRhqkwP77T08rI3YkNV6nTQ,26513
-konfai/metric/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-konfai/metric/measure.py,sha256=S65WGBRYMoalkwI_Le3C6K0tQmT4Qft82nUTBwhpvFU,26907
-konfai/metric/schedulers.py,sha256=VPp7zEwD0AtQz51XG0TlutD_NrsTZs4fstC5h8A8f8U,1309
-konfai/models/classification/convNeXt.py,sha256=hZQ_9I1lJW7DX8QnoeZ9L1br-98apV9VXGrsTAN25ds,9261
-konfai/models/classification/resnet.py,sha256=eBoS_zNczfa1EnECKKLr0Ow26ekR_4W5eji6p9cPfHk,7978
-konfai/models/generation/cStyleGan.py,sha256=7D8zZveDEZapFeaaDTe3wVhuDCCHLqq6cpXtl2-QaJA,8059
-konfai/models/generation/ddpm.py,sha256=jXb0eOU3i_gMHvj9pawVAWQDjMGl6SPfCc5m2Jzdrik,13175
-konfai/models/generation/diffusionGan.py,sha256=ZFKdRHvlajEPyw9_HCLo0Q92iChCtZd2dvi2nEF_tBI,33218
-konfai/models/generation/gan.py,sha256=RzpGNu8BlBVDCxFR5CCmEFYUzVBs2YrDl7trIos3ssw,7861
-konfai/models/generation/vae.py,sha256=zH8qk04z2lXDhXAVxTHiRIVMfRi3brB61lHGmEsE3kM,4675
-konfai/models/registration/registration.py,sha256=EAE3w8aic2fPWiJz0ilqrs2kCGUQD6NWvysVfHXxA_g,6329
-konfai/models/representation/representation.py,sha256=TiYcBBqZYySpwsRlnnBQh0QVW29Rcvb9GUjsqnCKKLM,2689
-konfai/models/segmentation/NestedUNet.py,sha256=hDSE7BJ17IqaiHWAD5zlVG7G_KvQzuQmVnAVsiYVE6E,10248
-konfai/models/segmentation/UNet.py,sha256=BktCRfAcCDtvGCw8wGfyZvBtT4G0Oy8teIcVgDFOurk,4078
-konfai/network/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-konfai/network/blocks.py,sha256=C_dOLmXovxANWCx7P439FzR_95Zisy2I1F7FEwwz7AE,14412
-konfai/network/network.py,sha256=LHeA7HtsVYO7BJu2_kqh23q2GIANn5ZSa4LhKMt7dJg,48642
-konfai/utils/ITK.py,sha256=OxTieDNNYHGkn7zxJsAG-6ecRG1VYMvn1dlBbBe1DOs,13955
-konfai/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-konfai/utils/config.py,sha256=f7o83ix5_oNbr2pki-Czqr-yHi-8n92ZL64nlo0XGwA,12514
-konfai/utils/dataset.py,sha256=6ZzevdhJ7e5zlXATAVwSh9O6acKXM7gYNxkMAa5DrmM,36351
-konfai/utils/registration.py,sha256=v1srEBOcgDnHrx0YtsK6bcj0yCMH7wNeaQ3wC7gEvOw,8898
-konfai/utils/utils.py,sha256=mQQ6FB0Jw7Odg5GxYeTApR6lvFhXn_iz-GVGZRngGQA,24934
-konfai-1.1.8.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-konfai-1.1.8.dist-info/METADATA,sha256=nHfQDnRa5XjIHXCoVSRY7JX7pP-kSoaWnb0Nt1IwQaU,2515
-konfai-1.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-konfai-1.1.8.dist-info/entry_points.txt,sha256=fG82HRN5-g39ACSOCtij_I3N6EHxfYnMR0D7TI_8pW8,81
-konfai-1.1.8.dist-info/top_level.txt,sha256=xF470dkIlFoFqTZEOlRehKJr4WU_8OKGXrJqYm9vWKs,7
-konfai-1.1.8.dist-info/RECORD,,