[Keras] VGG-net
Keras로 작성된 VGG 종류에 따른 코드이다. VGG11, VGG11-LRN, VGG13, VGG16, VGG16-conv1 모델들이다. VGG11 def create_model(input_shape, label_class_cnt: int): """ :param input_shape: 입력 이미지 형태 :param label_class_cnt: 출력 class 수 :return: 생성된 모델 """ # weight 초기화 He_normal = initializers.he_normal(seed=None) model = Sequential() model.add(Conv2D(64, kernel_size=(3, 3),activation='relu',padding='same', input_shape=input..
2022. 9. 1.