基于C++实现对UNet图像分割的部署

本博文利用UNet介绍图像分割的部署,重点介绍采用warpaffine对图像进行预处理,以及后处理decode部分的代码。

1. Unet网络介绍

基于C++实现对UNet图像分割的部署_第1张图片
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把代码下载下来。
基于C++实现对UNet图像分割的部署_第2张图片

你可能感兴趣的:(object,detection,c++,深度学习,人工智能)