728x90
반응형
Object detection같은 task할때마다 매번 모델 시각화 할 때 클래스 별 RGB를 맵핑해줘야 하는 경우.
다른 패키지보다 이게 젤 간단...
단점은 비슷한 컬러들이 생길 수 있음.
colors = {cls: (np.random.randint(0, 256), np.random.randint(0, 256), np.random.randint(0, 256)) for cls in CLASS}
728x90
반응형