AI 탐구노트

Ultralytics Yolo 모델의 로그 안 보이게 하기 본문

기술 팁

Ultralytics Yolo 모델의 로그 안 보이게 하기

42morrow 2024. 11. 21. 12:27

1.문제점

Yolov8 이후부터 객체 감지를 코드를 실행하면 로그가 아래와 같이 발생합니다. 흠... 내가 찍으라고 한 것도 아닌데 이건 뭐지? 그냥 python에서 로그 안 남기도록 하거나 log level을 높이면 되나? 했는데 그걸로도 안 돼네요.

 

0: 480x640 1 person, 67.0ms
Speed: 2.7ms preprocess, 67.0ms inference, 87.1ms postprocess per image at shape (1, 3, 480, 640)

0: 480x640 1 person, 8.1ms
Speed: 1.3ms preprocess, 8.1ms inference, 1.3ms postprocess per image at shape (1, 3, 480, 640)

0: 480x640 1 person, 8.4ms
Speed: 0.8ms preprocess, 8.4ms inference, 1.2ms postprocess per image at shape (1, 3, 480, 640)

0: 480x640 1 person, 8.4ms
Speed: 0.8ms preprocess, 8.4ms inference, 1.2ms postprocess per image at shape (1, 3, 480, 640)

0: 480x640 1 person, 7.6ms

 

 

2.원인 검색

검색을 해 본 결과, Yolov8에서 생성된 것으로 이와 관련된 것은 ultralytics 패키지라고 합니다. 

 

 

 

Hide console output when objects are detected · Issue #82 · roboflow/supervision

Search before asking I have searched the Supervision issues and found no similar feature requests. Question Is there a way to hide the output to console when objects are detected? Additional No res...

github.com

 

 

3.해결 방법

Ultralytics 패키지를 이용해 Yolo 모델 객체를 생성할 때 verbose 옵션을 아래와 같이 비활성화해 주면 됩니다. 기본이 활성화라고 하네요...

model.predict(source=0, verbose=False)