ImageUtils.cs 3.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108
  1. using OpenCvSharp;
  2. using OpenCvSharp.Extensions;
  3. using System;
  4. using System.Drawing;
  5. using System.IO;
  6. namespace DaJiaoYan.Utils
  7. {
  8. internal class ImageUtils
  9. {
  10. internal static Mat Read(string filename, ImreadModes mode = ImreadModes.AnyColor)
  11. {
  12. var res = Cv2.ImRead(filename, mode);
  13. if (res == null || res.Width < 1 || res.Height < 1)
  14. {
  15. // 判断图片异常
  16. try
  17. {
  18. Bitmap bitmap = new Bitmap(filename);
  19. res = ToMat(bitmap);
  20. }
  21. catch { }
  22. }
  23. return res;
  24. }
  25. /// <summary>
  26. /// image 转 mat
  27. /// </summary>
  28. /// <param name="image"></param>
  29. /// <returns></returns>
  30. internal static Mat ToMat(Image image)
  31. {
  32. Mat mat = null;
  33. if (image != null)
  34. {
  35. Bitmap bmp = new Bitmap(image);
  36. mat = BitmapConverter.ToMat(bmp);
  37. bmp.Dispose();
  38. }
  39. return mat;
  40. }
  41. /// <summary>
  42. /// 从 stream 转 mat
  43. /// </summary>
  44. /// <param name="stream"></param>
  45. /// <returns></returns>
  46. internal static Mat ToMat(Stream stream)
  47. {
  48. Mat mat = null;
  49. if (stream != null && stream.Length > 0)
  50. {
  51. mat = Mat.FromStream(stream, ImreadModes.AnyColor);
  52. }
  53. return mat;
  54. }
  55. /// <summary>
  56. /// 转Base64
  57. /// </summary>
  58. /// <param name="filename"></param>
  59. /// <returns></returns>
  60. internal static string ToBase64(string filename)
  61. {
  62. string res = "";
  63. try
  64. {
  65. using (FileStream fs = new FileStream(filename, FileMode.Open, FileAccess.Read))
  66. {
  67. byte[] buffer = new byte[fs.Length];
  68. fs.Read(buffer, 0, (int)fs.Length);
  69. res = Convert.ToBase64String(buffer);
  70. }
  71. }
  72. catch { }
  73. return res;
  74. }
  75. internal static void Red2OtherColor(ref Mat img, Scalar scalar, int minThreshold = 10, int maxThreshold = 40)
  76. {
  77. Cv2.CvtColor(img, img, ColorConversionCodes.BGR2HSV);
  78. Mat mask0 = new Mat(), mask1 = new Mat(), mask;
  79. Cv2.InRange(img,
  80. new Mat(1, 3, MatType.CV_32S, new Scalar(0, maxThreshold, maxThreshold)),
  81. new Mat(1, 3, MatType.CV_32S, new Scalar(minThreshold, 255, 255)),
  82. mask0);
  83. Cv2.InRange(img,
  84. new Mat(1, 3, MatType.CV_32S, new Scalar(150, maxThreshold, maxThreshold)),
  85. new Mat(1, 3, MatType.CV_32S, new Scalar(180, 255, 255)),
  86. mask1);
  87. mask = mask0 + mask1;
  88. Cv2.CvtColor(img, img, ColorConversionCodes.HSV2BGR);
  89. img.SetTo(scalar, mask);
  90. mask0?.Dispose();
  91. mask1?.Dispose();
  92. mask?.Dispose();
  93. }
  94. }
  95. }