图像分割分析效果2

这次加了结构化损失

训练集dice: 0.9219 - iou: 0.8611 - loss: 0.0318 - mae: 0.0220 - total: 0.8915

dropout后:dice: 0.9143 - iou: 0.8488 - loss: 0.0335 - mae: 0.0236 - total: 0.8816

加了结构化损失后:avg_score: 0.8917 - dice: 0.9228 - iou: 0.8627 - loss: 0.1098 - mae: 0.0191

其实效果有提升,之前自定义的指标有问题,它返回的是最后一个批次的统计信息,现在返回的是整个数据集的平均信息

model.evaluate(train_dataset)

验证集504个样本,dice: 0.8096 - iou: 0.7138 - loss: 0.1267 - mae: 0.0429 - total: 0.7617

dice: 0.8113 - iou: 0.7135 - loss: 0.1024 - mae: 0.0409 - total: 0.7624

加了结构化损失后:avg_score: 0.7687 - dice: 0.8189 - iou: 0.7242 - loss: 0.2894 - mae: 0.0393

len(val_dataset)*BATCH_SIZE,验证集

经过最后的一番优化后

模型在训练集上的表现,因为验证集和训练集不同,模型并没有训练验证集上数据

训练集dice: 0.9219 - iou: 0.8611 - loss: 0.0318 - mae: 0.0220 - total: 0.8915

dropout后:dice: 0.9143 - iou: 0.8488 - loss: 0.0335 - mae: 0.0236 - total: 0.8816

加了结构化损失后:avg_score: 0.8917 - dice: 0.9228 - iou: 0.8627 - loss: 0.1098 - mae: 0.0191

其实效果有提升,之前自定义的指标有问题,它返回的是最后一个批次的统计信息,现在返回的是整个数据集的平均信息

稍微修改了dropout后:avg_score: 0.9160 - dice: 0.9411 - iou: 0.8921 - loss: 0.2218 - mae: 0.0140

model.evaluate(train_dataset)

模型在训练集上的表现可以说相当好,再来看模型在验证集上的表现,模型从未拟合过验证集数据因为我不是用的随机拆分,也没重启内核

验证集504个样本,dice: 0.8096 - iou: 0.7138 - loss: 0.1267 - mae: 0.0429 - total: 0.7617

dice: 0.8113 - iou: 0.7135 - loss: 0.1024 - mae: 0.0409 - total: 0.7624

加了结构化损失后:avg_score: 0.7687 - dice: 0.8189 - iou: 0.7242 - loss: 0.2894 - mae: 0.0393

稍微修改了dropout后:avg_score: 0.7939 - dice: 0.8403 - iou: 0.7524 - loss: 0.5245 - mae: 0.0322

len(val_dataset)*BATCH_SIZE

虽然这张图片真实掩码有半截身子的人 ,但是我还是觉得模型预测的很正确,半截身子的人不能当前景,应该只有没被裁剪的人当前景,因为这个数据集中很多裁剪的人当背景的

相关推荐
小蒋观天下6 小时前
雷击与浪涌防护:前端SPD选型、接地网误区,接地不当反而引入干扰
深度学习·算法·机器学习
云和数据.ChenGuang6 小时前
Python 单继承与多继承深度剖析:从 MRO 到实战取舍
开发语言·人工智能·python·langchain
老金带你玩AI6 小时前
OpenAI先砍额度,又送额度,老用户看懵了
人工智能
AI小云7 小时前
从CRUD到LLM Engineering:一个后端工程师的AI转型技术栈梳理
人工智能·ai培训·ai培训机构·程序员转行
黄啊码7 小时前
【黄啊码】终于不用死磕提示词了!做了一个所见即所得的动作与运镜工具,现已开源
人工智能
阿狸猿7 小时前
MLOps 机器学习流水线架构设计
人工智能·机器学习
虎鲸不是鱼7 小时前
AI 自动化逆向(三十三)换一个「驱动器」,覆盖率跳了一个量级
人工智能·自动化·大模型·逆向·ai逆向
后端小肥肠7 小时前
群友靠公众号漫画工坊入池!10w+IP漫画一键复刻,新增矩阵功能
人工智能·aigc·agent
2601_965619697 小时前
AI视频制作:做历史战争视频,兵力部署的老地图难找,AI怎么自动配战争地图?
人工智能·音视频
IT_陈寒7 小时前
Vue的v-for里用了个临时变量,页面直接挂了
前端·人工智能·后端