konfai 1.2.7__py3-none-any.whl → 1.2.9__py3-none-any.whl

konfai/data/patching.py

@@ -218,7 +218,7 @@ class Patch(ABC):
             ),
         )
         slices = [s] + list(self._patch_slices[a][index][1:])
-        data_sliced = data[slices_pre + slices]
+        data_sliced = data[tuple(slices_pre + slices)]
         if extend_slice > 0 and data_sliced.shape[len(slices_pre)] < bottom + top + 1:
             pad_bottom = 0
             pad_top = 0

konfai/data/transform.py

@@ -311,7 +311,7 @@ class Padding(TransformInverse):
         slices = [slice(0, shape) for shape in tensor.shape]
         for dim in range(len(self.padding) // 2):
             slices[-dim - 1] = slice(self.padding[dim * 2], tensor.shape[-dim - 1] - self.padding[dim * 2 + 1])
-        result = tensor[slices]
+        result = tensor[tuple(slices)]
         return result
 
 
@@ -368,7 +368,7 @@ class ResampleToResolution(Resample):
         super().__init__(inverse)
         self.spacing = torch.tensor([0 if s < 0 else s for s in spacing])
 
-    def transform_shape(self, shape: list[int], cache_attribute: Attribute) -> list[int]:
+    def transform_shape(self, name: str, shape: list[int], cache_attribute: Attribute) -> list[int]:
         if "Spacing" not in cache_attribute:
             TransformError(
                 "Missing 'Spacing' in cache attributes, the data is likely not a valid image.",
@@ -383,7 +383,7 @@ class ResampleToResolution(Resample):
             if s == 0:
                 spacing[i] = image_spacing[i]
         resize_factor = spacing / image_spacing
-        return [int(x) for x in (torch.tensor(shape) * 1 / resize_factor)]
+        return [int(x) for x in (torch.tensor(shape) * 1 / resize_factor.flip(0))]
 
     def __call__(self, name: str, tensor: torch.Tensor, cache_attribute: Attribute) -> torch.Tensor:
         image_spacing = cache_attribute.get_tensor("Spacing")
@@ -394,7 +394,7 @@ class ResampleToResolution(Resample):
         resize_factor = spacing / cache_attribute.get_tensor("Spacing")
         cache_attribute["Spacing"] = spacing
         cache_attribute["Size"] = np.asarray([int(x) for x in torch.tensor(tensor.shape[1:])])
-        size = [int(x) for x in (torch.tensor(tensor.shape[1:]) * 1 / resize_factor)]
+        size = [int(x) for x in (torch.tensor(tensor.shape[1:]) * 1 / resize_factor.flip(0))]
         cache_attribute["Size"] = np.asarray(size)
         return self._resample(tensor, size)
 

{konfai-1.2.7.dist-info → konfai-1.2.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: konfai
-Version: 1.2.7
+Version: 1.2.9
 Summary: Modular and configurable Deep Learning framework with YAML and PyTorch
 Author-email: Valentin Boussot <boussot.v@gmail.com>
 License-Expression: Apache-2.0

{konfai-1.2.7.dist-info → konfai-1.2.9.dist-info}/RECORD

@@ -6,8 +6,8 @@ konfai/trainer.py,sha256=g_TkPDUjToFGDGB7aaRZMn-fQllHV_I2GHFKUzDGF8o,27106
 konfai/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 konfai/data/augmentation.py,sha256=7jrWcpw61t3cCIdHtUlnekRO7JwXIM5Q6RIXY8Ya-xM,27796
 konfai/data/data_manager.py,sha256=tZ2DZHDW4UySCCzwEzR2WIL0fTp7lqAfqEbNPiEw5NE,31064
-konfai/data/patching.py,sha256=P0TcjR4qcUWpB_Uph0-dd8bMeNVJC_IGNK_jkxStglQ,16526
-konfai/data/transform.py,sha256=VVDaQOBdgvHhOdoJlo9fkOJ2ln8th_0u8go1I4Uy7eo,30332
+konfai/data/patching.py,sha256=PNxRFk2wkB5yWRJ6krj6slvUXRq0Tb7ysp4E2IZkUp8,16533
+konfai/data/transform.py,sha256=RB0iT3GX9O4OeuwOdFV_kSdLN6ML2WiHTzQUEyRJGv0,30366
 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
@@ -30,9 +30,9 @@ 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.7.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-konfai-1.2.7.dist-info/METADATA,sha256=CVuc5KlcaickUPpttTyIdhdRHGm3h8bEWSeLidc5mUs,2475
-konfai-1.2.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-konfai-1.2.7.dist-info/entry_points.txt,sha256=fG82HRN5-g39ACSOCtij_I3N6EHxfYnMR0D7TI_8pW8,81
-konfai-1.2.7.dist-info/top_level.txt,sha256=xF470dkIlFoFqTZEOlRehKJr4WU_8OKGXrJqYm9vWKs,7
-konfai-1.2.7.dist-info/RECORD,,
+konfai-1.2.9.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+konfai-1.2.9.dist-info/METADATA,sha256=pGFfOyf57TF55Xi8mfsFPLhhexr3i_YaA7XCev2A3v4,2475
+konfai-1.2.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+konfai-1.2.9.dist-info/entry_points.txt,sha256=fG82HRN5-g39ACSOCtij_I3N6EHxfYnMR0D7TI_8pW8,81
+konfai-1.2.9.dist-info/top_level.txt,sha256=xF470dkIlFoFqTZEOlRehKJr4WU_8OKGXrJqYm9vWKs,7
+konfai-1.2.9.dist-info/RECORD,,