语义分割
游虾十八弯
吊袜带
展开
-
语义分割之copy_paste数据增强
记录一种另类的语义分割增强算法,copy and paste,简而言之,就是把A图的mask扣下来放到B图中,需要A图的原图,A图的8位mask,B图的原图,B图的8位mask。不过很可惜,在我这里尝试了以后,感觉造出来的数据极度不真实,后来也就没用了,仅仅记录一下,代码如下。原创 2023-04-19 15:32:58 · 535 阅读 · 0 评论 -
语义分割宝库之segmentation_models_pytorch
github地址:https://github.com/qubvel/segmentation_models.pytorch。这个集合由俄罗斯的程序员小哥Pavel Yakubovskiy一手打造,对于图像分割而言简直就是神器般的存在。segmentation_models_pytorch是一个基于PyTorch的。二、训练自己的多类别语义分割模型。原创 2023-04-17 22:22:20 · 624 阅读 · 0 评论 -
语义分割中最容易出错的图片格式转换
语义分割中经常会遇到图片格式转换的事情,labelme的json转换出来的mask是16位,训练的时候需要的mask一般是8位,如果要做离线数据增强的话,mask也要做增强,那么mask需要转换为24位,增强完毕以后,24位要转换为8位,以下三个脚本记录这些转换。原创 2023-04-17 16:52:48 · 199 阅读 · 0 评论 -
语义分割前奏之数据增强
我采用albumentations库进行数据增强,有些增强方式需要对label也做相应的增强,有些方式不需要,所以分开来做,我这里计划对每张图做10次增强。小数据集适合离线数据增强,大数据集适合在线数据增强。原始图像和原始lalbel大小不一样,将原始图片与增强后的图像合并,并统一到同一个尺寸大小,将24位label转换为8位单通道,图像中的像素值从0开始递增,0代表背景,1代表第一类前景,依次类推。我的原图带有source字样,mask图带有mask字样,因此我将其都替换为NB。原创 2023-04-11 00:08:48 · 533 阅读 · 1 评论 -
语义分割前奏之数据标签转换
语义分割前奏之数据标签转换原创 2023-04-10 20:21:58 · 595 阅读 · 1 评论 -
使用BiseNet从头训练&&微调自己的数据集
2、MASK图为8位的单通道图片,图片中的值从0开始,0代表背景,第一类前景为1,第二类前景为2,依次类推递增。labelme的json转换出来的mask图为24位彩色图,要想办法转成8位。我使用的是bisenetv2,配置文件是configs里面的bisenetv2_city.py,数据集分为原图和对应的MASK,尤其要主要的有两点:1、所有图片要resize到统一大小。本次训练采用的是pytorch版本的BiseNet,代码链接为。生成train.txt val.txt的脚本文件。原创 2023-04-09 21:25:27 · 940 阅读 · 0 评论