img: bgr, [0,255] heatmap: [0,1]
(img, heatmap)
| 25 | |
| 26 | |
| 27 | def colorize(img, heatmap): |
| 28 | """ img: bgr, [0,255] |
| 29 | heatmap: [0,1] |
| 30 | """ |
| 31 | heatmap = viz.intensity_to_rgb(heatmap, cmap='jet')[:, :, ::-1] |
| 32 | return img * 0.5 + heatmap * 0.5 |
| 33 | |
| 34 | |
| 35 | @memoized |