🍀

VGG16 모델로 이미지 분류 (훈련)

작성일자
Mar 22, 2023
태그
DIARY_DEVELOP
프로젝트
BeachCombine
책 종류

VGG16 모델로 이미지 분류

Train에서 발생한 에러

  • 아래 코드를 돌렸을 때
    • 아래와 같이 에러 로그가 떴다.
      • 해결
        • 이 에러는 fit_generator()함수가 더 이상 지원되지 않기 때문에 발생함. 대신, fit() 함수를 사용해야 함.
        • 에러 메시지에서 "Your input ran out of data"와 같은 메시지가 표시될 때, 주로 데이터셋이나 데이터 제너레이터에서 문제가 발생한 경우임. 이 에러 메시지는 데이터셋 또는 데이터 제너레이터에서 생성된 배치(batch)의 수가 모델이 지정한 배치 수보다 적어서 발생함. → 나는 train, test_set의 batch_size와 validation_steps를 32에서 10으로 줄여주어 해결함
        • 수정한 코드

        모델 Train 결과 - 이상 있음

        notion image
        • loss가…. 감소하다가 갑자기 에포크 8쯤에서 증가한다. 감소하는 거 보고 좋아했는데 잠시 눈 뗐다가 다시 보니,, 배신,,
        • 손실함수가 감소하다가 갑자기 커지는 건 오버피팅 떄문일 확률이 높다고 한다,,,
        • 해결하기 위한 방법은 일단 정규화를 해보고 안되면 데이터 더 찾아 넣음 된다,,

        Test 진행 중 발생한 에러

        • preds = model.predict(x) 코드 실행 시 하단 에러 발생
          • 해결
            • 원인) 모델 객체의 call메소드가 구현 안 되어 있어 발생한 에러임
            • 해결 방법) 모델 객체를 다시 구현하거나, inputsoutputs 인수를 명시하여 Functional API를 사용하여 모델 객체를 생성하면 됨

            하루 정리

            TIL 작성하기
            BeachCombine
            클라 수정사항 반영
            ML 모델 학습시키기
             
            VGG16 모델로 이미지 분류 (훈련)