直接看运行效果:
代码实现:
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, epo