Opencv dnn 模块推理不使用 Jetson TX2 上的 GPU for SSD mobilenet

Question

我使用以下方法在 Jetson TX2 上使用 opencv3.4.1 和 Cuda 9.0 使用 SSD mobilenet (CAffe) 检测对象

blob = cv2.dnn.blobFromImage(cv2.resize(frame, (300, 300)), 0.007843, (300, 300), 127.5)
net.setInput(blob)
detections = net.forward()

tegrastats 显示根本没有 GPU 使用。CPU 的所有内核都得到了正确利用，但 GPU 利用率为零。

如何在 Jetson 上启用 GPU 以提高视频的 FPS？

谢谢

Answer 1

OpenCV DNN 只能使用 Intel GPU 进行 OpenCL 推理。该检查放在 OpenCV 源代码中。