一次性跑n个图片和对应的内存,显卡 1080Ti
图片数 | 内存 |
---|---|
1000 | 8G |
2000 | 11G |
2800 | 19G |
2900 | 21G |
3100 | 24G |
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
import time
p = pipeline('ocr-detection', 'damo/cv_resnet18_ocr-detection-line-level_damo',model_revision='v1.0.0',device="cuda:1")
for i in range(3000):
result = p('font1.jpg')
print(f'{i+1}/3000')
使用官方文档的方法运行,其中 ocr_detection 循环处理图片,大约一张图片需要2s。 有无加速和解决内存泄露的方案?