[Object Detection] YOLO의 모든것 - YOLOv3, YOLOv5

YOLOv3 모델 (TensorFlow)

• YOLOv3 모델 다운로드
• 기존의 DarkNet으로 학습된 모델 파일을 Keras에서 사용할 수 있는 Tensorflow 모델 파일로 변환
• 객체 탐지를 위해 사용할 Detector 정의
• 예제 데이터를 Detector에 입력하여 객체 인식 수행
• 객체 인식 결과 확인

YOLOv3 모델 (PyTorch)

• YOLOv3 모델 다운로드
• 객체 탐지를 위해 YOLOv3를 받아서 사용할 Detector인 detect.py를 사용해 모든 입력 데이터에 대해 객체 인식 수행
• 저장된 객체 인식 결과 확인

YOLOv5 모델 (PyTorch)

• YOLOv3 모델 다운로드
• 객체 탐지를 위해 YOLOv5를 받아서 사용할 Detector인 detect.py를 사용해 모든 입력 데이터에 대해 객체 인식 수행
• 저장된 객체 인식 결과 확인

 

 

 

참고자료 및 코드

 

객체 탐지 Object Detection - YOLO의 모든것 / 포트홀 탐지, 연기 탐지, 안전모 탐지, 주차공간 탐지

https://youtu.be/fdWx3QV5n44

 

YOLOv3 모델 (TensorFlow)

https://github.com/GaGa-Kim/ML_Study/blob/main/YOLO%EC%9D%98%20%EB%AA%A8%EB%93%A0%EA%B2%83/YOLOv3_%EB%AA%A8%EB%8D%B8_(TensorFlow).ipynb 

 

YOLOv3 모델 (PyTorch)

https://github.com/GaGa-Kim/ML_Study/blob/main/YOLO%EC%9D%98%20%EB%AA%A8%EB%93%A0%EA%B2%83/YOLOv3_%EB%AA%A8%EB%8D%B8_(PyTorch).ipynb 

 

YOLOv5 모델 (PyTorch)

https://github.com/GaGa-Kim/ML_Study/blob/main/YOLO%EC%9D%98%20%EB%AA%A8%EB%93%A0%EA%B2%83/YOLOv5_%EB%AA%A8%EB%8D%B8_(PyTorch).ipynb