글 수 191

vgg16관련 질문드려요

조회 수 83 추천 수 0 2020.11.24 20:31:29


trainset : 7000

testset : 200

model : vgg16

Layer :

conv_base = VGG16(weights='imagenet', include_top=False, input_shape=(150, 150, 3))

conv_base.trainable = False

model = models.Sequential()

model.add(conv_base)

model.add(layers.Flatten())

model.add(layers.Dense(4096, activation='relu'))

model.add(layers.Dropout(0.5))

model.add(layers.Dense(2048, activation='relu'))

model.add(layers.Dropout(0.5))

model.add(layers.Dense(1024, activation='relu'))

model.add(layers.Dropout(0.5))

model.add(layers.Dense(10, activation='softmax'))

 

model.compile(optimizer=optimizers.RMSprop(lr=2e-6),

              loss='categorical_crossentropy',

              metrics=['acc'])

checkpoint_cb = keras.callbacks.ModelCheckpoint(

    "/content/drive/My Drive/Colab Notebooks/my_fish_model.h5",

    monitor="val_acc", verbose=1, save_best_only=True)

 

history = model.fit_generator(train_generator,

                    steps_per_epoch=math.ceil(train_generator.n / train_generator.batch_size),

                    epochs=50,

                    validation_data=validation_generator,

                    validation_steps=math.ceil(validation_generator.n / validation_generator.batch_size),

                    callbacks=[checkpoint_cb])

 

 

지금 상황이 이런데 오버피팅 발생한것 같아 Dropout도 추가하고 했지만 지금 acc는 거의 90~93만 왔다갔다 해서 어떻게 해결해야 될까요 ㅜㅜ 머신러닝 관련공부를 한지 얼마안되 실습용으로 공부중인데 acc97이상 나오게 하려면 어떻게 해야되나요??

 

ipynb파일 첨부합니다. Colab 환경에서 사용했어요

 


깊은바다

2020.11.25 23:07:14
*.111.22.194

데이터셋의 크기가 7000개면 너무 적은 듯 합니다. 그게 과대적합의 가장 큰 이유인 것 같습니다. 제너레이터를 사용하고 있으면 증식은 적용했는지요.

또한 지금은 feature extraction인데 finetuning으로 베이스모델의 일부분이 학습되도록 변경해보세요. 약간 적확도가 상승합니다.
List of Articles
제목 글쓴이 날짜 조회 수
안녕하세요 이제 막 공부하기 시작한 학부생입니다. sk211 2021-01-16 30
MZ 인공지능 해커톤 대회 - 의사(간호사)-환자 간 대화로부터 의료 관련 주요 항목 응답(=MedicalQuAD) file 락희베쑤 2021-01-11 106
파일을 AI를 통해 자동 분류가 가능할까요? [1] ghghgh 2021-01-10 88
AIoT 개발총괄 해외근무 영어 가능 8년 이상 ( 연봉 세전 1.6억~2.5억 ) KoreaHR 2021-01-06 105
Dialogflow로 이미지 불러오기가 가능할까요? [1] 재앤더플루이드 2021-01-06 78
웹 상에서 그래프 그리기 nari 2021-01-05 94
(온라인 컨퍼런스안내) 인공지능기업 아크릴 20년12월22일 'Acryl Next2020'개최 / i-pad 등 상품추첨 event file 아크릴20 2020-12-10 211
MZ 인공지능 해커톤 대회 - AI 장치용 STT(Speech To Text)를 위한 의도 분류 (AI-NLU with STT) [2] 락희베쑤 2020-12-08 233
[가입인사] AI STARTUP 기업과 개발자 자료를 검색하다가 등록합니다. [1] hornet 2020-12-07 77
데이터 라벨링 후 궁금증 [1] fex5360 2020-11-26 102
vgg16관련 질문드려요 file [1] UOU 2020-11-24 83
딥러닝 어플리케이션을 쉽게 만들수있는 서비스 DLBuilder file DLBuilder 2020-11-20 117
스타트업 관련 인공지능 머신러닝 궁금증 Kyle13 2020-11-10 114
오픈 컨트리뷰터의 코드 참고법에 대해 질문드립니다. [2] Roger 2020-10-19 75
딥러닝을 위한 레이블링 질문입니다. [1] Jang 2020-10-15 87