pytorch-grad-cam
pytorch-grad-cam copied to clipboard
Address cv2.resize float16 issue in scale_cam_image
https://github.com/jacobgil/pytorch-grad-cam/issues/198 https://github.com/jacobgil/pytorch-grad-cam/issues/493
Putting in a PR to go along with my issue post.
cv2.resize doesn't support float16 which causes this step to break. This fixes the issue with not downstream consequences that I can identify since the resulting image gets converted to lfoat32 anyways.
any chance we can merge and release? @Christophe-Foyer
Merged. Thank you for the contribution.