코드 6-8 필요한 라이브러리 호출
%load_ext tensorboard
import numpy as np
import tensorflow as tf
import matplotlib.pyplot as plt
from tensorflow.keras import Model
from tensorflow.keras.models import Sequential
from tensorflow.keras.utils import to_categorical
from tensorflow.keras.losses import categorical_crossentropy
from tensorflow.keras.preprocessing.image import ImageDataGenerator
from tensorflow.keras.layers import Dense, Flatten, Conv2D, MaxPooling2D, Dropout
데이터셋은 앞서 진행했던 개와 고양이 이미지를 계속 사용합니다. 이제 준비된 데이터를 이용할 네트워크를 생성해야 합니다. 원래 AlexNet의 마지막 계층에는 뉴런이 1000개 있지만 예제에서는 클래스 두 개만 사용합니다. 또한, 합성곱 신경망을 구축하는 데 Model Subclassing API를 사용합니다.
▲ 그림 6-11 AlexNet 예제 네트워크