- 博客(4)
- 收藏
- 关注
RSS订阅原创 mmsegmentation 训练自己的数据集，三通道标签转换为单通道标签
摘要：本文解决了mmsegmentation框架中因数据尺寸不匹配导致的报错问题。通过将三通道彩色标签图转换为单通道调色板模式(P模式)，实现了标签数据的统一处理。解决方案包含：1)建立RGB颜色到类别ID的映射；2)遍历像素匹配预定义调色板；3)创建符合PIL规范的768长度调色板；4)批量转换处理train/val/test数据集。该方法有效解决了"RuntimeError: stack expects each tensor to be equal size"错误，适用于多类别语义
2025-06-29 19:40:04
182
原创 CUDA version (11.8) mismatches
这个错误我一搜都说是版本不匹配让你重装，但是我只是用来编码几行CUDA内核代码，不应该版本差这么大。首先提示我(11.8) mismatches，我电脑上根本没有装过11.8，于是nvcc --version发现查不到CUDA，于是发现问题，应该是编译代码找不到cuda路径。解决：由于我是普通用户，在bashrc文件里面没有cuda-home，于是添加进去，nvcc --version成功，编译成功！
2025-02-19 14:56:10
549
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人