paddleocr快速使用tensorrt进行推理加速

self.ocr = PaddleOCR(
	         use_angle_cls=True,
	          use_space_char=True,
	          use_gpu=True,
	          use_tensorrt=True,
	          precision='fp32', #根据需要改fp16或fp32
	          det_model_dir='models/ocr/ch_PP-OCRv4_det_infer/',
	          rec_model_dir='models/ocr/ch_PP-OCRv4_rec_infer/',
	          cls_model_dir='models/ocr/ch_ppocr_mobile_v2.0_cls_infer/',
	          shape_info_filename="shape.txt",
	          thread_num=8#线程数
                             )

注意事项:

报错:ppocr INFO: when using tensorrt, dynamic shape is a suggested option, you can use ‘–shape_info_filename=shape.txt’ for offline dygnamic shape tuning,

–shape_info_filename=“shape.txt”,首次运行:程序会分析输入图像的尺寸范围,并生成shape.txt文件。后续运行:直接复用该文件,避免重复优化。但是,在执行新的进程时一定要删除shape.txt文件再运行否则会报错。
误区:使用tensorrt就一定要用到engine文件×,这里的文件地址输入是’models/ocr/ch_PP-OCRv4_det_infer/',不需要我们自己转换。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值