如何在 GPU 上沿着深度学习模型进行 cv2 操作？

Question

cv2 在对深度学习模型执行推理时使用 CPU 我想使用 gpu 对深度学习模型进行推理有什么办法吗？我尝试做 cv2.UMat() 但它仅在我开​​始使用深度学习时不使用深度学习时才支持，但它不起作用。请让我知道我们当时如何进行 cv2 操作和深度学习。

Answer 1

Tensorflow/Keras（带有 tensorflow 后端）保留整个 gpu 内存，即使它们只需要其中的一部分。如果可以只保留部分内容，您必须查看文档。