File size: 314 Bytes
752c636
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import numpy as np
import tensorflow.keras.backend as K



cell_size=32
class_names=['face']
class_colors={class_name:np.random.rand(3) for class_name in class_names}



class_to_idx={class_name:i for i,class_name in enumerate(class_names)}
idx_to_class={i:class_name for i,class_name in enumerate(class_names)}