Imported from https://github.com/adityac94/Grad_CAM_plus_plus