Created by: lanzhou2012

欢迎您反馈PaddleHub使用问题，非常感谢您对PaddleHub的贡献！ 在留下您的问题时，辛苦您同步提供如下信息：

• 版本、环境信息 1）PaddleHub和PaddlePaddle版本： paddlehub 1.5.0 paddlepaddle-gpu 1.6.3.post97 2）系统环境：4.15.0-76-generic #86 (closed)~16.04.1-Ubuntu， python版本: Python 2.7.12 (default, Oct 8 2019, 14:14:10) [GCC 5.4.0 20160609] on linux2
• 复现信息：如为报错，请给出复现环境、复现步骤

核心代码:

cap = cv2.VideoCapture(0） success ,frame = cap.read() module = hub.Module(name="pyramidbox_lite_mobile_mask") input_dict = {"data": [frame ]} results = module.face_detection(data=input_dict, use_cuda=True, use_gpu=True, batch_size=1）

~/.bashrc 添加:

export CUDA_VISIBLE_DEVICES=0

执行命令

CUDA_VISIBLE_DEVICES='0' python paddle_mask.py

进程信息

nvidia-smi 0 27717 C python 3549MiB top 7995 root 20 0 24.060g 396740 182028 S 83.1 4.9 0:08.06 python

反馈

GPU推理时间 与 CPU (取消相关GPU设置)推理，做过对比， 推理时间相差无多，均在87ms~170ms波动。