直接上效果图
首先需要搜集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.Adapti