深度学习 Unet+Resnet101 实战分割项目、多尺度训练、迁移学习、多类别分割:腹部多脏器5类别分割数据集下载

weixin_39820780 2024-02-23 22:30:16
本项目为 Unet+Resnet101 多尺度分割实战项目(包含数据集),unet的backbone更换为resnet101 数据集采用腹部多脏器5类别分割数据集 项目介绍:总大小202MB 1.train 脚本会自动训练,代码会自动将数据随机缩放为设定尺寸的0.5-1.5倍之间,实现多尺度训练。为了实现多分割项目,utils中的compute_gray函数会将mask灰度值保存在txt文本,并且自动为UNET网络定义输出的channel 2.项目的预处理函数全部重新实现,可以在transforms.py自行查看。 3.网络训练了100个epochs,miou达到0.9左右,学习率采用cos衰减,训练集和测试集的损失和iou曲线可以在run_results文件内查看,图像由matplotlib库绘制。除此外,还保存了训练日志,最好权重等,在训练日志可以看到每个类别的iou、recall、precision以及全局像素点的准确率等等 4.预测脚本可以自动推理inference下所有图片 代码做了注释,自行下载查看,想要训练自己的数据,参考README文件,傻瓜式运行 , 相关下载链接:https://download.csdn.net/download/qq_44886601/88821127?utm_source=bbsseo
...全文
26 回复 打赏 收藏 转发到动态 举报
写回复
用AI写文章
回复
切换为时间正序
请发表友善的回复…
发表回复

13,654

社区成员

发帖
与我相关
我的任务
社区描述
CSDN 下载资源悬赏专区
其他 技术论坛(原bbs)
社区管理员
  • 下载资源悬赏专区社区
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧