딥러닝 - 콜백을 이용하여 가장 좋은 모델과 로그 저장하기
이번 포스팅은 콜백을 이용하여 모델/로그를 저장하는 것을 중점으로 작성하였습니다. 딥러닝의 모델링 부분은 포함하지 않습니다. 모델/로그 저장 할 디렉토리 만들기 # 파이썬으로 디렉토리를 만드는 코드 작성 PROJECT_PATH = '.' # 기본 경로 설정, 현재 위치 model_type='mobilenetv2' # 사용중인 모델 # 모델과 로그기록 저장 경로 설정 # CHECKPOINT_PATH: ./checkpoints/mobilenetv2/mobilenetv2.h5 CHECKPOINT_PATH = PROJECT_PATH+'/checkpoints/'+model_type+'/'+model_type+'.h5' # LOGFILE_PATH: ./log/mobilenetv2/mobilenetv2-trainin..
- AI & ML/Machine Learning (Python)
- · 2022. 6. 19.
파일을 train/test로 디렉토리를 나눠서 저장하는 방법
해당 포스팅은 구글 코랩(Colab)을 기준으로 작성하였습니다. 데이터 파일 데이터 파일 다운로드 후 압축해제 # /tmp 경로에 압축파일을 다운로드 받고 압축 푼다. !wget --no-check-certificate \ "https://block-edu-test.s3.ap-northeast-2.amazonaws.com/kagglecatsanddogs_5340.zip" \ -O "/tmp/cats-and-dogs.zip" import zipfile local_zip = '/tmp/cats-and-dogs.zip' zip_ref = zipfile.ZipFile(local_zip, 'r') zip_ref.extractall('/tmp') zip_ref.close() 파일을 분리하고 저장 할 디렉토리 생성 ..
- AI & ML/Machine Learning (Python)
- · 2022. 6. 15.