MiDaS深度估计教程：热力图颜色映射原理详解

1. 引言：AI 单目深度估计的视觉革命

在计算机视觉领域，从二维图像中恢复三维空间信息一直是极具挑战性的任务。传统方法依赖双目立体视觉或多传感器融合，而近年来，单目深度估计（Monocular Depth Estimation）凭借深度学习的强大表征能力，实现了仅凭一张RGB图像即可推断场景深度结构的技术突破。

Intel ISL 实验室推出的MiDaS（Mixed Data Set）模型正是这一方向的代表性成果。它通过在大规模混合数据集上训练，能够泛化到各种复杂场景，输出高质量的相对深度图。更关键的是，其结果可通过热力图可视化，将不可见的“距离”转化为直观的色彩分布——近处暖色、远处冷色，极大增强了人机交互的理解性。

本文将围绕基于 MiDaS 的深度估计服务展开，重点解析其核心功能背后的热力图颜色映射机制，并结合工程实践说明如何实现稳定、高效的 CPU 推理与可视化输出。

2. MiDaS 模型架构与推理流程解析

2.1 MiDaS 的核心技术原理

MiDaS 的设计目标是跨数据集泛化能力，即在一个统一框架下处理来自不同来源、标注方式各异的深度数据。为达成此目标，MiDaS 采用了一种尺度不变的损失函数（Scale-Invariant Loss），使得模型不关心绝对物理距离，而是专注于学习像素间的相对远近关系。

其主干网络通常基于EfficientNet-B5 或 Transformer 架构（如 DPT 模型），具备强大的特征提取能力。输入图像经过归一化后送入网络，输出一个与原图尺寸一致的单通道深度张量（Depth Map），其中每个值代表对应像素点的“深度响应强度”。

📌 注意：这里的“深度值”并非真实世界中的米或厘米，而是经过归一化的相对数值，越大表示越远，越小表示越近。

2.2 推理流程拆解

以MiDaS_small模型为例，整个推理过程可分为以下步骤：

图像预处理：
调整图像大小至 384×384（模型输入要求）
归一化像素值到 [0,1] 区间
转换为 PyTorch Tensor 并添加 batch 维度
模型前向传播：python import torch model = torch.hub.load("intel-isl/MiDaS", "MiDaS_small") depth_map = model(img_tensor)
后处理与归一化：
将输出的深度图从 GPU/CPU 张量转为 NumPy 数组
进行 min-max 归一化至 [0, 255] 范围，便于后续颜色映射
热力图生成：
使用 OpenCV 的applyColorMap函数，将灰度深度图映射为彩色热力图

该流程高度模块化，适合部署在资源受限的 CPU 环境中，尤其适用于边缘设备和轻量级 WebUI 应用。

3. 热力图颜色映射机制深度剖析

3.1 什么是热力图？为什么需要它？

原始深度图是一个单通道灰度图像，亮度越高表示距离越远。然而，人类对灰度变化的感知有限，难以快速识别细微差异。为此，引入伪彩色映射（Pseudocolor Mapping）技术，将灰度值映射到丰富的色彩空间，形成所谓的“热力图”。

在本项目中，选用的是 OpenCV 内置的Inferno 颜色表，其特点是： - 黑 → 深蓝 → 紫红 → 橙黄 → 白 - 动态范围广，对比度高 - 视觉冲击力强，科技感十足

3.2 Inferno 映射原理与数学表达

OpenCV 提供了多种预定义的颜色查找表（Colormap LUT），cv2.COLORMAP_INFERNO是其中之一。其本质是一个长度为 256 的 RGB 查找表，每个索引对应一种颜色。

映射过程如下：

import cv2 import numpy as np # 假设 depth_normalized 是 [0, 255] 范围内的 uint8 深度图 depth_colored = cv2.applyColorMap(depth_normalized, cv2.COLORMAP_INFERNO)

具体映射逻辑可理解为分段函数逼近：

输入灰度值区间	主导颜色趋势
0 ~ 64	黑 → 深蓝（暗部细节）
64 ~ 128	蓝 → 紫
128 ~ 192	紫 → 红 → 橙
192 ~ 255	橙 → 黄 → 白（亮部峰值）

这种非线性映射能有效增强中间区域的视觉区分度，避免传统线性映射导致的“过曝”或“死黑”问题。

3.3 自定义颜色映射的可能性

虽然 Inferno 效果出众，但在特定应用场景下也可自定义调色板。例如医学影像偏好Jet或Plasma，而工业检测可能倾向Viridis（对色盲友好）。

示例：构建自定义渐变映射

def create_custom_colormap(): # 创建 (256, 1, 3) 的 LUT 表 lut = np.zeros((256, 1, 3), dtype=np.uint8) for i in range(256): if i < 85: lut[i, 0] = [int(255 * i / 85), 0, int(255 * (85 - i) / 85)] # 绿→黑 elif i < 170: lut[i, 0] = [int(255 * (i - 85) / 85), int(255 * (i - 85) / 85), 0] # 黑→黄 else: lut[i, 0] = [255, int(255 * (255 - i) / 85), 0] # 黄→红 return lut custom_lut = create_custom_colormap() depth_custom = cv2.LUT(depth_normalized, custom_lut)

这为开发者提供了极大的灵活性，可根据业务需求调整视觉风格。

4. WebUI 集成与工程优化实践

4.1 系统架构设计

本项目的整体架构如下：

[用户上传图片] ↓ [Flask/FastAPI 后端接收] ↓ [图像预处理 → MiDaS 推理 → 归一化 → 热力图生成] ↓ [返回深度图 + 热力图 Base64 编码] ↓ [前端页面展示]

所有组件均运行于 CPU 环境，依赖精简，无 Token 验证环节，确保高可用性和稳定性。

4.2 关键代码实现

以下是核心推理与可视化模块的完整实现：

import torch import cv2 import numpy as np from PIL import Image def predict_depth(image_path: str) -> tuple: """ 输入图像路径，返回原始深度图和彩色热力图 """ # 加载模型（建议全局初始化一次） device = torch.device("cpu") model = torch.hub.load("intel-isl/MiDaS", "MiDaS_small").to(device) model.eval() # 图像加载与预处理 img = Image.open(image_path).convert("RGB") transform = torch.hub.load("intel-isl/MiDaS", "transforms").small_transform img_tensor = transform(img).unsqueeze(0).to(device) # 推理 with torch.no_grad(): prediction = model(img_tensor) # 后处理 depth_map = prediction.squeeze().cpu().numpy() depth_normalized = cv2.normalize(depth_map, None, 0, 255, cv2.NORM_MINMAX).astype(np.uint8) # 生成热力图 heat_map = cv2.applyColorMap(depth_normalized, cv2.COLORMAP_INFERNO) return depth_normalized, heat_map # 示例调用 gray_depth, color_heatmap = predict_depth("input.jpg") cv2.imwrite("depth_gray.png", gray_depth) cv2.imwrite("depth_inferno.png", color_heatmap)