目录
1. 文件目录介绍
完整下载路径:unet对DRIVE数据集的完整项目
文件夹信息如下所示:
剩下的就是常用的py文件
- dataset 加载DRIVE数据集
- model 存放unet模型
- predict 根据最好的权重对图片进行预测
- train 网络训练的主函数
- utils 需要用到的工具函数
2. tools 结构
tools 里面存放三个py文件:计算训练集的mean和std、混淆矩阵的实现、自定义预处理的实现
因为防止utils的代码太多冗余,所以将其他的类定义在tools里面,然后utils调用tools里面的函数