FaceMaskDetection icon indicating copy to clipboard operation
FaceMaskDetection copied to clipboard

Published 20 hours ago • verified publisher icon AIZOOTech

python opencv_dnn_infer.py 是默认使用cpu还是gpu，想切换gpu要怎么办？

Open fumingren opened this issue 3 years ago • 5 comments

fumingren avatar Aug 18 '21 08:08 fumingren

默认CPU

xinnie-the-pooh avatar Aug 18 '21 08:08 xinnie-the-pooh

能换gpu吗

fumingren avatar Aug 18 '21 09:08 fumingren

这个得用对应版本cuda框架 但是程序貌似没有引用 （仅限于模型执行过程）

xinnie-the-pooh avatar Aug 18 '21 09:08 xinnie-the-pooh

好的谢谢哈

fumingren avatar Aug 18 '21 09:08 fumingren

可通过换下面代码解决换gpu问题 Net = cv2.dnn.readNetFromCaffe(args.proto, args.model) Net.setPreferableBackend(cv2.dnn.DNN_BACKEND_CUDA) Net.setPreferableTarget(cv2.dnn.DNN_TARGET_CUDA)

fumingren avatar Aug 19 '21 01:08 fumingren