bmf icon indicating copy to clipboard operation
bmf copied to clipboard

Published 20 hours ago • verified publisher icon BabitMF

使用torch.from_dlpack(vf.reformat(rgb).frame().plane(0))方法，会增加很多显存

Open zhangsong1234 opened this issue 9 months ago • 2 comments

人脸检测模型中使用torch.from_dlpack(vf.reformat(rgb).frame().plane(0))方法后显存明显增加了很多，输入是4K图片，使用与不使用这个方法显存差了500M左右

zhangsong1234 avatar May 23 '24 07:05 zhangsong1234