add 'pynrrd' to README

juglab · Jan 21, 2024 · 52168d1 · 52168d1
1 parent 3d6ee3e
commit 52168d1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -36,6 +36,7 @@ conda env create -f ./env.yml
 - `pytorch`
 - `tochvision`
 - `timm`
+- `pynrrd`
 
 ### Installing The Plugin
 To install the plugin, open the command prompt in the plugin directory, and run the following command:

diff --git a/src/napari_sam_labeling_tools/utils/data.py b/src/napari_sam_labeling_tools/utils/data.py
@@ -48,9 +48,6 @@ def unpatchify(
     batch_patches = embed_patches.reshape(
         batch_size, num_patches_h, num_patches_w, c, patch_size, patch_size
     ).contiguous()
-    padded_images = torch.zeros((
-        batch_size, c, img_h + pad_bottom, img_w + pad_right
-    ))
     # print(padded_images.shape)
     target_patches = batch_patches[
         :, :, :, :, margin: margin + target_size, margin: margin + target_size
@@ -60,6 +57,9 @@ def unpatchify(
         batch_size, c, num_patches_h * target_size, num_patches_w * target_size
     )
     # print(target_patches.shape)
+    padded_images = torch.zeros((
+        batch_size, c, img_h + pad_bottom, img_w + pad_right
+    ))
     padded_images[
         :, :, : num_patches_h * target_size, : num_patches_w * target_size
     ] = target_patches