머신러닝/Pytorch

define colors using random tuples

조마조마 2021. 1. 6. 16:24
COLORS = np.random.randint(0, 255, size=(len(classes), 3), dtype="uint8")
# draw loop
	color = [int(c) for c in COLORS[idx]]

※ reference: pytorch computer vision codebook