[人工智能]基于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把代码下载下来。

了解本专栏 订阅专栏 解锁全文 超级会员免费看