$ hub install DriverStatusRecognition==1.0.0
$ hub run DriverStatusRecognition --image /PATH/TO/IMAGE --use_gpu True
def predict(data)
驾驶员状态识别预测接口,输入一张图像,输出该图像上驾驶员的状态。
参数
import cv2
import paddlehub as hub
module = hub.Module(directory='DriverStatusRecognition') # 一行代码实现模型调用
images = [cv2.imread('/PATH/TO/IMAGE')]
results = module.predict(images=images)
for result in results:
print(result)
paddlepaddle >= 2.0.0
paddlehub >= 2.0.0