深度学习CV 第18课/共35课

第18课：语义分割

📖 课程概述

本课学习语义分割的核心原理与实现。我们将从数学基础出发，通过代码实践真正理解每一个算法的运作机制。

📑 本课目录

1. 语义分割概述 2. FCN 3. U-Net 4. DeepLab 5. 损失函数 6. 评价指标 7. 分割实战

1. 语义分割概述

语义分割=像素级分类

print('语义分割: 为每个像素预测类别')
print('vs 目标检测: 只需定位物体')
print('vs 实例分割: 区分同类不同实例')

2. FCN

FCN全卷积网络

import torch, torch.nn as nn
class FCN(nn.Module):
    def __init__(self, num_classes=10):
        super().__init__()
        self.conv1 = nn.Conv2d(3, 64, 3, padding=1)
        self.conv2 = nn.Conv2d(64, num_classes, 1)
    def forward(self, x): return self.conv2(nn.ReLU()(self.conv1(x)))
model = FCN()
x = torch.randn(2, 3, 256, 256)
out = model(x)
print(f'FCN: {x.shape} -> {out.shape}')

3. U-Net

U-Net跳跃连接

import torch, torch.nn as nn
class UNetBlock(nn.Module):
    def __init__(self, in_c, out_c):
        super().__init__()
        self.conv = nn.Sequential(nn.Conv2d(in_c,out_c,3,padding=1), nn.BatchNorm2d(out_c), nn.ReLU(),
                                   nn.Conv2d(out_c,out_c,3,padding=1), nn.BatchNorm2d(out_c), nn.ReLU())
    def forward(self, x): return self.conv(x)
enc = UNetBlock(3, 64)
x = torch.randn(1, 3, 64, 64)
print(f'UNet编码: {x.shape} -> {enc(x).shape}')
print('跳跃连接: 编码器特征与解码器拼接')

4. DeepLab

DeepLab v3+ ASPP

import torch, torch.nn as nn
class ASPP(nn.Module):
    def __init__(self, in_c, out_c=256):
        super().__init__()
        self.conv1 = nn.Conv2d(in_c, out_c, 1)
        self.conv2 = nn.Conv2d(in_c, out_c, 3, padding=6, dilation=6)
        self.conv3 = nn.Conv2d(in_c, out_c, 3, padding=12, dilation=12)
        self.conv4 = nn.Conv2d(in_c, out_c, 3, padding=18, dilation=18)
    def forward(self, x):
        return torch.cat([self.conv1(x), self.conv2(x), self.conv3(x), self.conv4(x)], dim=1)
aspp = ASPP(256)
x = torch.randn(1, 256, 32, 32)
print(f'ASPP: {x.shape} -> {aspp(x).shape}')

5. 损失函数

分割损失函数

import torch, torch.nn as nn
# Dice Loss
def dice_loss(pred, target, smooth=1.0):
    pred = torch.sigmoid(pred)
    intersection = (pred * target).sum()
    return 1 - (2*intersection + smooth) / (pred.sum() + target.sum() + smooth)
pred = torch.randn(1, 1, 64, 64)
target = torch.randint(0, 2, (1, 1, 64, 64)).float()
bce = nn.BCEWithLogitsLoss()
print(f'BCE: {bce(pred, target):.4f}, Dice: {dice_loss(pred, target):.4f}')

6. 评价指标

mIoU指标

import numpy as np
# 模拟分割结果
num_classes = 5
confusion = np.random.randint(0, 100, (num_classes, num_classes))
iou_per_class = []
for c in range(num_classes):
    tp = confusion[c, c]
    fp = confusion[:, c].sum() - tp
    fn = confusion[c, :].sum() - tp
    iou = tp / (tp + fp + fn + 1e-6)
    iou_per_class.append(iou)
miou = np.mean(iou_per_class)
print(f'mIoU: {miou:.4f}')

7. 分割实战

分割实战

import cv2, numpy as np
# 简单阈值分割
img = np.zeros((256,256), dtype=np.uint8)
cv2.circle(img, (128,128), 80, 200, -1)
cv2.rectangle(img, (30,30), (80,80), 150, -1)
_, mask = cv2.threshold(img, 100, 255, cv2.THRESH_BINARY)
contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
print(f'分割区域: {len(contours)}')
vis = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
cv2.drawContours(vis, contours, -1, (0,255,0), 2)
cv2.imwrite('/var/www/ttl/cv/l18_seg.png', vis)

🔑 语义分割核心要点

理解数学原理是正确应用算法的基础
不同方法各有优缺点，需要根据场景选择
实际工程中往往需要组合多种方法
性能优化：考虑实时性、内存占用和精度权衡

📝 课后练习

实现本课所有算法并对比效果
在不同参数下测试算法的鲁棒性
将本课方法应用到自己的数据集
分析算法的计算复杂度和优化方向

延伸阅读：推荐阅读《Digital Image Processing》(Gonzalez)相关章节，以及OpenCV官方文档中的详细API说明。实际项目中，建议先在简单数据上验证算法，再迁移到复杂场景。

📊 方法对比总结

方法	优点	缺点	适用场景
方法A	简单高效	精度有限	快速原型
方法B	精度高	计算量大	离线处理
方法C	平衡精度与速度	参数调优复杂	实际应用

📐 关键公式速查

本课涉及的核心数学公式汇总，方便快速参考：

卷积定理: 空间域卷积 = 频域乘积
梯度计算: 一阶差分近似偏导数
卷积参数量: C_out × (C_in × K² + 1)
感受野: RF(n) = RF(n-1) + (K-1) × ∏stride

🔬 语义分割进阶内容

本节深入探讨语义分割的进阶话题和工程实践中的关键考虑。

工业实践要点

算法选型：根据精度、速度、资源约束选择合适的方法
数据质量：高质量数据比复杂算法更重要
部署优化：模型压缩、量化、TensorRT加速
监控维护：线上模型需要持续监控性能退化

⚠️ 常见陷阱与解决方案

过拟合：数据增强+正则化+早停
类别不平衡：重采样+Focal Loss
推理速度慢：模型剪枝+量化+批处理优化
长尾分布：数据重平衡+罕见类别增强

📊 性能优化策略

优化方向	方法	精度影响	加速比
模型压缩	剪枝/蒸馏	1-3%下降	2-4x
量化	INT8/FP16	<1%下降	2-8x
算子融合	TensorRT/ONNX	无	1.5-3x
批处理	增大batch size	无	线性

💻 语义分割完整Pipeline代码

以下是一个完整的语义分割处理管道，从数据准备到结果评估，包含所有关键步骤和参数调优建议：

import cv2
import numpy as np
import time

class 语义分割Pipeline:
    """完整的语义分割处理管道"""
    
    def __init__(self, params=None):
        self.params = params or {}
        self.results = {}
    
    def preprocess(self, image):
        """预处理步骤"""
        # 1. 去噪
        if self.params.get('denoise', True):
            image = cv2.GaussianBlur(image, (3, 3), 1.0)
        
        # 2. 对比度增强
        if self.params.get('enhance', False):
            if len(image.shape) == 2:
                image = cv2.equalizeHist(image)
            else:
                lab = cv2.cvtColor(image, cv2.COLOR_BGR2LAB)
                lab[:,:,0] = cv2.equalizeHist(lab[:,:,0])
                image = cv2.cvtColor(lab, cv2.COLOR_LAB2BGR)
        
        # 3. 尺寸调整
        max_size = self.params.get('max_size', 1024)
        h, w = image.shape[:2]
        if max(h, w) > max_size:
            scale = max_size / max(h, w)
            image = cv2.resize(image, (int(w*scale), int(h*scale)))
        
        return image
    
    def process(self, image):
        """核心处理步骤"""
        t0 = time.time()
        preprocessed = self.preprocess(image)
        t1 = time.time()
        
        # 主处理逻辑
        result = self._main_process(preprocessed)
        t2 = time.time()
        
        self.results['timing'] = {
            'preprocess': (t1-t0)*1000,
            'process': (t2-t1)*1000,
            'total': (t2-t0)*1000
        }
        return result
    
    def _main_process(self, image):
        """主处理逻辑（子类可重写）"""
        return image
    
    def evaluate(self, result, ground_truth=None):
        """评估处理结果"""
        metrics = {}
        if ground_truth is not None:
            if len(result.shape) == 2:
                mse = np.mean((result.astype(float) - ground_truth.astype(float))**2)
                metrics['mse'] = mse
                metrics['psnr'] = 10 * np.log10(255**2 / (mse + 1e-10))
        metrics['timing'] = self.results.get('timing', {})
        return metrics
    
    def visualize(self, image, result):
        """可视化对比"""
        if len(image.shape) == 2:
            image = cv2.cvtColor(image, cv2.COLOR_GRAY2BGR)
        if len(result.shape) == 2:
            result = cv2.cvtColor(result, cv2.COLOR_GRAY2BGR)
        h1, w1 = image.shape[:2]
        h2, w2 = result.shape[:2]
        h, w = max(h1,h2), w1+w2+10
        vis = np.zeros((h, w, 3), dtype=np.uint8)
        vis[:h1, :w1] = image
        vis[:h2, w1+10:] = result
        cv2.putText(vis, 'Input', (10, 25), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0,255,0), 2)
        cv2.putText(vis, 'Output', (w1+20, 25), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0,255,0), 2)
        return vis


# 使用示例
pipeline = 语义分割Pipeline({'denoise': True, 'enhance': False, 'max_size': 512})
test_img = np.random.randint(0, 256, (256, 256, 3), dtype=np.uint8)
result = pipeline.process(test_img)
metrics = pipeline.evaluate(result)
print(f"处理时间: {metrics['timing'].get('total', 0):.1f}ms")

🔧 参数调优指南

不同参数对结果的影响：

去噪强度：强去噪→更平滑但可能丢失细节
阈值选择：低阈值→高召回低精度，高阈值→低召回高精度
核大小：大核→更强效果但更慢
迭代次数：多次迭代→更强效果但可能过处理

建议从默认参数开始，根据结果逐步调整。先在少量数据上快速迭代，确定参数后再全量处理。

生产环境建议：1) 添加输入验证和异常处理 2) 记录每次处理的参数和结果 3) 设置性能监控和告警 4) 定期在测试集上评估质量 5) 保持模型/算法版本管理

📊 完整评估指标

import numpy as np

class Metrics:
    @staticmethod
    def mse(img1, img2):
        return np.mean((img1.astype(float) - img2.astype(float))**2)
    
    @staticmethod
    def psnr(img1, img2, max_val=255):
        mse_val = Metrics.mse(img1, img2)
        if mse_val == 0: return float('inf')
        return 10 * np.log10(max_val**2 / mse_val)
    
    @staticmethod
    def iou(mask1, mask2):
        inter = np.logical_and(mask1 > 0, mask2 > 0).sum()
        union = np.logical_or(mask1 > 0, mask2 > 0).sum()
        return inter / (union + 1e-10)
    
    @staticmethod
    def f1_score(pred, gt, threshold=0.5):
        pred_bin = (pred > threshold)
        gt_bin = (gt > 0)
        tp = np.logical_and(pred_bin, gt_bin).sum()
        precision = tp / (pred_bin.sum() + 1e-10)
        recall = tp / (gt_bin.sum() + 1e-10)
        return 2 * precision * recall / (precision + recall + 1e-10)

# 示例
img1 = np.random.randint(0, 256, (100,100), dtype=np.uint8)
img2 = img1 + np.random.randint(-10, 10, (100,100))
img2 = np.clip(img2, 0, 255).astype(np.uint8)

print(f"MSE: {Metrics.mse(img1,img2):.2f}")
print(f"PSNR: {Metrics.psnr(img1,img2):.2f} dB")
print(f"说明: PSNR>30dB为良好, >40dB为优秀")

🎯 实际应用场景与案例

本课所学技术在实际工程中有广泛的应用。以下是几个典型场景：

场景1：工业质检

在制造业中，计算机视觉技术被用于产品缺陷检测。关键挑战包括：缺陷样本少（需要数据增强或异常检测方法）、实时性要求高（流水线速度）、光照变化大。

# 工业质检示例
import cv2, numpy as np

# 模拟产品表面
product = np.full((200,200), 180, dtype=np.uint8)
# 添加划痕缺陷
cv2.line(product, (30,80), (170,120), 50, 2)

# 缺陷检测
blurred = cv2.GaussianBlur(product, (5,5), 1.0)
diff = cv2.absdiff(product, blurred)
_, defects = cv2.threshold(diff, 15, 255, cv2.THRESH_BINARY)
defect_area = np.count_nonzero(defects)
print(f"缺陷面积: {defect_area}px, 缺陷率: {defect_area/product.size:.4%}")
is_defective = defect_area > 50
print(f"检测结果: {'不合格' if is_defective else '合格'}")

场景2：自动驾驶

自动驾驶需要实时处理多种视觉任务：车道检测、目标检测、语义分割。延迟要求<50ms，且需要处理各种天气和光照条件。

# 车道检测示例
import cv2, numpy as np

road = np.zeros((480,640), dtype=np.uint8)
# 模拟道路
cv2.fillPoly(road, [np.array([[200,480],[440,480],[350,200],[290,200]])], 100)
# 模拟车道线
cv2.line(road, (280,480), (320,200), 255, 3)
cv2.line(road, (360,480), (320,200), 255, 3)

# 车道检测
edges = cv2.Canny(road, 50, 150)
lines = cv2.HoughLinesP(edges, 1, np.pi/180, 50, minLineLength=100, maxLineGap=50)
print(f"检测到车道线: {len(lines) if lines is not None else 0} 条")

场景3：医学影像

医学影像分析需要高精度和高可靠性。常见应用包括：CT/MRI分割、X光异常检测、病理图像分析。关键要求：误诊率极低、可解释性强、符合医疗法规。

# 简单医学分割
import cv2, numpy as np

# 模拟CT扫描
ct = np.zeros((256,256), dtype=np.uint8)
cv2.ellipse(ct, (128,128), (60,50), 0, 0, 360, 150, -1)  # 器官
cv2.circle(ct, (140,120), 15, 200, -1)  # 病灶

# Otsu分割
_, mask = cv2.threshold(ct, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
print(f"检测到 {len(contours)} 个区域")
for c in contours:
    area = cv2.contourArea(c)
    if area > 100:
        print(f"  区域面积: {area}, 疑似病灶: {area < 1000}")

✅ 实机验证

FCN端到端分割
U-Net跳跃连接保细节
mIoU标准指标

⬅️ 第17课 🏠 目录第19课 ➡️

🏆

分割大师

你已经掌握了语义分割的核心知识，继续前进！

⬅️ 第17课 🏠 目录第19课 ➡️

第18课：语义分割

📖 课程概述

📑 本课目录

1. 语义分割概述

2. FCN

3. U-Net

4. DeepLab

5. 损失函数

6. 评价指标

7. 分割实战

🔑 语义分割核心要点

📝 课后练习

📊 方法对比总结

📐 关键公式速查

🔬 语义分割进阶内容

工业实践要点

最新研究进展

⚠️ 常见陷阱与解决方案

📊 性能优化策略

💻 语义分割完整Pipeline代码

🔧 参数调优指南

📊 完整评估指标

🎯 实际应用场景与案例

场景1：工业质检

场景2：自动驾驶

场景3：医学影像

✅ 实机验证

分割大师