fix: handle 4D

8c7cf9d3 · Jonathan Passerat-Palmbach · b2fe9465 · 8c7cf9d3
Commit 8c7cf9d3 authored Apr 25, 2017 by Jonathan Passerat-Palmbach
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

nithumb.py nithumb.py +2 -1

No files found.
--- a/nithumb.py
+++ b/nithumb.py
@@ -34,7 +34,8 @@ def main():
    image = ((image - image.min()) *
             (255 / (image.max() - image.min()))).astype('uint8')

-    # Select the centre slice if 3D.
+    # Select the centre slice if 3D. Centre frame if 4D
+    image = image[:, :, :, image.shape[3]//2] if image.ndim == 4 else image
    image = image[:, :, image.shape[2]//2] if image.ndim == 3 else image

    # Generate and save the snapshot and its thumbnail.