固定随机数，并生成指定数量的颜色，每次生成的颜色相同

分类: IT文章 • 2023-11-14 19:41:54

参考mmdetection代码

地址：https://github.com/open-mmlab/mmdetection/blob/master/mmdet/core/visualization/image.py：

state = np.random.get_state()

# random color

np.random.seed(42)

mask_colors = [

np.random.randint(0, 256, (1, 3), dtype=np.uint8)[0].tolist()

for _ in range(len(categories))

]

np.random.set_state(state)