1.1语义分割基础

1.1.1概念

简而言之,是将一张(含有多种物体的)图片,识别出其中一种物体。

最简单的语义分割,当属于二分类,即识别出图片中的两种地物,一种归为背景,另一种归为想要提取的物体。

将原图放入模型后,产生预测图,预测图宇实现已有的标签计算loss,调整参数继续循环这个过程,直至loss函数最小,即参数最优为止。

深入理解:

图像shape的变化:

编码:图像(size)不断变小,同时不断变厚(通道数增加)

解码:图像(size)不断变大(最终恢复到原图的1/2),同时不断变薄(通道数减小)

1.2代码部分

1.2.1训练流程

通过gitup下在完整代码https://github.com/bubbliiiing/Semantic-Segmentation
数据集的链接为:

通过修改报错进行训练

1.2.2训练结果

原图:
在这里插入图片描述
处理后:
在这里插入图片描述

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐