eballesteros

I have noticed the output shape for visualize_cam is different for 0.41 version and for master. with `input_cam.shape` (544, 544, 3) this: ```python out_cam = visualize_cam(model = my_model, layer_idx =...

I ended up adding ```python if len(out_cam.shape) == 2: out_cam = (cm.jet(out_cam)[:,:,:-1]*255).astype(np.uint8) #to RGBA - A ``` to make the code compatible with bot 0.41 and posterior versions

+1 to this feature request since I think it also tackles my use-case. I am collaborating with a team, working with a loading script which takes some time to generate...