[컴퓨터비전(이미지/영상)] Object Detection - Part2. Yolo 알고리즘 실습

1. Yolov5 모델 실습

 - 실습 내용 Code 및 설명

---

Open in Colab으로 해당 코드 실습 - 아래 작성된 설명 및 코드를 따라 실습 진행하시면 됩니다.

 

 

- 훈련 모델 결과

---

https://wandb.ai/에 방문하여 metric, loss, learning rate등을 확인이 가능.

- 생성된 모델의 추론 테스트

---

첨부된 테스트 이미지를 드라이브에 업로드 후 "!python detect.py --weight 훈련된pt파일경로 --source 해당경로" 테스트 진행

(학습이 계속 진행 중이어서 기존에 추론했던 이미지 업로드)

Public으로 공개된 오픈 데이터셋을 통해 학습하였으며, 내부 요구사항을 반영하여 데이터셋을 Customize한다면 적합한 모델이 생성 될것 으로 기대됨.

 

추론 결과는 각 객체에 바운딩 박스되어 confidence값이 나오게 됩니다.

Flask로 구현한 이미지 추론 웹사이트(영상 및 실시간 CCTV또한 가능하다.)