xiaohuimin/DHDHSoftware
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

保存图片有问题

Browse Source
This commit is contained in:
TD
2025-04-02 18:26:34 +08:00
parent babc40d36a
commit d63a6c42b3
6 changed files with 217 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files

219
DH.Devices.Vision/SimboVisionDriver.cs
View File

@ -8,6 +8,8 @@ using OpenCvSharp.Extensions;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Diagnostics.Eventing.Reader;
using System.Drawing.Imaging;
using System.Linq;
using System.Runtime.ExceptionServices;
using System.Text;
@ -76,12 +78,13 @@ namespace DH.Devices.Vision
//未能获得检测配置
return detectResult;
}
detectResult.DetectName = detectConfig.Name;
detectResult.ImageSaveDirectory=detectConfig.ImageSaveDirectory;
detectResult.SaveNGDetect=detectConfig.SaveNGDetect;
detectResult.SaveNGOriginal=detectConfig.SaveNGOriginal;
detectResult.SaveOKDetect=detectConfig.SaveOKDetect;
detectResult.SaveOKOriginal=detectConfig.SaveOKOriginal;
detectResult.DetectionOriginImage = originImgSet.Clone().ToBitmap();
//detectResult.DetectionOriginImage = originImgSet.Clone().ToBitmap();
Stopwatch sw = new Stopwatch();
#region 1.预处理
sw.Start();
@ -263,22 +266,22 @@ namespace DH.Devices.Vision
}
foreach (IGrouping<ResultState, DetectionFilter> group in conditionList)
{
bool b = group.ToList().Any(f =>
{
return f.FilterOperation(d);
});
//foreach (IGrouping<ResultState, DetectionFilter> group in conditionList)
//{
// bool b = group.ToList().Any(f =>
// {
// return f.FilterOperation(d);
// });
if (b)
{
d.FinalResult = group.Key;
break;
}
// if (b)
// {
// d.FinalResult = group.Key;
// break;
// }
}
//}
});
#endregion
#region 5.统计缺陷过滤结果或预处理直接NG
@ -455,9 +458,9 @@ namespace DH.Devices.Vision
}
catch (Exception ex)
{
LogAsync(DateTime.Now, LogLevel.Error, $"模型加载成功；是否GPU:{isGPU} CoreInx:{coreInx} - {dc.Name}" + $" {dc.ModelType.GetEnumDescription()}:{dc.ModelPath}");
// LogAsync(DateTime.Now, LogLevel.Error, $"模型加载成功；是否GPU:{isGPU} CoreInx:{coreInx} - {dc.Name}" + $" {dc.ModelType.GetEnumDescription()}:{dc.ModelPath}");
//throw new ProcessException($"异常：是否GPU:{isGPU} CoreInx:{coreInx} - {dc.Name}模型加载异常：{ex.GetExceptionMessage()}");
throw new ProcessException($"异常：是否GPU:{isGPU} CoreInx:{coreInx} - {dc.Name}模型加载异常：{ex.GetExceptionMessage()}");
}
return mLEngineSet;
}
@ -1000,14 +1003,14 @@ namespace DH.Devices.Vision
List<IShapeElement> detectionResultShapesClone = new List<IShapeElement>(detectionResultShapes);
DetectionDone(DetectionId, resultMask, detectionResultShapes);
detectResult.DetectionOriginImage = CopyBitmapWithLockBits(resultMask);
SaveDetectResultImageAsync(detectResult);
// SaveDetectResultCSVAsync(detectResult);
}
catch (Exception ex)
{
LogAsync(DateTime.Now, LogLevel.Exception,
$"{Name}显示{detectResult.DetectName}检测结果异常,{ex.GetExceptionMessage()}");
//LogAsync(DateTime.Now, LogLevel.Exception,
// $"{Name}显示{detectResult.DetectName}检测结果异常,{ex.GetExceptionMessage()}");
}
finally
{
@ -1016,17 +1019,58 @@ namespace DH.Devices.Vision
}
});
}
public static Bitmap CopyBitmapWithLockBits(Bitmap original)
{
Bitmap copy = new Bitmap(original.Width, original.Height, original.PixelFormat);
BitmapData originalData = original.LockBits(
new Rectangle(0, 0, original.Width, original.Height),
ImageLockMode.ReadOnly,
original.PixelFormat
);
BitmapData copyData = copy.LockBits(
new Rectangle(0, 0, copy.Width, copy.Height),
ImageLockMode.WriteOnly,
copy.PixelFormat
);
int bytesPerPixel = Image.GetPixelFormatSize(original.PixelFormat) / 8;
int byteCount = originalData.Stride * original.Height;
byte[] buffer = new byte[byteCount];
System.Runtime.InteropServices.Marshal.Copy(originalData.Scan0, buffer, 0, byteCount);
System.Runtime.InteropServices.Marshal.Copy(buffer, 0, copyData.Scan0, byteCount);
original.UnlockBits(originalData);
copy.UnlockBits(copyData);
return copy;
}
/// <summary>
///图片异步保存
/// </summary>
public void SaveDetectResultImageAsync(DetectStationResult detectResult)
public void SaveDetectResultImageAsync1(DetectStationResult detectResult)
{
string format = detectResult.ImageFormat.ToString().ToLower();
if(detectResult.ImageSaveDirectory!=null)
{
if (!Directory.Exists(detectResult.ImageSaveDirectory))
{
}
else
{
}
}else
{
return;
}
//根目录
string rootPath = Path.Combine(detectResult.ImageSaveDirectory,
DateTime.Now.ToString("yyyyMMdd"), BatchNO, detectResult.DetectName);
DateTime.Now.ToString("yyyyMMdd"), detectResult.DetectName);
if (detectResult.ResultState != ResultState.OK)
{
@ -1078,12 +1122,21 @@ namespace DH.Devices.Vision
Bitmap detectionFitImage = StaticHelper.HConnectBitmap(preTreatedBitmap, resultMap);
string prefix = Path.Combine(rootPath, "NGFitImages", detectResult.ResultLabel);
Directory.CreateDirectory(prefix); // 自动创建所有缺失的目录
string fullname = Path.Combine(prefix, $"{detectResult.Pid}_NGFitImage_{detectResult.DetectName}_{detectResult.Id}.{format}");
SaveImageAsync(fullname, detectionFitImage, detectResult.ImageFormat);
// SaveImageAsync(fullname, detectionFitImage, detectResult.ImageFormat);
// 使用回调或 Task.ContinueWith 确保保存完成后再释放资源
//SaveImageAsync(fullname, detectionFitImage, detectResult.ImageFormat)
// .ContinueWith(t =>
// {
// resultMask?.Dispose();
// preTreatedBitmap?.Dispose();
// resultMap?.Dispose();
// detectionFitImage?.Dispose();
// }, TaskScheduler.Default);
resultMask?.Dispose();
preTreatedBitmap?.Dispose();
@ -1134,8 +1187,7 @@ namespace DH.Devices.Vision
Bitmap resultMap = GetResultImage(resultMask, detectionResultShapes);
resultDisplay.Dispose();
detectionResultShapes.Clear();
Bitmap detectionFitImage = StaticHelper.HConnectBitmap(preTreatedBitmap, resultMap);
@ -1148,11 +1200,12 @@ namespace DH.Devices.Vision
SaveImageAsync(fullname, detectionFitImage, detectResult.ImageFormat);
resultMask?.Dispose();
preTreatedBitmap?.Dispose();
resultMap?.Dispose();
detectionFitImage?.Dispose();
//resultDisplay.Dispose();
//detectionResultShapes.Clear();
//resultMask?.Dispose();
//preTreatedBitmap?.Dispose();
//resultMap?.Dispose();
//detectionFitImage?.Dispose();
}
@ -1163,6 +1216,7 @@ namespace DH.Devices.Vision
}
public virtual Bitmap GetResultImage(Bitmap baseImage, List<IShapeElement> eleList)
{
try
@ -1189,5 +1243,112 @@ namespace DH.Devices.Vision
}
}
public void SaveDetectResultImageAsync(DetectStationResult detectResult)
{
if (detectResult.ImageSaveDirectory == null) return;
string format = detectResult.ImageFormat.ToString().ToLower();
string rootPath = Path.Combine(detectResult.ImageSaveDirectory,
DateTime.Now.ToString("yyyyMMdd"), detectResult.DetectName);
try
{
if (detectResult.ResultState != ResultState.OK)
{
SaveNGImages(detectResult, rootPath, format);
}
else
{
SaveOKImages(detectResult, rootPath, format);
}
}
catch (Exception ex)
{
// Logger.Error($"保存检测结果失败: {ex.Message}");
}
}
private void SaveNGImages(DetectStationResult result, string rootPath, string format)
{
// NG 原图
if (result.SaveNGOriginal && result.DetectionOriginImage != null)
{
string prefix = Path.Combine(rootPath, "NGRawImages", result.ResultLabel);
Directory.CreateDirectory(prefix);
string fullname = Path.Combine(prefix, $"{result.Pid}_NGRawImage_{result.DetectName}_{result.Id}.{format}");
SaveImageAsync(fullname, result.DetectionOriginImage, result.ImageFormat);
}
// NG 结果图
if (result.SaveOKDetect && result.DetectionOriginImage != null)
{
string displayTxt = BuildDisplayText(result);
using (Bitmap resultMask = result.DetectionOriginImage.CopyBitmap())
using (Bitmap preTreatedBitmap = result.DetectionOriginImage.CopyBitmap())
{
var detectionResultShapes = new List<IShapeElement>(result.DetectionResultShapes)
{
new DetectResultDisplay(result, resultMask, displayTxt)
};
using (Bitmap resultMap = GetResultImage(resultMask, detectionResultShapes))
using (Bitmap detectionFitImage = StaticHelper.HConnectBitmap(preTreatedBitmap, resultMap))
{
string prefix = Path.Combine(rootPath, "NGFitImages", result.ResultLabel);
Directory.CreateDirectory(prefix);
string fullname = Path.Combine(prefix, $"{result.Pid}_NGFitImage_{result.DetectName}_{result.Id}.{format}");
SaveImageAsync(fullname, detectionFitImage, result.ImageFormat);
}
}
}
}
private void SaveOKImages(DetectStationResult result, string rootPath, string format)
{
// OK 原图
if (result.SaveOKOriginal && result.DetectionOriginImage != null)
{
string prefix = Path.Combine(rootPath, "OKRawImages", result.ResultLabel);
Directory.CreateDirectory(prefix);
string fullname = Path.Combine(prefix, $"{result.Pid}_OKRawImage_{result.DetectName}_{result.Id}.{format}");
SaveImageAsync(fullname, result.DetectionOriginImage, result.ImageFormat);
}
// OK 结果图
if (result.SaveOKDetect && result.DetectionOriginImage != null)
{
string displayTxt = BuildDisplayText(result);
using (Bitmap resultMask = result.DetectionOriginImage.CopyBitmap())
using (Bitmap preTreatedBitmap = result.DetectionOriginImage.CopyBitmap())
{
var detectionResultShapes = new List<IShapeElement>(result.DetectionResultShapes)
{
new DetectResultDisplay(result, resultMask, displayTxt)
};
using (Bitmap resultMap = GetResultImage(resultMask, detectionResultShapes))
using (Bitmap detectionFitImage = StaticHelper.HConnectBitmap(preTreatedBitmap, resultMap))
{
string prefix = Path.Combine(rootPath, "OKFitImages", result.ResultLabel);
Directory.CreateDirectory(prefix);
string fullname = Path.Combine(prefix, $"{result.Pid}_OKFitImage_{result.DetectName}_{result.Id}.{format}");
SaveImageAsync(fullname, result.DetectionOriginImage, result.ImageFormat);
}
}
}
}
private string BuildDisplayText(DetectStationResult result)
{
StringBuilder sb = new StringBuilder();
result.DetectDetails.ForEach(d => sb.AppendLine($"{d.LabelName} score:{d.Score:f2} area:{d.Area:f2}"));
if (result.realSpecs?.Count > 0)
{
result.realSpecs.ForEach(d => sb.AppendLine($"{d.Code} score:{d.ActualValue}"));
}
return sb.ToString();
}
}
}