keras UNET视网膜血管分割调试记录

本文记录了使用Keras UNET在视网膜血管分割任务中的调试过程,包括环境配置、代码修改和解决的错误。在训练过程中遇到过拟合问题,通过调整数据预处理策略来缓解。在调试代码时,解决了Tensorflow 2.x与Keras的兼容性问题,以及文件路径、sklearn.metrics导入和模型加载等方面的问题。
摘要由CSDN通过智能技术生成

随随便便的废话

跑自己的代码的时候发现CHASE_DB1数据集总是过拟合,即使是做了数据增强将图像增加到600+,验证集准确率依旧达到0.99,怀疑是在数据预处理的时候简单的参考别的开源代码将图像简单的划分为无重叠的64*64 patch导致类不平衡,遂找到当年因为硬件查没跑通的代码,开始调试,学习如何在训练集上随机提取patch进行训练。
调试的代码来自这位大佬

环境

老板大手一挥给配了个3080,当初配置环境的时候简直是摸着石头过河,都说新版本容易碰壁,原先想要直接用上以前tensorflow1.6+keras的代码,就弄了个CUDA9,结果配的天昏地暗,还是报错。想过在虚拟环境中配置CUDA,也不行。问了同学,同学答:我觉得pytorch不错,你可以试试。 最后老老实实卸了重装,按照网上不多的教程配置好了3080能运行的环境。

  1. GPU RTX 3080,cuda 11;
  2. tensorflow==2.5.0-dev;
  3. keras==2.3.0;
  4. opencv-python==4.4.0.46;
  5. scikit-learn==0.23.2;
  6. pillow==8.0.1;
  7. h5py==2.10.0;
  8. configparser==3.5.0

调试记录

1) Cannot Run Git & Invalid VCS root mapping

和github相关的报错,可以按照网上下载并配置Git,不过我一直都倾向于自己新建工程,复制粘贴。新建工程后不存在这个问题。

1.retinaNN_training.py

2)
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值