【unet语义分割训练自己的数据集---失败】

最新推荐文章于 2023-10-26 21:10:56 发布

蛮有用

最新推荐文章于 2023-10-26 21:10:56 发布

阅读量713

点赞数

文章标签：深度学习机器学习神经网络

本文链接：https://blog.csdn.net/yym33/article/details/125116759

版权

复现的Unet网络，得出的测试图是这样的：

在这里插入图片描述

而当我换成自己的数据集，训练完成后确是这样。

测试原图：
请添加图片描述

测试分割出的结果图：
在这里插入图片描述
以下是代码：train.py的参数设置


def train_net(net,
              device,
              epochs=10,
              batch_size=1,
              lr=1e-4,
              val_percent=0.1,
              save_cp=True,
              img_scale=0.5):

    logging.info(f'''Starting training:
        Epochs:          {epochs}
        Batch size:      {batch_size}
        Learning rate:   {lr}
        Training size:   {n_train}
        Validation size: {n_val}
        Checkpoints:     {save_cp}
        Device:          {device.type}
        Images scaling:  {img_scale}
    ''')

    optimizer = optim.RMSprop(net.parameters(), lr=1e-4, weight_decay=1e-8)