ValueError: Floating point image RGB values must be in the 0..1 range. while using matplotlib

Question

I want to visualize weights of the layer of a neural network. I\'m using pytorch.

import torch
import torchvision.models as models
from matplotlib import pyp         


        

Answer 1

It sounds as if you already know your values are not in that range. Yes, you must re-scale them to the range 0.0 - 1.0. I suggest that you want to retain visibility of negative vs positive, but that you let 0.5 be your new "neutral" point. Scale such that current 0.0 values map to 0.5, and your most extreme value (largest magnitude) scale to 0.0 (if negative) or 1.0 (if positive).

---

Thanks for the vectors. It looks like your values are in the range -2.25 to +2.0. I suggest a rescaling new = (1/(2*2.25)) * old + 0.5