다부처 인수인계용 정리본 edge-analysis-module 테스트

edge-analysis-module Test

yolo 모델 받기 및 TensorRT 진행

edge module 같은 경우에는 CCTV 디바이스에 사용해야 하기 때문에 TensorRT로 경량화 작업을 해줘야함.
설치는 방법 URL에서 yolov7설치 및 경량화 작업 코드드 확인해 봐야함.
해당 코드는 보안상으로 올리진 않음.
yolov7는 현재 깃허브에 올라와 있는데 yolov4는 찾아본 결과 없음. (yolov4는 송진하 박사님께 따로 얘기해볼 예정)

edge-analysis-module Test

edge anlaysis module test를 하기 위해서는 test/test_video_all_model.py를 이용해서 돌려야 됨.

video_path => video 경로
resol => 360p or 1080p
params_path => yolov4 or yolov7으로 진행하면 됨
process_time => 각 이벤트들의 속도를 확인하는 작업임.

이렇게 총 4개로 설정을 진행할 수 있는데 필수적으로 필요한건 --video_path임.

python3 test/test_video_all_model.py --video_path=/workspace/videos/videos/total_videos/360p/mix_09_360p.mp4

위 코드를 실행시켜서 테스트를 진행하면 됨. 결과는 results/video에 저장이 됨.
제대로 동작했을 때 나오는 결과

[2023-03-24 14:15:12.400298] -     INFO: parameter Info:
                                         input video path: /workspace/videos/videos/total_videos/360p/mix_09_360p.mp4
                                         input json path: /workspace/videos/videos/total_videos/360p/mix_09_360p.json
                                         decoding parameters
                                         - mode: cpu
                                         - fps: 20
                                         running model
                                         - batch size: 20
                                         - object detection: yolov7-tiny
                                         - tracker: ['byte_tracker', 'sort_tracker']
                                         - event: ['assault', 'falldown', 'kidnapping', 'tailing', 'wanderer']
                                         result parameters
                                         - final result video path: results/video/mix_09_360p_edge_20fps_230324.avi
                                         visual parameters:
                                                object:
                                                - result visualize: True
                                                - visualize score threshold: 0.1
                                                event:
                                                - result visualize: True
[2023-03-24 14:15:15.218489] -     INFO: object detection model(yolov7-tiny) model is loaded - 2.804 sec
[2023-03-24 14:15:15.218576] -     INFO: loading trackers
                                                byte_tracker is loaded - 0.000 sec
                                                sort_tracker is loaded - 0.000 sec
[2023-03-24 14:15:15.218757] -     INFO: loading event models
                                                assault is loaded - 0.000 sec
                                                falldown is loaded - 0.000 sec
                                                kidnapping is loaded - 0.000 sec
                                                tailing is loaded - 0.000 sec
                                                wanderer is loaded - 0.000 sec
[2023-03-24 14:16:05.570065] -     INFO: frame number:   5751/5751      / timestamp: 0:03:11.700[2023-03-24 14:16:05.586588] -     INFO: processing time:
                                                object detection(yolov7-tiny) - total: 34.139944        / average: 0.005936     / max: 0.019321  / min: 0.005551
                                                tracker
                                                        byte_tracker - total: 2.750163  / average: 0.000478     / max: 0.002935  / min: 0.000115
                                                        sort_tracker - total: 1.779376  / average: 0.000309     / max: 0.012637  / min: 0.000087
                                                event detection
                                                        assault - total: 0.255030       / average: 0.000044     / max: 0.000605  / min: 0.000010
                                                        falldown - total: 0.179451      / average: 0.000031     / max: 0.000173  / min: 0.000012
                                                        kidnapping - total: 0.068029    / average: 0.000012     / max: 0.000248  / min: 0.000005
                                                        tailing - total: 0.235289       / average: 0.000041     / max: 0.001218  / min: 0.000012
                                                        wanderer - total: 0.055662      / average: 0.000010     / max: 0.000066  / min: 0.000004
                                                total process time: 50.351420
[2023-03-24 14:16:05.618798] -     INFO: event result file is successfully extracted.(path: results/video/mix_09_360p_edge_20fps_230324.csv)
[2023-03-24 14:16:05.619104] -     INFO: sequence result file is successfully extracted.(path: results/video/mix_09_360p_edge_20fps_230324.json)
[2023-03-24 14:16:39.094570] -     INFO: frame number:   3833/3834 - {'assault': False, 'falldown': False, 'kidnapping': False, 'tailing': False, 'wanderer': False}
[2023-03-24 14:16:39.099828] -     INFO: sequence result video is successfully generated(path: results/video/mix_09_360p_edge_20fps_230324.avi)

다부처 결과 확인하는 URL

다부처 결과 확인 서버를 접속하면 다부처 결과를 확인할 수 있음.