add YOLP heatmap