Inconsistent conversion of VectorImage to numpy

[ncullen93]

In itk Python, if I have a VectorImage of size (10,10) with 2 channels, then converting it to numpy gives me an array of shape (10,10,2). But if I have a VectorImage with 1 channel, it squeezes that channel and gives me a numpy array of (10,10). Shouldn't the expected behavior be (10,10,1). Or is there a way to ensure this?

import itk
import numpy as np

arr = np.ones((10,10,2))
img = itk.image_from_array(arr, is_vector=True)
arr2 = itk.array_from_image(img)
print(arr2.shape) # (10,10,2)

arr = np.ones((10,10,1))
img = itk.image_from_array(arr, is_vector=True)
arr2 = itk.array_from_image(img)
print(arr2.shape) # (10,10)

[github-actions[bot]]

Thank you for contributing an issue! 🙏

Welcome to the ITK community! 🤗👋☀️

We are glad you are here and appreciate your contribution. Please keep in mind our community participation guidelines. 📜 Also, please check existing open issues and consider discussion on the ITK Discourse. 📖

This is an automatic message. Allow for time for the ITK community to be able to read the issue and comment on it.

[thewtex]

Hi, thanks for the nice report! Patch is here: #4648