Error using the `Show test images in ImageJ` button

[IvanHCenalmor]

Hi @petebankhead ,
I was testing the Bioimage Model Zoo extension and I have had an issue while using the Show test images in ImageJ button.

I have used the 2D UNet nucleid broad model to test the extension and using the default values, when using the Show test images in ImageJ button the output that I receive in ImageJ is:

The target image is uploaded as a 256x2 pixel image with 256 channels, instead of what I suppose that it should be a 256x256 pixel image with 2 channels. Therefore, as the comparison between target and prediction images cannot be done, I suppose that might be the reason why no difference image is neither loaded.

If you need more information or anything, let me know.
Iván

[petebankhead]

Hi @IvanHCenalmor I can replicate this. Will need to investigate - it seems to be because the NumPy dimensions are especially awkward (I think 1, 2, 256, 256) and it seems we don't handle that yet.