C# emgu dnn 调用语义分割模型

记录下备忘

            Net net = DnnInvoke.ReadNetFromDarknet(@"J:\model.cfg", @"J:\model.weights");
    

            net.SetPreferableBackend(Emgu.CV.Dnn.Backend.Cuda);
            net.SetPreferableTarget(Target.CudaFp16);

            Microsoft.Win32.OpenFileDialog openFile = new Microsoft.Win32.OpenFileDialog();
            openFile.Filter = "图片格式|*.jpg;*.png;*.bmp";
            if (openFile.ShowDialog() != true) return;
            Image<Rgb, byte> img = new Image<Rgb, byte>(openFile.FileName);

            var size = new System.Drawing.Size(((int)(img.Width / 32d + 16) * 32), ((int)(img.Height / 32d + 16) * 32));
            var blob = DnnInvoke.BlobFromImage(img, 1 / 255d, size);
            net.SetInput(blob);

            var mat = net.Forward();
            int rows = mat.SizeOfDimension[2];
            int cols = mat.SizeOfDimension[3];
            int chhn = mat.SizeOfDimension[1];
            float* p = (float*)mat.DataPointer;
            Image<Rgb, byte> res = new Image<Rgb, byte>(cols, rows);

            for (int i = 0; i < rows; i++)
            {
                for (int j = 0; j < cols; j++)
                {
                    var v = p[i * cols + j] * 255 + 0.5;
                    v = v > 255 ? 255 : (v < 0 ? 0 : v);
                    res[i, j] = new Rgb(v, v, v);
                }
            }

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值