111
This commit is contained in:
290
Check.Main/Common/LogoMatcher.cs
Normal file
290
Check.Main/Common/LogoMatcher.cs
Normal file
@@ -0,0 +1,290 @@
|
||||
//using HalconDotNet;
|
||||
//using System;
|
||||
//using System.Collections.Generic;
|
||||
//using System.Drawing.Imaging;
|
||||
//using System.IO;
|
||||
|
||||
//namespace HalconTemplateMatch
|
||||
//{
|
||||
// public class LogoMatcher
|
||||
// {
|
||||
// private List<HTuple> modelHandles = new List<HTuple>();
|
||||
|
||||
// /// <summary>
|
||||
// /// 从文件加载多个模板
|
||||
// /// </summary>
|
||||
// public void LoadTemplates(string dir)
|
||||
// {
|
||||
// foreach (var file in Directory.GetFiles(dir, "*.shm"))
|
||||
// {
|
||||
// HTuple modelID;
|
||||
// HOperatorSet.ReadShapeModel(file, out modelID);
|
||||
// modelHandles.Add(modelID);
|
||||
// Console.WriteLine($"加载模板: {file}");
|
||||
// }
|
||||
// }
|
||||
|
||||
// /// <summary>
|
||||
// /// 在测试图像中查找 Logo
|
||||
// /// </summary>
|
||||
// /// <returns>true = OK，false = NG</returns>
|
||||
// public bool FindLogo(string testImagePath)
|
||||
// {
|
||||
// HObject ho_TestImage;
|
||||
// HOperatorSet.ReadImage(out ho_TestImage, testImagePath);
|
||||
// HOperatorSet.Rgb1ToGray(ho_TestImage, out ho_TestImage);
|
||||
|
||||
// foreach (var modelID in modelHandles)
|
||||
// {
|
||||
// HOperatorSet.FindShapeModel(
|
||||
// ho_TestImage,
|
||||
// modelID,
|
||||
// new HTuple(0).TupleRad(),
|
||||
// new HTuple(360).TupleRad(),
|
||||
// 0.5, // 最低分数
|
||||
// 1, // 最大匹配数
|
||||
// 0.5, // 重叠度
|
||||
// "least_squares",
|
||||
// 0,
|
||||
// 0.9,
|
||||
// out HTuple row,
|
||||
// out HTuple col,
|
||||
// out HTuple angle,
|
||||
// out HTuple score);
|
||||
|
||||
// if (score.Length > 0 && score[0].D > 0.5)
|
||||
// {
|
||||
// Console.WriteLine($"找到 Logo: Row={row[0]}, Col={col[0]}, Score={score[0]}");
|
||||
// return true; // 找到即返回成功
|
||||
// }
|
||||
// }
|
||||
|
||||
// return false; // 没找到
|
||||
// }
|
||||
|
||||
|
||||
|
||||
// /// <summary>
|
||||
// /// 重载FindLogo函数（double返回）
|
||||
// /// </summary>
|
||||
// /// <param name="bmp"></param>
|
||||
// /// <returns></returns>
|
||||
// public double FindLogo(Bitmap bmp)
|
||||
// {
|
||||
// // Bitmap 转 HObject
|
||||
// HObject ho_TestImage;
|
||||
// Bitmap2HObject(bmp, out ho_TestImage);
|
||||
|
||||
// HOperatorSet.Rgb1ToGray(ho_TestImage, out ho_TestImage);
|
||||
|
||||
// double bestScore = -1;
|
||||
|
||||
// foreach (var modelID in modelHandles)
|
||||
// {
|
||||
// HOperatorSet.FindShapeModel(
|
||||
// ho_TestImage,
|
||||
// modelID,
|
||||
// new HTuple(0).TupleRad(),
|
||||
// new HTuple(360).TupleRad(),
|
||||
// 0.5, // 最低分数
|
||||
// 1, // 最大匹配数
|
||||
// 0.5, // 重叠度
|
||||
// "least_squares",
|
||||
// 0,
|
||||
// 0.9,
|
||||
// out HTuple row,
|
||||
// out HTuple col,
|
||||
// out HTuple angle,
|
||||
// out HTuple score);
|
||||
|
||||
// if (score.Length > 0 && score[0].D > bestScore)
|
||||
// {
|
||||
// bestScore = score[0].D;
|
||||
// }
|
||||
// }
|
||||
|
||||
// ho_TestImage.Dispose();
|
||||
// return bestScore; // -1 = 没找到
|
||||
// }
|
||||
|
||||
// /// <summary>
|
||||
// /// Bitmap 转 Halcon HObject
|
||||
// /// </summary>
|
||||
// private void Bitmap2HObject(Bitmap bmp, out HObject hobj)
|
||||
// {
|
||||
// HOperatorSet.GenEmptyObj(out hobj);
|
||||
// Rectangle rect = new Rectangle(0, 0, bmp.Width, bmp.Height);
|
||||
// BitmapData bmpData = bmp.LockBits(rect, ImageLockMode.ReadOnly, PixelFormat.Format24bppRgb);
|
||||
|
||||
// try
|
||||
// {
|
||||
// HOperatorSet.GenImageInterleaved(
|
||||
// out hobj,
|
||||
// bmpData.Scan0,
|
||||
// "bgr", // Bitmap 默认是 BGR
|
||||
// bmp.Width,
|
||||
// bmp.Height,
|
||||
// 0,
|
||||
// "byte",
|
||||
// bmp.Width,
|
||||
// bmp.Height,
|
||||
// 0,
|
||||
// 0,
|
||||
// -1,
|
||||
// 0
|
||||
// );
|
||||
// }
|
||||
// finally
|
||||
// {
|
||||
// bmp.UnlockBits(bmpData);
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
//}
|
||||
|
||||
|
||||
|
||||
using HalconDotNet;
|
||||
using NPOI.OpenXmlFormats.Vml;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Drawing;
|
||||
using System.Drawing.Imaging;
|
||||
using System.IO;
|
||||
|
||||
namespace HalconTemplateMatch
|
||||
{
|
||||
public class LogoMatcher
|
||||
{
|
||||
private readonly List<HTuple> modelHandles = new List<HTuple>();
|
||||
|
||||
/// <summary>
|
||||
/// 从指定目录加载所有 .shm 模板文件
|
||||
/// </summary>
|
||||
public void LoadTemplates(string dir)
|
||||
{
|
||||
try
|
||||
{
|
||||
string fullPath = Path.GetFullPath(dir);
|
||||
|
||||
if (!Directory.Exists(fullPath))
|
||||
{
|
||||
Console.WriteLine($"[警告] 模型目录不存在: {fullPath}");
|
||||
return;
|
||||
}
|
||||
|
||||
string[] modelFiles = Directory.GetFiles(fullPath, "*.shm", SearchOption.TopDirectoryOnly);
|
||||
|
||||
if (modelFiles.Length == 0)
|
||||
{
|
||||
Console.WriteLine($"[警告] 模型目录中没有任何 .shm 文件: {fullPath}");
|
||||
return;
|
||||
}
|
||||
|
||||
foreach (var file in modelFiles)
|
||||
{
|
||||
try
|
||||
{
|
||||
HTuple modelID;
|
||||
HOperatorSet.ReadShapeModel(file, out modelID);
|
||||
modelHandles.Add(modelID);
|
||||
Console.WriteLine($"[加载成功] 模板: {file}");
|
||||
}
|
||||
catch (HOperatorException ex)
|
||||
{
|
||||
Console.WriteLine($"[错误] 无法加载模板 {file}: {ex.Message}");
|
||||
}
|
||||
}
|
||||
|
||||
if (modelHandles.Count == 0)
|
||||
{
|
||||
Console.WriteLine($"[警告] 没有成功加载任何模板文件。");
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Console.WriteLine($"[异常] 加载模板目录出错: {ex.Message}");
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 匹配并返回最高得分（double返回）
|
||||
/// </summary>
|
||||
public double FindLogo(Bitmap bmp)
|
||||
{
|
||||
if (modelHandles.Count == 0)
|
||||
{
|
||||
Console.WriteLine("[警告] 尚未加载任何模板。");
|
||||
return -1;
|
||||
}
|
||||
|
||||
// Bitmap 转 Halcon 对象
|
||||
HObject ho_TestImage;
|
||||
Bitmap2HObject(bmp, out ho_TestImage);
|
||||
HOperatorSet.Rgb1ToGray(ho_TestImage, out ho_TestImage);
|
||||
|
||||
double bestScore = -1;
|
||||
|
||||
foreach (var modelID in modelHandles)
|
||||
{
|
||||
try
|
||||
{
|
||||
HOperatorSet.FindScaledShapeModel(
|
||||
ho_TestImage,
|
||||
modelID,
|
||||
new HTuple(0).TupleRad(),
|
||||
new HTuple(360).TupleRad(),
|
||||
0.8, 1.2,
|
||||
0.5, 1, 0.5,
|
||||
"least_squares_high",
|
||||
0, 0.9,
|
||||
out HTuple row, out HTuple col, out HTuple angle, out HTuple scale, out HTuple score
|
||||
);
|
||||
|
||||
|
||||
if (score.Length > 0 && score[0].D > bestScore)
|
||||
bestScore = score[0].D;
|
||||
}
|
||||
catch (HOperatorException ex)
|
||||
{
|
||||
Console.WriteLine($"[错误] 模板匹配失败: {ex.Message}");
|
||||
}
|
||||
}
|
||||
|
||||
ho_TestImage.Dispose();
|
||||
return bestScore;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Bitmap 转 Halcon HObject
|
||||
/// </summary>
|
||||
private void Bitmap2HObject(Bitmap bmp, out HObject hobj)
|
||||
{
|
||||
HOperatorSet.GenEmptyObj(out hobj);
|
||||
Rectangle rect = new Rectangle(0, 0, bmp.Width, bmp.Height);
|
||||
BitmapData bmpData = bmp.LockBits(rect, ImageLockMode.ReadOnly, PixelFormat.Format24bppRgb);
|
||||
|
||||
try
|
||||
{
|
||||
HOperatorSet.GenImageInterleaved(
|
||||
out hobj,
|
||||
bmpData.Scan0,
|
||||
"bgr",
|
||||
bmp.Width,
|
||||
bmp.Height,
|
||||
0,
|
||||
"byte",
|
||||
bmp.Width,
|
||||
bmp.Height,
|
||||
0,
|
||||
0,
|
||||
-1,
|
||||
0);
|
||||
}
|
||||
finally
|
||||
{
|
||||
bmp.UnlockBits(bmpData);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user