模型优化
文章平均质量分 89
一呆飞仙
你那么努力,一定能活到大结局!
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
pytorch 网络预处理与后处理中基于numpy操作的GPU加速
0、背景python脚本运行在服务器端的卷积神经网络往往需要将图片数据从cv2(numpy.ndarray)->tensor送入网络,之后进行inference,再将结果从tensor-> numpy.ndarray的过程。由于cv2读取的数据存于内存中,以pytorch框架举例,在把数据送入GPU前会产生如下的数据转换:GPU准备进行inference之前会判断torch.c...原创 2019-11-04 09:51:33 · 4062 阅读 · 2 评论 -
TensorRT与pycuda配置运行yolov3
一、pycuda安装http://pypi.python.org/pypi/pucuda点击左侧download files,下载最新版本的pycuda#解压tar xfz pycuda-VERSION.tar.gz##编译 VERSION为你自己的版本号cd pycuda-VERSIONpython configure.py##此时会生成siteconf.py配置文件pyth...原创 2019-05-10 11:28:10 · 2731 阅读 · 2 评论
分享