...e para os outros o (valor*255)/max, parece que funcionou mas é assim mesmo que normaliza?
na verdade não, mas para esse EP é suficiente escalar o intervalo (0 - max) para (0 - 255).
Se a gente for mais rigoroso, tem ainda o problema do tipo (8 bits e não um real) e uma normalização "genérica" deveria levar em conta números negativos. Como valores negativos não aparecem no resultado do módulo do gradiente e o Python (plt) tolera o tipo, a formula de conversão sugerida é suficiente para esse EP.