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
'머신러닝 > Pytorch' 카테고리의 다른 글
Save the model object into a file for restoration/de-serialization (0) | 2021.04.02 |
---|---|
define the optimizer and the learning rate schedule (0) | 2021.01.06 |
visualize the filters of the first CNN layer (0) | 2021.01.06 |
store best weights (0) | 2021.01.04 |
Storing and loading models (0) | 2020.12.31 |