科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种基于稠密连接的废旧塑料图像分割方法,方法包括以下步骤：采集废旧塑料图像,得到采集图像,将采集图像划分为验证集和训练集,利用图像标注工具对采集图像进行语义标签标注；进行局部数据增广,对增广后的局部图像进行预处理,将局部图像与对应的语义标签转化为张量,对图像张量进行正则化；模型训练,将训练数据输入基于稠密连接的语义分割模型,利用带聚焦参数和平衡因子的交叉熵损失函数评估训练结果,利用Adam算法进行反向传播更新模型参数；循环训练模型步骤,直至达到停止条件,保存模型以及模型参数。本发明充分利用了稠密连接的特征复用,有效增强算法的性能与效率,实现废旧塑料图像精确高效的端对端语义分割。
- 单位
