[YOLOv4] 인식 결과 저장하기

Yolov4를 이용한 프로젝트를 진행하다 보면 인식하고 있는 장면을 보여줘야 하는 경우가 종종 발생합니다.

 

이전에는 실시간으로 인식하고 있는 장면을 화면 녹화 프로그램으로 찍었었는데

 

여러 인식기를 만들어 같은 영상을 틀어줘서 성능을 비교하는 작업을 하다보니 녹화 프로그램을 쓰기가 굉장히 불편했습니다.

 

그래서 검색을 해보니 Yolov4에 자체적으로 인식 결과를 저장하는 명령어가 있더라구요.

 

이번에도 AlexeyAB의 GitHub를 참고했습니다.

 

https://github.com/AlexeyAB/darknet

GitHub - AlexeyAB/darknet: YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Da

 

 

./darknet detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights test.mp4 -out_filename res.avi

동영상을 실행하는 명령어 뒤에 (data, cfg, weight, 동영상 파일은 본인의 것에 맞게 바꿔야 합니다.)

 

-out_filename res.avi만 달아주면 인식을 마친 뒤 darknet 실행 파일이 있는 경로에 결과 영상이 저장됩니다.

 

영상이름은 res.avi가 아니라 다른 걸로 바꿔줘도 됩니다.