yolov8推理视频爆内存的处理方式

- 事件概述

使用yolov8推理1080P视频来测试其在服务器上的资源占用,结果发现推理时,内存会随着时间增长,直到内存耗尽进程被killed,最终视频没有全部推理完毕。

ultralytics>engine>predictor

在上述文件位置写有一个warning,提示如果不使用stream=true,有内存不够的风险,下面也给了使用的例子(其实在模型启动后后台也有显示这段内容,不过都一闪而过没仔细看)

- 方式一(仅针对视频流)

测试发现只有输入是流时,stream=true才能正确启动模型,若输入为视频文件,则模型不会启动。

正确的启动格式如下(以track为例):

from ultralytics import YOLO

def yolo_start():
    # Load a model
    model = YOLO("yolov8l.pt")  # load a pretrained model (recommended for training)
    # Use the model
    results=model.track(source=source,tracker="bytetrack.yaml",stream=True)
    for r in results:
        boxes = r.boxes  # Boxes object for bbox outputs
        masks = r.masks  # Masks object for segment masks outputs
        probs = r.probs  # Class probabilities for classification outputs
yolo_start()

和官方例子一样for循环不能省,省去同样模型不启动

- 方式二

如果是推理较大视频,推荐使用CLI命令直接后台开启,或者python脚本用subprocess执行CLI命令,具体内存耗损没有测,但是使用model.track()没有推理完的视频,用CLI是可以推理完的

  • 6
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值