dnn: add Vulkan device check for BackendRegistry

be6a837e · Wu Zhiwen · e1f5b886 · be6a837e
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

modules/dnn/src/dnn.cpp modules/dnn/src/dnn.cpp +2 -1

未找到文件。
--- a/modules/dnn/src/dnn.cpp
+++ b/modules/dnn/src/dnn.cpp
@@ -137,7 +137,8 @@ private:
        backends.push_back(std::make_pair(DNN_BACKEND_OPENCV, DNN_TARGET_CPU));

 #ifdef HAVE_VULKAN
-        backends.push_back(std::make_pair(DNN_BACKEND_VKCOM, DNN_TARGET_VULKAN));  // TODO Add device check
+        if (haveVulkan())
+            backends.push_back(std::make_pair(DNN_BACKEND_VKCOM, DNN_TARGET_VULKAN));
 #endif
    }
    static inline bool checkIETarget(int target)