Exemplo n.º 1
0
        /// <summary>
        /// use c# to do preprocess, the result will a little different from others
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void classifyPreBtn_Click(object sender, EventArgs e)
        {
            if (!netInited)
            {
                MessageBox.Show("Network is not inited");
                return;
            }

            if (pictureBox1.Image == null)
            {
                MessageBox.Show("there is no pic in picBox!");
                return;
            }

            MsnhnetDef.Dim dim    = net.GetInputDim();
            Bitmap         bitmap = (Bitmap)pictureBox1.Image;//image转bitmap
            Bitmap         outBitmap;

            ImgPro.ReSize(bitmap, dim.width, dim.height, out outBitmap, true);// a little different from opencv
            bitmap = ImgPro.ConvertTo24bpp(outBitmap);

            Rectangle rect = new Rectangle(0, 0, bitmap.Width, bitmap.Height);

            BitmapData bmpData = bitmap.LockBits(rect, ImageLockMode.ReadWrite, bitmap.PixelFormat);

            float[] inputData = new float[3 * bmpData.Width * bmpData.Height];

            unsafe
            {
                byte *ptr = (byte *)(bmpData.Scan0);

                for (int i = 0; i < bmpData.Height; i++)
                {
                    for (int k = 0; k < 3; ++k)
                    {
                        for (int j = 0; j < bmpData.Width; j++)
                        {
                            inputData[(k * bmpData.Width * bmpData.Height + i * bmpData.Width + j)] = ptr[i * 3 * bmpData.Width + j * 3 + k] / 255.0f;
                        }
                    }
                }
            }

            bitmap.UnlockBits(bmpData);

            int best = net.RunClassifyNoPred(inputData, false);

            string[] labelList = labels.Split('\n');

            // actually time need to add preprocess time , you can do it by your self
            float time = net.GetCpuForwardTime();

            richTextBox1.AppendText("CPU inference time:" + time.ToString() + " ms\n");

            richTextBox1.AppendText("CPU inferece result: " + labelList[best] + "\n");
        }
Exemplo n.º 2
0
 private void resetImgBtn_Click(object sender, EventArgs e)
 {
     try
     {
         pictureBox1.Image = Image.FromFile(savedImagePath);
         pictureBox1.Image = ImgPro.ConvertTo24bpp(pictureBox1.Image);
     }
     catch (Exception ex)
     {
         MessageBox.Show(ex.Message);
     }
 }
Exemplo n.º 3
0
        private void yoloGPUBtn_Click(object sender, EventArgs e)
        {
            if (!netInited)
            {
                MessageBox.Show("Network is not inited");
                return;
            }

            if (!Msnhnet.WithGPU())
            {
                MessageBox.Show("Msnhnet not build with GPU");
                return;
            }

            try
            {
                Rectangle rect = new Rectangle(0, 0, pictureBox1.Image.Width, pictureBox1.Image.Height); // image的长宽,非pictureBox的长宽

                Bitmap bitmap = (Bitmap)pictureBox1.Image;                                               //image转bitmap

                BitmapData bmpdata = bitmap.LockBits(rect, ImageLockMode.ReadWrite, bitmap.PixelFormat);

                List <Msnhnet.BBox> bboxes = net.RunYoloList(bmpdata, false, true);

                bitmap.UnlockBits(bmpdata);

                for (int i = 0; i < bboxes.Count; i++)
                {
                    bboxes[i] = ImgPro.bboxResize2Org(bboxes[i], net.GetInputDim().width, net.GetInputDim().height, bitmap.Width, bitmap.Height);
                }

                pictureBox1.Image = drawYolo(bitmap, bboxes, labels);

                float time = net.GetGpuForwardTime();

                richTextBox1.AppendText("GPU inference time:" + time.ToString() + " ms\n");
            }
            catch (Exception ex)
            {
                MessageBox.Show(ex.Message);
            }
        }
Exemplo n.º 4
0
        private void readImgBtn_Click(object sender, EventArgs e)
        {
            OpenFileDialog of = new OpenFileDialog();

            of.Filter = picOpenFilter;
            if (of.ShowDialog() == DialogResult.OK)
            {
                string temp = of.FileName;
                savedImagePath      = temp;
                resetImgBtn.Enabled = true;
                try
                {
                    pictureBox1.Image = Image.FromFile(temp);
                    pictureBox1.Image = ImgPro.ConvertTo24bpp(pictureBox1.Image);
                }
                catch (Exception ex)
                {
                    MessageBox.Show(ex.Message);
                }
            }
        }