图像分割
文章平均质量分 69
CV-deeplearning
小米AI Lab图像算法工程师,擅长表格识别、目标检测、行人属性识别、车牌识别等。
展开
-
mmsegmentation 训练自己的数据集
介绍基于MMSegmentation训练自己的数据集。原创 2023-04-16 15:58:45 · 1831 阅读 · 4 评论 -
语义分割中的常用Loss实战
1.nn.BCEWithLogitsLoss() 与nn.CrossEntropyLoss() 工程代码参考:https://github.com/milesial/Pytorch-UNet.git (1)nn.BCEWithLogitsLoss() 一般用于两个类别的分割,只有前景和背景。 (2)nn.CrossEntropyLoss() 一般用于多个类别的分割。 具体应用见代码和注释:import torchimport torch...原创 2021-01-19 17:49:58 · 2306 阅读 · 2 评论 -
Unet++ 实战
最近在做人体解析,即把人体的各个部位分割出来。博主用了Unet和Unet++作为baseline进行了对比, 发现Unet++的效果不错,在此分享给大家。原创 2021-01-18 18:18:14 · 9486 阅读 · 5 评论 -
如何将mask rcnn预测的mask(二值图)转化为标注信息(segmentation)
用训练好的mask rcnn模型对图片进行预测时,如何生成标签。即将mask rcnn预测的mask信息转化为标注需要的segmentaton信息。原创 2020-11-03 15:39:29 · 1398 阅读 · 4 评论 -
将mask的标注信息转化为二值图,mask iou计算
做实例分割时,经常会遇到mask的相关问题,本文着重介绍由segmentation生成mask的二值图以及两个mask的iou计算。原创 2020-11-03 15:13:49 · 6547 阅读 · 5 评论 -
Cascade Mask R-CNN实战——训练自己的数据集
Cascade Mask R-CNN是在Mask R-CNN基础上的改进版本,其精度要高于Mask R-CNN, 本文讲带你如何训练Cascade Mask R-CNN。原创 2020-06-11 18:37:21 · 5112 阅读 · 1 评论 -
SOLO实战——用自己的数据集训练实例分割模型
SOLO实例分割算法原理简单,并且分割精度达到mask R-CNN的水平,用了Matrix NMS(SOLOV2中的内容)后,其推理速度明显快于mask R-CNN,本文会带你一步一训练自己的数据集。solov2官方也开源了,同样适用。原创 2020-06-08 18:02:33 · 13455 阅读 · 84 评论