묻고 답해요

keras-yolo 모델 학습관련 질문

미해결

[개정판] 딥러닝 컴퓨터 비전 완벽 가이드

안녕하세요 선생님 먼저 좋은 강의 정말 감사합니다. 저는 [yolo raccoon custom 강의]를 보고 keras-yolo에 제 데이터에 custom하는 작업을 진행 중인데요. 문제는 제 데이터 양 때문인지 한 1epoch 당 3시간 정도가 걸립니다. 저는 aws 를 사용하고 있는데 이게 오래 학습을 시키면 저절로 서버다운이 되고... 중간에 학습모델 저장이 쉽지가 않습니다. 그래서 2epoch씩 학습시키고 저장하고 -> 저장된 모델을 로드해서 다시 2epoch 학습하고 이런 방식으로 노동을 하고자 하는데 그 방법을 여쭈어보고 싶습니다. 여기서 저장된 모델 이름은 trained_weight_stage_1.h5 입니다. fine-tune은 할 엄두도 못내고 있구요... 1. 2epoch 학습한 trained_weight_stage_1.h5 를 yolo.h5 자리에 넣고 학습시키면 되는 걸까요? 2. 어느정도 학습한 모델을 이어서 그다음 코드인 fine-tune 단계만 진행하려면 어떻게 해야할까요?

• keras-yolo
• 머신러닝 배워볼래요?
• tensorflow
• keras
• 컴퓨터-비전
• python
• 딥러닝