yolo_standard_libray/tensorrtx-master/efficient_ad/src/postprocess.h
2025-03-07 11:35:40 +08:00

10 lines
280 B
C

#pragma once
#include <opencv2/opencv.hpp>
void genHeatMap(cv::Mat originImg, cv::Mat& anomalyGrayMap, cv::Mat& HeatMap) {
cv::Mat colorMap;
cv::applyColorMap(colorMap, anomalyGrayMap, cv::COLORMAP_JET);
cv::addWeighted(originImg, 0.5, colorMap, 0.5, 0, HeatMap);
}