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

[fumingren]

[xinnie-the-pooh]

默认CPU

[fumingren]

能换gpu吗

[xinnie-the-pooh]

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

[fumingren]

好的谢谢哈

[fumingren]

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