더북(TheBook)

모델 설계

모델을 설계해보겠습니다. 다음 코드는 모델의 학습 시간을 고려하여 VGG16 아키텍처를 활용하여 YOLO 모델을 구축하는 작업을 수행합니다.

max_num = len(tf.keras.applications.VGG16(weights='imagenet', include_top=False,  
input_shape=(224, 224, 3)).layers) # ①
YOLO = tf.keras.models.Sequential(name = "YOLO")
for i in range(0, max_num-1):
    YOLO.add(tf.keras.applications.VGG16(weights='imagenet', include_top=False, input_shape=(224, 224, 3)).layers[i])

initializer = tf.keras.initializers.RandomNormal(mean=0.0, stddev=0.01, seed=None) # ②
leaky_relu = tf.keras.layers.LeakyReLU(alpha=0.01) # ③
regularizer = tf.keras.regularizers.l2(0.0005) 

for layer in YOLO.layers: # ④
    layer.trainable=False
    if (hasattr(layer,'activation'))==True:
        layer.activation = leaky_relu
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.