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 |