基于C++实现对UNet图像分割的部署
本博文利用UNet介绍图像分割的部署,重点介绍采用warpaffine对图像进行预处理,以及后处理decode部分的代码。
1. Unet网络介绍
UNet它是一个encoder - decoder
的结构,那么encoder对应就是这个u
型网络左边这半部分,也就是我们特征提取及下采样这部分。decoder解码就是右边部分通过一系列上采样,我们得到最终的一个分割图。 对于网络结构左边这一侧,作者叫做contracting path
,对于右边这一侧作者称为expansive path
。详情请参考博客:图像分割UNet (1) : 网络结构讲解
1.1 unet 代码的下载
本项目使用的unet pytorch
代码见:https://github.com/bubbliiiing/unet-pytorch,利用git clone
把代码下载下来。