Noise2Void shape broadcast error during prediction

[astronerdF]

I have followed the 293_denoising_RGB_images_using_deep_learning.ipynb and everything seemed fine except while trying to predict.

This is the error that I get: `pred = model.predict(img, axes='YXC')

---

ValueError Traceback (most recent call last) Cell In[26], line 5 1 # Here we denoise the image (predict) 2 # The parameter 'n_tiles' can be used if images are to big for the GPU memory. 3 # If we do not provide the n_tiles' parameter the system will automatically try to find an appropriate tiling. 4 # This can take longer. ----> 5 pred = model.predict(img, axes='YXC')

File ~/Desktop/N2VTensorFlow/n2vTF/lib/python3.8/site-packages/n2v/models/n2v_standard.py:382, in N2V.predict(self, img, axes, resizer, n_tiles, tta) 380 if n_tiles: 381 new_n_tiles = tuple([n_tiles[axes.index(c)] for c in axes if c != 'C']) + (n_tiles[axes.index('C')],) --> 382 normalized = self.normalize(np.moveaxis(img, axes.index('C'), -1), means, stds) 383 else: 384 normalized = self.normalize(img[..., np.newaxis], means, stds)

File ~/Desktop/N2VTensorFlow/n2vTF/lib/python3.8/site-packages/n2v/models/n2v_standard.py:341, in N2V.normalize(self, data, means, stds) 340 def normalize(self, data, means, stds): --> 341 return (data - means) / stds

ValueError: operands could not be broadcast together with shapes (359,497,4) (1,1,3)`