How did you use custom image for pytorch cnn visualization? I am getting this error:

Blockquote
Traceback (most recent call last):
File “scorecam.py”, line 98, in 
save_class_activation_images(original_image, cam, file_name_to_export)
File “/home/pracheta/COVID19/pytorch-cnn-visualizations-master/src/misc_functions.py”, line 65, in save_class_activation_images
heatmap, heatmap_on_image = apply_colormap_on_image(org_img, activation_map, ‘hsv’)
File “/home/pracheta/COVID19/pytorch-cnn-visualizations-master/src/misc_functions.py”, line 97, in apply_colormap_on_image
heatmap_on_image = Image.alpha_composite(heatmap_on_image, heatmap)
File “/home/pracheta/.local/lib/python3.6/site-packages/PIL/Image.py”, line 2916, in alpha_composite
return im1._new(core.alpha_composite(im1.im, im2.im))
ValueError: images do not match


Blockquote