图像分割分析效果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

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

相关推荐
rocksun几秒前
OneUptime MCP服务器:AI原生可观测性融入你的工作流程
人工智能·监控
禺垣几秒前
支持向量机(SVM)分类
机器学习
禺垣2 分钟前
协同过滤推荐算法
机器学习
weisian1519 分钟前
人工智能-基础篇-10-什么是卷积神经网络CNN(网格状数据处理:输入层,卷积层,激活函数,池化层,全连接层,输出层等)
人工智能·神经网络·cnn
这里有鱼汤21 分钟前
90%的人都会搞错的XGBoost预测逻辑,未来到底怎么预测才对?
后端·机器学习
静心问道33 分钟前
SELF-INSTRUCT:使用自生成指令对齐语言模型
人工智能·语言模型·大模型
芷栀夏39 分钟前
基于Anything LLM的本地知识库系统远程访问实现路径
数据库·人工智能
AI生存日记39 分钟前
AI 日报:阿里、字节等企业密集发布新技术,覆盖语音、图像与药物研发等领域
人工智能·华为云·语音识别·open ai大模型
小庞在加油39 分钟前
《dlib库中的聚类》算法详解:从原理到实践
c++·算法·机器学习·数据挖掘·聚类
hjs_deeplearning1 小时前
认知篇#10:何为分布式与多智能体?二者联系?
人工智能·分布式·深度学习·学习·agent·智能体