RGB_to_L RGB标签转类别图工具
三通道 RGB 图的标签太占空间？RGB_to_L.py帮你省事不少。它能把 RGB 图批量转成单通道的 L 图，像素点直接代表类别，做语义分割的标注挺方便。完之后，你训练模型就轻松多了，图像小了、内存省了，速度也快了不少。
脚本用法也简单，路径设好，运行就完事。没有太多花哨逻辑，核心就一个：把每个 RGB 值映射成一个类的整数，你要是语义标注的颜色表固定，那转起来稳。配合你自己的palette表，用起来就更灵活了。
语义分割任务里，不少数据集都喜欢用 RGB 图当标签，比如PASCAL VOC和PASCALContext，但模型一般要的是class id。用这脚本一转，模型输入就顺了。顺便说一句，如果你想了解 PASCALContext 的格式，可以看看这篇。
嗯，对了，要是你对图像分割相关的算法感兴趣，基于深度学习的实时语义分割算法研究这篇也挺值得看，能帮你更好理解这个脚本背后的应用场景。
如果你刚好在标注图、准备训练语义分割模型，那RGB_to_L.py真挺适合放进你的数据预流程里的。简单高效，还能少掉多手动的麻烦。
文件大小：1.14KB
评论区