[教学]路面裂缝检测_瑕疵检测等图像分割

直接上效果图

 

首先需要搜集n张路面缺陷的图片，然后使用labelme进行打标签。

 沿着轮廓描一遍，打上标签

这里以unet图像分割网络为例，主干网络是vgg，如果有想通过加注意力机制优化网络的，可以尝试这样添加。

import torch
import torch.nn as nn
from torchvision.models.utils import load_state_dict_from_url


from torch.nn import init
from collections import OrderedDict


class ECAAttention(nn.Module):

    def __init__(self, kernel_size=3):
        super().__init__()
        self.gap = nn.AdaptiveAvgPool2d(1)
        self.co