- 博客(4)
- 收藏
- 关注
RSS订阅原创 mmsegmentation1.2.2+mmcv的安装(Win10)
我安装的是CUDA11.7，所以下载对应版本的cudnn，cudnn下载后解压，解压后目录下的内容全部复制，然后进入CUDA的根目录，我的是默认的路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7，将复制的文件粘贴到这里。安装CUDA的时候一定要主要，如果已经安装了显卡驱动，那么一定要选择自定义安装，然后在安装选项页面那知勾选CUDA，其他几项不要勾选，其他的诸如安装路径什么的，建议不要更改，如果更改的话要记着装在哪里了。
2023-12-29 12:24:37
1264
1
原创 语义分割任务中使用混淆矩阵计算IOU的原理及代码解读
为了搞明白这几行代码的原理，令a = torch.tensor([ 3, 255, 255, 4, 4, 5, 4, -1, 3, 3, 7, 7, 255, 2, 2, 3])，b=tensor([ 3, 255, 255, 3, 3, 5, 3, 255, 3, 3, 7, 7, 255, 2, 2, 3])。, 3, 3, 7, 7, 2, 2, 3])，红色位置的是预测错误的像素。
2023-12-26 16:46:59
957
原创 PASCALContext提取语义分割任务的Mask(Win10+Pytorch)
最近在做语义分割的实验，期间用到了PASCALContext数据集，该数据集是基于PASCAL VOC2010数据集划分出来的。其图片部分与VOC2010目录下的JPEGImages完全一致，使用了其中的4996张作为训练集，5104张作为验证集。不同于VOC2010的是，PASCALContext的标注信息扩展到了459类，其中用于语义分割任务的类别是59（不含背景）。我自己在执行上述代码的时候mode和transform指定的都是None，这样相当于仅提取了Mask并生成了训练和验证的txt文件。
2023-12-23 12:23:27
323
1
原创 从COCO2017数据集中提取语义分割任务的图片及Mask
从COCO2017数据集提取用于语义分割的训练集和验证集，解决了mask重叠导致的索引溢出。
2023-12-19 12:14:01
836
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人