predtiler 0.0.1__tar.gz → 0.0.2__tar.gz

{predtiler-0.0.1 → predtiler-0.0.2}/PKG-INFO

@@ -1,7 +1,7 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: predtiler
-Version: 0.0.1
-Summary: Converting your dataset class into a class that can be used for tiled prediction and eventually obtain stiched prediction.
+Version: 0.0.2
+Summary: Subclassing your dataset class into a new class that can be used for tiled prediction to obtain the stitched prediction.
 Project-URL: homepage, https://github.com/ashesh-0/PredTiler
 Project-URL: repository, https://github.com/ashesh-0/PredTiler
 Author: Ashesh
@@ -86,7 +86,7 @@ For this we also need a tile manager that will manage the tiles.
 
 ```python
 
-from predtiler.dataset import get_tiling_dataset, get_tile_manager
+from predtiler import get_tiling_dataset, get_tile_manager, stitch_predictions
 patch_size = 256
 tile_size = 128
 data_shape = (10, 2048, 2048) # size of the data you are working with

{predtiler-0.0.1 → predtiler-0.0.2}/README.md

@@ -57,7 +57,7 @@ For this we also need a tile manager that will manage the tiles.
 
 ```python
 
-from predtiler.dataset import get_tiling_dataset, get_tile_manager
+from predtiler import get_tiling_dataset, get_tile_manager, stitch_predictions
 patch_size = 256
 tile_size = 128
 data_shape = (10, 2048, 2048) # size of the data you are working with

{predtiler-0.0.1 → predtiler-0.0.2}/pyproject.toml

@@ -16,8 +16,8 @@ sources = ["src"]
 # https://peps.python.org/pep-0621/
 [project]
 name = "predtiler"
-version = "0.0.1"
-description = "Converting your dataset class into a class that can be used for tiled prediction and eventually obtain stiched prediction."
+version = "0.0.2"
+description = "Subclassing your dataset class into a new class that can be used for tiled prediction to obtain the stitched prediction."
 readme = "README.md"
 requires-python = ">=3.9"
 license = { text = "MIT" }

predtiler-0.0.2/src/predtiler/__init__.py

@@ -0,0 +1,8 @@
+from .tile_stitcher import stitch_predictions
+from .dataset import get_tiling_dataset, get_tile_manager
+
+__all__ = [
+    'stitch_predictions',
+    'get_tiling_dataset',
+    'get_tile_manager'
+]

{predtiler-0.0.1 → predtiler-0.0.2}/src/predtiler/tile_stitcher.py

@@ -15,8 +15,7 @@ def stitch_predictions(predictions:np.ndarray, manager):
     mng = manager
     shape = list(mng.data_shape)
     shape.append(predictions.shape[1])
-    print(shape)
-
+    assert mng.patch_shape[-2:] == predictions.shape[-2:], 'Patch shape and predictions shape must match. Please set the patch shape correctly'
     output = np.zeros(shape, dtype=predictions.dtype)
     for dset_idx in range(predictions.shape[0]):
         # grid start, grid end

{predtiler-0.0.1 → predtiler-0.0.2}/tests/test_full_tiling_setup.py

@@ -1,7 +1,6 @@
 from unittest.mock import Mock
 import numpy as np
-from predtiler.dataset import get_tiling_dataset, get_tile_manager
-from predtiler.tile_stitcher import stitch_predictions
+from predtiler import get_tiling_dataset, get_tile_manager, stitch_predictions
 
 def get_data_3D(n=5,Z=9, H=512,W=512,C=2):
     data = np.arange(n*Z*H*W*C).reshape(n,Z,H,W,C)