mask图转成cityscapes格式
自定义数据的读取是快速验证模型的最重要的一步。模型结构可以复用,但自己的数据集和公开的citysacpes,coco数据集标注格式还是有差异。以下是针对三通道彩色或单通道白色mask图转成cityscapes mask图格式。
import numpy as np
from PIL import Image
###15个类别,背景是黑色。类别1对应[1,1,1]####
PALETTE = [[0, 0, 0], [1, 1, 1], [2, 2, 2], [3, 3, 3],
原创
2021-09-29 09:45:52 ·
482 阅读 ·
0 评论