02 2021 档案
摘要:基于 "LibTorch (Pytorch C++)" 的C++开源图像分割神经网络库. 分享一个C++的图像分割开源库 "LibtorchSegmentation" ,支持C++训练分割模型,可以训练自己的数据集。支持FPN,UNet,PAN,LinkNet,DeepLabV3和DeepLabV3
阅读全文
摘要:本章 简要 介绍如何如何用C++实现一个语义分割器模型,该模型具有训练和预测的功能。本文的分割模型架构使用简单的U Net结构,代码结构参考了 "qubvel segmentation" 中的U Net部分,该项目简称SMP,是基于pytorch实现的开源语义分割项目。本文分享的c++模型几乎完美复
阅读全文
摘要:前面的章节中我们介绍了libtorch的环境搭建(VS和Qt),libtorch张量常用操作,简单的MLP,CNN和LSTM模型搭建,以及数据加载类的使用。本章将以图像分类任务为例,详细介绍如何使用c++训练一个图片分类器。 模型 本文以VGG为例,对比pytorch下的模型搭建和训练,阐述Libt
阅读全文