PyTorch:基于UNet和camvid数据集的道路分割

背景 语义分割是深度学习中的一个非常重要的研究方向,并且UNet是语义分割中一个非常经典的模型。在本次博客中,我尝试用UNet对camvid dataset数据集进行道路分割,大致期望的效果如下: 原图 道路分割效果 本博客的代码参考了以下链接: https://github.com/milesial/Pytorch-UNet https://github.com/qubvel/segmentation_models.pytorch 数据集介绍及处理 之前的博客里,我几乎不怎么介绍数据集,因为用到的数据集比较简单;但是在使用camvid dataset的时候,我脑袋都大了,用了两三个小