python裂缝检测_小黑的Python日记：Unet简单实现裂缝分割

大噶好，我系小黑喵

裂缝数据集

--project

main.py

--image

--train

--data

--groundTruth

--val

--data

--groundTruth

我手动将数据集做成这个格式，其中trian84张，val34张，都保存为了jpg图像。

Unet

Unet结构

需要修改dataset.py为自己的数据集，其他小小改动即可。

#dataset.py

import torch.utils.data as data

import PIL.Image as Image

import os

def make_dataset(rootdata,roottarget):#获取img和mask的地址

imgs = []

filename_data = [x for x in os.listdir(rootdata)]

for name in filename_data:

img = os.path.join(rootdata, name)

mask = os.path.join(roottarget, name)

imgs.append((img, mask))#作为元组返回

return imgs

class MyDataset(data.Dataset):

def __init__(sel