github:Machine_Learning/网络模型/UNet at main · golitter/Machine_Learning (github.com)
因为VOC数据集太大,上传github很慢,所以就没有上传VOC数据,只有参考的目录位置。
数据集自行下载:https://host.robots.ox.ac.uk/pascal/VOC/voc2007/
该模型是在B站学习敲出来的,B站出处:UNet测试及代码实现_哔哩哔哩_bilibili
UP主github仓库:qiaofengsheng/pytorch-UNet: pytorch搭建自己的unet网络,训练自己的数据集。 (github.com)
个人总结: