现在有颜色字典class_color_dict = {0: (0, 0, 0), 1: (128,0,0), 2: (0,128,0), 3: (128,128,0), 4: (0,0,128), 5: (128,0,128), 6: (0,128,128), 7: (128,128,128), 8: (64,0,0), 9: (192,0,0), 10: (64,128,0), 11: (192,128,0), 12: (64,0,128), 13: (192,0,128), 14: (64,128,128), 15: (192,128,128), 16: (0,64,0), 17: (128,64,0), 18: (0,192,0), 19: (128,192,0), 20: (0,64,128)},字典中的值为颜色值输入掩码mask为RGB模式,形状为[16,3,511,511],其中16为batchsize,现在我想得到一个形状为[16,511,511]的张量,使得每个位置按照mask的颜色值转换为对应的键值,基于pytorch的代码该如何实现?
最新发布