使用yolo预测的时候一直在输出信息,类似下面这种,比较影响debug,查了一查是由于ultralytics自己实时输出推理结果。
0: 1080×1920 1 person, 47.8ms
Speed: 1.0ms preprocess, 47.8ms inference, 0.0ms postprocess per image at shape (1, 3, 1080, 1920)
解决办法:在不修改源代码的情况下,修改预测语句:
原代码(这样会输出大量的诊断信息):
results = yolo_model(frame)或者results = yolo_model.predict(frame)
修改为:
results = yolo_model.predict(frame,verbose=False)
这样就能停止输出推理信息了。