[Object Detection] YOLO의 모든것 - 커스텀 학습

권총 탐지 모델

• YOLOv5 모델 다운로드
• 권총 데이터셋 다운로드
  
  
  • 테스트 데이터를 분리해 검증용 데이터 생성
  • 데이터셋 이미지 경로들을 yaml 파일로 생성
• 학습
  
  
  • 배치 크기, 학습 기간, 모델 구성, 가중치 등을 정의하여 학습
  • 텐서보드를 통해 학습 진행 확인
• 추론
  
  
  • 가장 수치가 좋았던 가중치 파일 모델을 이용해 테스트 이미지 추론
• 모델 내보내기
  
  
  • 가장 수치가 좋았던 가중치 파일 모델을 구글 드라이브로 내보내기
• 학습된 모델을 이용해서 영상에 있는 총을 인식하기
  
  
  • 저장된 모델을 사용해 동영상 파일 경로를 source로 정의하여 총 인식

합성 과일 탐지 모델

• YOLOv5 모델 다운로드
• 합성 과일 데이터셋 다운로드
• 학습
  
  
  • 배치 크기, 학습 기간, 모델 구성, 가중치 등을 정의하여 학습
• 추론
  
  
  • 가장 수치가 좋았던 가중치 파일 모델을 이용해 이미지 추론
• 모델 내보내기
  
  
  • 가장 수치가 좋았던 가중치 파일 모델을 구글 드라이브로 내보내기
• 학습된 모델을 이용해서 영상에 있는 총을 인식하기
  
  
  • 저장된 모델을 사용해 동영상 파일 경로를 source로 정의하여 총 인식

 

 

 

참고자료 및 코드

 

YOLO v5 커스텀 학습 튜토리얼

https://youtu.be/T0DO1C8uYP8

 

권총 탐지 모델

https://github.com/GaGa-Kim/ML_Study/blob/main/YOLO%EC%9D%98%20%EB%AA%A8%EB%93%A0%EA%B2%83/%EA%B6%8C%EC%B4%9D_%ED%83%90%EC%A7%80_%EB%AA%A8%EB%8D%B8.ipynb

 

합성 과일 탐지 모델

https://github.com/GaGa-Kim/ML_Study/blob/main/YOLO%EC%9D%98%20%EB%AA%A8%EB%93%A0%EA%B2%83/%ED%95%A9%EC%84%B1_%EA%B3%BC%EC%9D%BC_%ED%83%90%EC%A7%80_%EB%AA%A8%EB%8D%B8.ipynb