图像分割
文章平均质量分 94
姜饼饼
这个作者很懒,什么都没留下…
展开
-
PyTorch图像分割模型——多类别图像分割数据集制作
上一篇文章中,主要介绍了使用segmentation_models_pytorch库进行UNet++模型训练,我们使用的数据集是已经标注好的CamVid数据集,但在实际应用中,我们需要标注自己的多分类图像分割数据集,这篇文章中,就重点介绍下怎么创建自己的图像分割数据集。首先需强调的是在这里我们用的数据集都是png格式的,生成的标注图像也都是png格式的,因为png图像可以做到无损压缩,能在保证最不失真的情况下尽可能压缩图像文件的大小,而且png用来存储灰度图像时,灰度图像的深度可多到16位,存储彩色图像时原创 2021-08-10 16:11:11 · 8861 阅读 · 3 评论 -
PyTorch图像分割模型——segmentation_models_pytorch库的使用
一、概要segmentation_models_pytorch是一个基于PyTorch的图像分割神经网络这个新集合由俄罗斯的程序员小哥Pavel Yakubovskiy一手打造,对于图像分割而言简直就是神器般的存在。github地址:https://github.com/qubvel/segmentation_models.pytorch该库的主要功能有:高级API(只需两行即可创建神经网络);用于二分类和多类分割的9种模型架构(包括传奇的Unet)(Unet、Unet++、MAnet、Lin原创 2021-08-10 01:39:19 · 36019 阅读 · 27 评论