直接看运行效果:
代码实现:
from model.UNet_model import UNet
from UNet_dataset import ISBI_Loader
from torch import optim
import torch.nn as nn
import torch
# 基于UNet模型的细胞图像分割 细胞分割 完整数据+代码 可直接运行
# 运行 我提前运行好了
def train_net(net, device, data_path, epochs=40, batch_size=2, lr=0.00001):
# 加载训练集
isbi_dataset = ISBI_Loader(data_path)
train_loader = torch.utils.data.DataLoader(dataset=isbi_dataset,