C# OpenCvSharp DNN 黑白老照片上色

C# OpenCvSharp DNN 黑白老照片上色

目录

效果

项目

代码

下载 

参考

效果

项目

代码

using OpenCvSharp;
using OpenCvSharp.Extensions;
using System;
using System.Diagnostics;
using System.Drawing;
using System.Drawing.Imaging;
using System.Runtime.InteropServices;
using System.Windows.Forms;

namespace OpenCvSharp_Demo
{
    public partial class Form1 : Form
    {
        public Form1()
        {
            InitializeComponent();
        }

        string fileFilter = "*.*|*.bmp;*.jpg;*.jpeg;*.tiff;*.tiff;*.png";
        string startupPath;
        string image_path;

        Stopwatch stopwatch = new Stopwatch();

        Mat image;
        Mat result_image;

        /*
         // 初始化
        extern "C" _declspec(dllexport) void* __cdecl init(char* prototxt, char* caffe_model,int* res);

        // 上色
        extern "C" _declspec(dllexport) int __cdecl colorization(void* CvDNN_Net,Mat* image, Mat** returnValue);
         
         */

        const string DllName = "ColorizationSharp.dll";

        [DllImport(DllName, EntryPoint = "init", CallingConvention = CallingConvention.Cdecl)]
        public extern static IntPtr init(string prototxt, string caffe_model, ref int res);

        [DllImport(DllName, EntryPoint = "colorization", CallingConvention = CallingConvention.Cdecl)]
        public extern static int colorization(IntPtr CvDNN_Net, IntPtr image, out IntPtr returnValue);

        IntPtr CvDNN_Net;

        private void Form1_Load(object sender, EventArgs e)
        {
            startupPath = System.Windows.Forms.Application.StartupPath;

            image_path = "images/1.jpg";
            pictureBox1.Image = new Bitmap(image_path);
            image = new Mat(image_path);

            string prototxt = startupPath + "\\model\\colorization_deploy_v2.prototxt";
            string caffe_model = startupPath + "\\model\\colorization_release_v2.caffemodel";
            int res = 0;
            CvDNN_Net = init(prototxt, caffe_model, ref res);
            if (res != 0)
            {
                MessageBox.Show("初始化失败！");
            }
            else
            {
                button2.Enabled = true;
            }
        }

        private void button1_Click(object sender, EventArgs e)
        {
            OpenFileDialog ofd = new OpenFileDialog();
            ofd.Filter = fileFilter;
            if (ofd.ShowDialog() != DialogResult.OK) return;

            pictureBox1.Image = null;
            pictureBox2.Image = null;
            textBox1.Text = "";

            image_path = ofd.FileName;
            pictureBox1.Image = new Bitmap(image_path);
            image = new Mat(image_path);

        }

        private void button2_Click(object sender, EventArgs e)
        {
            if (image_path == "")
            {
                return;
            }

            textBox1.Text = "";
            pictureBox2.Image = null;
            button2.Enabled = false;

            Application.DoEvents();

            stopwatch.Restart();

            IntPtr returnValue = IntPtr.Zero;
            int res = colorization(CvDNN_Net, image.Clone().CvPtr, out returnValue);
            if (res == 0)
            {
                result_image = new Mat(returnValue);

                double costTime = stopwatch.Elapsed.TotalMilliseconds;

                textBox1.Text = $"耗时:{costTime:F2}ms";
                pictureBox2.Image = result_image.ToBitmap();
            }
            else
            {
                MessageBox.Show("代码异常，上色失败！");
            }
            button2.Enabled = true;

        }

        private void button3_Click(object sender, EventArgs e)
        {
            if (pictureBox2.Image == null)
            {
                return;
            }
            Bitmap output = new Bitmap(pictureBox2.Image);
            var sdf = new SaveFileDialog();
            sdf.Title = "保存";
            sdf.Filter = "Images (*.jpg)|*.jpg|Images (*.png)|*.png|Images (*.bmp)|*.bmp|Images (*.emf)|*.emf|Images (*.exif)|*.exif|Images (*.gif)|*.gif|Images (*.ico)|*.ico|Images (*.tiff)|*.tiff|Images (*.wmf)|*.wmf";
            if (sdf.ShowDialog() == DialogResult.OK)
            {
                switch (sdf.FilterIndex)
                {
                    case 1:
                        {
                            output.Save(sdf.FileName, ImageFormat.Jpeg);
                            break;
                        }
                    case 2:
                        {
                            output.Save(sdf.FileName, ImageFormat.Png);
                            break;
                        }
                    case 3:
                        {
                            output.Save(sdf.FileName, ImageFormat.Bmp);
                            break;
                        }
                    case 4:
                        {
                            output.Save(sdf.FileName, ImageFormat.Emf);
                            break;
                        }
                    case 5:
                        {
                            output.Save(sdf.FileName, ImageFormat.Exif);
                            break;
                        }
                    case 6:
                        {
                            output.Save(sdf.FileName, ImageFormat.Gif);
                            break;
                        }
                    case 7:
                        {
                            output.Save(sdf.FileName, ImageFormat.Icon);
                            break;
                        }
                    case 8:
                        {
                            output.Save(sdf.FileName, ImageFormat.Tiff);
                            break;
                        }
                    case 9:
                        {
                            output.Save(sdf.FileName, ImageFormat.Wmf);
                            break;
                        }
                }
                MessageBox.Show("保存成功，位置：" + sdf.FileName);
            }
        }

    }
}

  1. using OpenCvSharp;
  2. using OpenCvSharp.Extensions;
  3. using System;
  4. using System.Diagnostics;
  5. using System.Drawing;
  6. using System.Drawing.Imaging;
  7. using System.Runtime.InteropServices;
  8. using System.Windows.Forms;
  9.  
  10. namespace OpenCvSharp_Demo
  11. {
  12.     public partial class Form1 : Form
  13.     {
  14.         public Form1()
  15.         {
  16.             InitializeComponent();
  17.         }
  18.  
  19.         string fileFilter = "*.*|*.bmp;*.jpg;*.jpeg;*.tiff;*.tiff;*.png";
  20.         string startupPath;
  21.         string image_path;
  22.  
  23.         Stopwatch stopwatch = new Stopwatch();
  24.  
  25.         Mat image;
  26.         Mat result_image;
  27.  
  28.         /*
  29.          // 初始化
  30.         extern "C" _declspec(dllexport) void* __cdecl init(char* prototxt, char* caffe_model,int* res);
  31.  
  32.         // 上色
  33.         extern "C" _declspec(dllexport) int __cdecl colorization(void* CvDNN_Net,Mat* image, Mat** returnValue);
  34.          
  35.          */
  36.  
  37.         const string DllName = "ColorizationSharp.dll";
  38.  
  39.         [DllImport(DllName, EntryPoint = "init", CallingConvention = CallingConvention.Cdecl)]
  40.         public extern static IntPtr init(string prototxt, string caffe_model, ref int res);
  41.  
  42.         [DllImport(DllName, EntryPoint = "colorization", CallingConvention = CallingConvention.Cdecl)]
  43.         public extern static int colorization(IntPtr CvDNN_Net, IntPtr image, out IntPtr returnValue);
  44.  
  45.         IntPtr CvDNN_Net;
  46.  
  47.         private void Form1_Load(object sender, EventArgs e)
  48.         {
  49.             startupPath = System.Windows.Forms.Application.StartupPath;
  50.  
  51.             image_path = "images/1.jpg";
  52.             pictureBox1.Image = new Bitmap(image_path);
  53.             image = new Mat(image_path);
  54.  
  55.             string prototxt = startupPath + "\\model\\colorization_deploy_v2.prototxt";
  56.             string caffe_model = startupPath + "\\model\\colorization_release_v2.caffemodel";
  57.             int res = 0;
  58.             CvDNN_Net = init(prototxt, caffe_model, ref res);
  59.             if (res != 0)
  60.             {
  61.                 MessageBox.Show("初始化失败！");
  62.             }
  63.             else
  64.             {
  65.                 button2.Enabled = true;
  66.             }
  67.         }
  68.  
  69.         private void button1_Click(object sender, EventArgs e)
  70.         {
  71.             OpenFileDialog ofd = new OpenFileDialog();
  72.             ofd.Filter = fileFilter;
  73.             if (ofd.ShowDialog() != DialogResult.OK) return;
  74.  
  75.             pictureBox1.Image = null;
  76.             pictureBox2.Image = null;
  77.             textBox1.Text = "";
  78.  
  79.             image_path = ofd.FileName;
  80.             pictureBox1.Image = new Bitmap(image_path);
  81.             image = new Mat(image_path);
  82.  
  83.         }
  84.  
  85.         private void button2_Click(object sender, EventArgs e)
  86.         {
  87.             if (image_path == "")
  88.             {
  89.                 return;
  90.             }
  91.  
  92.             textBox1.Text = "";
  93.             pictureBox2.Image = null;
  94.             button2.Enabled = false;
  95.  
  96.             Application.DoEvents();
  97.  
  98.             stopwatch.Restart();
  99.  
  100.             IntPtr returnValue = IntPtr.Zero;
  101.             int res = colorization(CvDNN_Net, image.Clone().CvPtr, out returnValue);
  102.             if (res == 0)
  103.             {
  104.                 result_image = new Mat(returnValue);
  105.  
  106.                 double costTime = stopwatch.Elapsed.TotalMilliseconds;
  107.  
  108.                 textBox1.Text = $"耗时:{costTime:F2}ms";
  109.                 pictureBox2.Image = result_image.ToBitmap();
  110.             }
  111.             else
  112.             {
  113.                 MessageBox.Show("代码异常，上色失败！");
  114.             }
  115.             button2.Enabled = true;
  116.  
  117.         }
  118.  
  119.         private void button3_Click(object sender, EventArgs e)
  120.         {
  121.             if (pictureBox2.Image == null)
  122.             {
  123.                 return;
  124.             }
  125.             Bitmap output = new Bitmap(pictureBox2.Image);
  126.             var sdf = new SaveFileDialog();
  127.             sdf.Title = "保存";
  128.             sdf.Filter = "Images (*.jpg)|*.jpg|Images (*.png)|*.png|Images (*.bmp)|*.bmp|Images (*.emf)|*.emf|Images (*.exif)|*.exif|Images (*.gif)|*.gif|Images (*.ico)|*.ico|Images (*.tiff)|*.tiff|Images (*.wmf)|*.wmf";
  129.             if (sdf.ShowDialog() == DialogResult.OK)
  130.             {
  131.                 switch (sdf.FilterIndex)
  132.                 {
  133.                     case 1:
  134.                         {
  135.                             output.Save(sdf.FileName, ImageFormat.Jpeg);
  136.                             break;
  137.                         }
  138.                     case 2:
  139.                         {
  140.                             output.Save(sdf.FileName, ImageFormat.Png);
  141.                             break;
  142.                         }
  143.                     case 3:
  144.                         {
  145.                             output.Save(sdf.FileName, ImageFormat.Bmp);
  146.                             break;
  147.                         }
  148.                     case 4:
  149.                         {
  150.                             output.Save(sdf.FileName, ImageFormat.Emf);
  151.                             break;
  152.                         }
  153.                     case 5:
  154.                         {
  155.                             output.Save(sdf.FileName, ImageFormat.Exif);
  156.                             break;
  157.                         }
  158.                     case 6:
  159.                         {
  160.                             output.Save(sdf.FileName, ImageFormat.Gif);
  161.                             break;
  162.                         }
  163.                     case 7:
  164.                         {
  165.                             output.Save(sdf.FileName, ImageFormat.Icon);
  166.                             break;
  167.                         }
  168.                     case 8:
  169.                         {
  170.                             output.Save(sdf.FileName, ImageFormat.Tiff);
  171.                             break;
  172.                         }
  173.                     case 9:
  174.                         {
  175.                             output.Save(sdf.FileName, ImageFormat.Wmf);
  176.                             break;
  177.                         }
  178.                 }
  179.                 MessageBox.Show("保存成功，位置：" + sdf.FileName);
  180.             }
  181.         }
  182.  
  183.     }
  184. }

下载 

源码下载

参考

https://github.com/richzhang/colorization/tree/caffe

天天代码码天天
微信公众号
.NET 人工智能实践
注：本文转载自blog.csdn.net的天天代码码天天的文章"https://lw112190.blog.csdn.net/article/details/138863488"。版权归原作者所有，此博客不拥有其著作权，亦不承担相应法律责任。如有侵权，请联系我们删除。
复制链接
相关推荐
发表评论

评论记录：

未查询到任何数据！