자유게시판

YOLOv8을 이용한 multi counting(car, truck, motorcycle) 예제 python 코드
2023. 8. 2 (수) 13:14 goingdol 조회 41 좋아요 0 스크랩 0 댓글 0

안녕하세요.

YOLOv8에 관심있는 분들께 조금이나마 정보를 드리고자 글을 올립니다.

동영상을 보시면 아시겠지만 관련 예제코드는 Freedom tech (https://www.youtube.com/@freedomtech5307/videos)채널에서 공유한 코드를 이용하여 수정/추가 했습니다. 해당채널에 여러가지 도움이 될만한 예제가 있으니 참고하시면 코드를 작성할때 도움이 될거 같네요.

 

 

multi counting에 관련된 코드는 coco.txt, multi_counting_speed.py, tracker.py 그리고 테스트용 동영상 highway_1280.zip 입니다.

동영상 zip파일은 압축을 풀으셔야 합니다.

 

관련 코드는 다음의 github에 공유되어 있습니다.

https://github.com/goingdol/yolov8_example

 

코드를 보시면 아시겠지만 MS coco dataset으로 pre-trained된 model.predict의 results에서 car, truck, motorcycle 중 하나이면 

해당 object의 box좌표를 가져와서 center좌표를 계산하고 Line을 지날때 counting하고 속도를 계산하여 display해 주도록 되어 있습니다.

참고로 motorcycle은 속도계산 부분이 안들어가 있습니다. pixabay.com같은 무료 동영상 배포해주는 site에서 영상을 받아서 한번 추가해 보시면 어떨까요?

 

YOLOv8로 custom dataset을 만들고 이것을 이용해서 응용기능을 추가하고자 하는분들께 좋은 정보가 됬으면 좋겠습니다.

혹시 좋은 정보를 가지고 있으면 공유 부탁드립니다.

 

감사합니다.

 

로그인 후
참가 상태를 확인할 수 있습니다.