深度学习
文章平均质量分 69
深度学习
xiandong20
这个作者很懒,什么都没留下…
展开
-
U-Net入门(一)构建一个简单的U-Net网络
【代码】U-Net入门(一)构建一个简单的U-Net网络。原创 2024-06-13 17:38:04 · 212 阅读 · 0 评论 -
编写一个简单的UNet图像分割网络_(四)测试模块
加载网络-加载权重-把网络挂载在设备上-输入图片-图片处理-把图片输入到网络-保存输出图片。当然,这只是一个很简单的图像分割训练,可以适当休息,但不要骄傲自满。代码仍然有很多可以优化的地方,继续大步的往前走吧!对应步骤都有注释标注。原创 2024-06-17 00:46:03 · 124 阅读 · 0 评论 -
编写一个简单的UNet图像分割网络_(三)训练模块
首先我们要明确我们要用于训练的设备是什么,是gpu还是cpu,当前使用cuda训练的技术较为成熟,教程较多,但如果确实没有cuda,也没必要强行给自己创造需求,现在才刚开始学习,没必要几千上万块买一张卡,使用cpu训练一些小模型小数据集也是可以的,或者上云平台。训练前的准备工作,我们需要确定训练设备,加载数据集路径,加载权重文件(权重文件就是别人从0训练调试出来的一个模型,我们基于人家调试出来的模型进行训练,训练效果会更好,但是需要确保权重文件的网络结构匹配,网络结构不匹配权重文件不通用)原创 2024-06-17 00:45:14 · 501 阅读 · 0 评论 -
编写一个简单的UNet图像分割网络_(二)网络结构搭建
当前这个unet网络loss在收敛到0.1左右时,仍会上下剧烈波动,如有大佬知道是啥情况的话,有空的话烦请指点一二。此处使用的网络结构搭建比较基础,不做过多说明,说明都写在注释里,方便一行一行对照查看。示例,loss不能正确收敛,如有大佬知道是啥情况的话,有空的话烦请指点一二。原创 2024-06-17 00:44:30 · 323 阅读 · 0 评论 -
编写一个简单的UNet图像分割网络_(一)数据预处理
编写数据预处理则要有数据,如果你拿到的标签格式不是.png,而是.xml或者.seg或者更多,请具体查阅关于格式转换的代码或工具,如果确实是初学者,建议别多折腾,去找找标签是.png的数据集。我们的数据集中的数据往往是成千上万条的,不可能一条一条给到程序中,所以写的路径一般只写到程序所在的文件夹, 其中的数据交给程序自己去遍历,用到os.listdir()遍历。此处我给他提供的path就有可能是D:\datasets,当然,你的路径肯定比这个长的多,仅作举例方便理解,但结尾一定是整个数据集的一级文件夹名。原创 2024-06-17 00:43:35 · 421 阅读 · 0 评论