add Net.Backend and Net.Target enums

shimat · shimat · commit e80a89a1a25d · 2019-02-03T07:58:29.000+09:00
diff --git a/samples b/samples
@@ -1 +1 @@
-Subproject commit 2c1941bb6eacaf4fc46b0e78b8c8b39407785559
+Subproject commit 6d0b58f2af5bd3dd364071d24165e2ed6b244f72
diff --git a/src/OpenCvSharp/Modules/dnn/Net.cs b/src/OpenCvSharp/Modules/dnn/Net.cs
@@ -337,21 +337,21 @@ public void SetHalideScheduler(string scheduler)
         /// Ask network to use specific computation backend where it supported.
         /// </summary>
         /// <param name="backendId">backend identifier.</param>
-        public void SetPreferableBackend(int backendId)
+        public void SetPreferableBackend(Backend backendId)
         {
             ThrowIfDisposed();
-            NativeMethods.dnn_Net_setPreferableBackend(ptr, backendId);
+            NativeMethods.dnn_Net_setPreferableBackend(ptr, (int)backendId);
             GC.KeepAlive(this);
         }
 
         /// <summary>
         /// Ask network to make computations on specific target device.
         /// </summary>
         /// <param name="targetId">target identifier.</param>
-        public void SetPreferableTarget(int targetId)
+        public void SetPreferableTarget(Target targetId)
         {
             ThrowIfDisposed();
-            NativeMethods.dnn_Net_setPreferableTarget(ptr, targetId);
+            NativeMethods.dnn_Net_setPreferableTarget(ptr, (int)targetId);
             GC.KeepAlive(this);
         }
 
@@ -451,5 +451,45 @@ public long GetPerfProfile(out double[] timings)
         }
 
         #endregion
+
+        #region Enum
+
+        /// <summary>
+        /// Enum of computation backends supported by layers.
+        /// </summary>
+        /// <remarks>
+        /// DNN_BACKEND_DEFAULT equals to DNN_BACKEND_INFERENCE_ENGINE if
+        /// OpenCV is built with Intel's Inference Engine library or 
+        /// DNN_BACKEND_OPENCV otherwise.
+        /// </remarks>
+        public enum Backend
+        {
+            //! DNN_BACKEND_DEFAULT equals to DNN_BACKEND_INFERENCE_ENGINE if
+            //! OpenCV is built with Intel's Inference Engine library or
+            //! DNN_BACKEND_OPENCV otherwise.
+#pragma warning disable CS1591
+            DEFAULT,
+            HALIDE,
+            INFERENCE_ENGINE,
+            OPENCV,
+            VKCOM
+#pragma warning restore CS1591
+        }
+
+        /// <summary>
+        /// Enum of target devices for computations.
+        /// </summary>
+        public enum Target
+        {
+#pragma warning disable CS1591
+            CPU,
+            OPENCL,
+            OPENCL_FP16,
+            MYRIAD,
+            VULKAN
+#pragma warning restore CS1591
+        }
+
+        #endregion
     }
 }
