直接上效果图
首先需要搜集n张路面缺陷的图片,然后使用labelme进行打标签。
沿着轮廓描一遍,打上标签
这里以unet图像分割网络为例,主干网络是vgg,如果有想通过加注意力机制优化网络的,可以尝试这样添加。
import torch
import torch.nn as nn
from torchvision.m
直接上效果图
首先需要搜集n张路面缺陷的图片,然后使用labelme进行打标签。
沿着轮廓描一遍,打上标签
这里以unet图像分割网络为例,主干网络是vgg,如果有想通过加注意力机制优化网络的,可以尝试这样添加。
import torch
import torch.nn as nn
from torchvision.m