프로젝트
프로젝트ㅣ- 시니어 건강관리 음식인식(4) 머신러닝
kingsgirl
2023. 6. 16. 09:46
728x90
yolov5로 모델을 학습시키면 가중치 파일이 생성됨
가중치 파일은 모델의 학습 과정에서 얻은 가중치 값을 저장하고 나중에 모델을 재사용하거나 이어서 학습하는데 사용함
django 웹에서 학습한 모델을 이용하기 위해서
가중치파일이 필요
가중치파일 을 얻기위해 colab에서 gpu하드웨어 가속기를 사용해서 학습하는 중 20번 직전 코랩 사용가능 메모리소진으로인해 멈춰서 일반으로
돌리는데 시간이 너무 오래걸림
yolov8모델을 찾아 실행시켜봄
yolov8의 기본 데이터셋은 잘 작동됨
커스텀데이터로변경(train, val)하고 data.yaml 파일 train,val
데이터셋 설정
-------------------------
실행
실행결과 exp 폴더는 모든 실험결과가 저장되고 name-- 으로 지정하면 이 이름의 폴더로 결과가 저장됨
메모리 초과했다고 뜨고 종료됨
728x90