Exemplo n.º 1
0
        public static void Main()
        {
            try
            {
                //using var bitmap = (System.Drawing.Bitmap)System.Drawing.Image.FromFile(ImageName);
                using var image = Image.Load <Rgb24>(ImageName);
                Console.WriteLine($"image: {image}");
                if (image.TryGetSinglePixelSpan(out var pixelSpan))
                {
                    DumpVersion();

                    using var core = new InferenceEngineCore();

                    DumpCoreInformation(core);

                    using var network = new InferenceEngineNetwork(core, NetworkName);

                    DumpNetwork(network);

                    var mainInputName  = network.GetInputName(0);
                    var mainOutputName = network.GetOutputName(0);

                    network.SetInputResizeAlgorithm(mainInputName, resize_alg_e.RESIZE_BILINEAR);
                    network.SetInputLayout(mainInputName, layout_e.NCHW);
                    network.SetInputPrecision(mainInputName, precision_e.U8);

                    Console.WriteLine("Create executable network");
                    using var executableNetwork = new InferenceEngineExecutableNetwork(network, "GPU");

                    Console.WriteLine("Create request");
                    using var request = new InferenceEngineRequest(executableNetwork);

                    var imageDimensions   = new dimensions_t(1, 3, image.Height, image.Width);
                    var tensorDescription = new tensor_desc_t(layout_e.NHWC, imageDimensions, precision_e.U8);

                    Console.WriteLine("Create blob");
                    using var inputBlob = new Blob(tensorDescription, MemoryMarshal.Cast <Rgb24, byte>(pixelSpan));
                    request.SetBlob(mainInputName, inputBlob);

                    for (var i = 0; i < 10; i++)
                    {
                        Console.WriteLine($"Infer {i}");
                        request.Infer();
                        Console.WriteLine($"Infer {i} done");
                    }

                    using var outputBlob = request.GetBlob(mainOutputName);

                    Console.WriteLine($"Output blob. Sizes = {outputBlob.Size} {outputBlob.ByteSize}. [{outputBlob.Layout} {outputBlob.Precision}] {outputBlob.Dimensions}");
                }
            }
            catch (Exception ex)
            {
                Console.Error.WriteLine($"Error: {ex.Message}");
            }
        }
Exemplo n.º 2
0
            static Blob createBlob(int height, int lineWidth, int width, byte *data)
            {
                var dimensions = new dimensions_t(1, 1, height, lineWidth);
                var tensor     = new tensor_desc_t(layout_e.NHWC, dimensions, precision_e.U8);

                using var basicBlob = new Blob(tensor, new Span <byte>(data, height * lineWidth));
                var roi = new roi_t(1, 0, 0, width, height);

                return(new Blob(basicBlob, roi));
            }
Exemplo n.º 3
0
        private unsafe Blob ConvertAndInitialise(IntPtr memory, State state)
        {
            var pointer = memory.ToPointer();

            state.Decoder.Convert(state.Frame, (byte *)pointer);

            var dimensions = new dimensions_t(1, C, H, W);
            var tensor     = new tensor_desc_t(layout_e.NHWC, dimensions, precision_e.U8);
            var blob       = new Blob(tensor, new Span <byte>(pointer, FrameSize));

            return(blob);
        }