모델 설계
모델을 설계해보겠습니다. 다음 코드는 모델의 학습 시간을 고려하여 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