using System;
|
using System.Collections.Generic;
|
using System.IO;
|
using System.Linq;
|
using System.Text;
|
using System.Threading.Tasks;
|
using System.Windows;
|
using System.Windows.Controls;
|
using System.Windows.Data;
|
using System.Windows.Documents;
|
using System.Windows.Input;
|
using System.Windows.Media;
|
using System.Windows.Media.Imaging;
|
using System.Windows.Navigation;
|
using System.Windows.Shapes;
|
using System.Xml;
|
using Emgu.CV;
|
using RichCreator.Editor.Utils;
|
using ZTImage;
|
|
namespace RichCreator.Editor.Tools.CV
|
{
|
/// <summary>
|
/// MainWindow.xaml 的交互逻辑
|
/// </summary>
|
public partial class MainCV : Window
|
{
|
public string XmlFile { get; set; }
|
|
public string ImageFile { get; set; }
|
|
public Int32 WinWidth { get; set; }
|
|
public Int32 WinHeight { get; set; }
|
|
|
public Int32 BlockWidth { get; set; }
|
|
|
public Int32 BlockHeight { get; set; }
|
|
public Int32 StrideWidth { get; set; }
|
|
public Int32 StrideHeight { get; set; }
|
|
public Int32 CellWidth { get; set; }
|
|
public Int32 CellHeight { get; set; }
|
|
public Int32 ChannelNumber { get; set; }
|
|
|
public MainCV()
|
{
|
InitializeComponent();
|
//detector = new HOGDetector();
|
}
|
|
/// <summary>
|
/// 开始识别
|
/// </summary>
|
/// <param name="sender"></param>
|
/// <param name="e"></param>
|
private void Detector_Click(object sender, RoutedEventArgs e)
|
{
|
if (!CheckAndFillAllParameter())
|
{
|
return;
|
}
|
|
|
HOGDetector detector = new HOGDetector(
|
new System.Drawing.Size(this.WinWidth, this.WinHeight),
|
new System.Drawing.Size(this.BlockWidth, this.BlockHeight),
|
new System.Drawing.Size(this.StrideWidth, this.StrideHeight),
|
new System.Drawing.Size(this.CellWidth, this.CellHeight),
|
this.ChannelNumber,
|
this.XmlFile
|
);
|
|
|
//ZTRect[] regions = HOGDetector.DetectorPeople(this.ImageFile);
|
ZTRect[] regions = detector.DetectorImage(this.ImageFile);
|
if (regions.Length <= 0)
|
{
|
MessageBox.Show("没有识别到");
|
return;
|
}
|
|
DrawImageAndRegion(this.ImageFile, regions);
|
}
|
|
|
/// <summary>
|
/// 检查并填充参数
|
/// </summary>
|
/// <returns></returns>
|
private Boolean CheckAndFillAllParameter()
|
{
|
this.XmlFile = this.txtXmlFile.Text.Trim();
|
if (string.IsNullOrEmpty(this.XmlFile))
|
{
|
ShowMessage("请填写结果Xml名");
|
return false;
|
}
|
|
if (!System.IO.File.Exists(this.XmlFile))
|
{
|
MessageBox.Show("xml不存在", "提示");
|
return false;
|
}
|
|
this.ImageFile = this.txtImgFile.Text.Trim();
|
if (string.IsNullOrEmpty(this.ImageFile))
|
{
|
ShowMessage("请选择要识别的图像");
|
return false;
|
}
|
|
if (!System.IO.File.Exists(this.ImageFile))
|
{
|
MessageBox.Show("Image不存在", "提示");
|
return false;
|
}
|
|
|
|
this.WinWidth = TypeConverter.StringToInt(this.txtWinWidth.Text, 0);
|
if (this.WinWidth <= 0)
|
{
|
ShowMessage("取样宽不合格");
|
return false;
|
}
|
|
this.WinHeight = TypeConverter.StringToInt(this.txtWinHeight.Text, 0);
|
if (this.WinHeight <= 0)
|
{
|
ShowMessage("取样高不合格");
|
return false;
|
}
|
|
|
|
|
this.BlockWidth = TypeConverter.StringToInt(this.txtBlockWidth.Text, 0);
|
if (this.BlockWidth <= 0 || (this.WinWidth % this.BlockWidth) != 0)
|
{
|
ShowMessage("Block宽不合格");
|
return false;
|
}
|
|
this.BlockHeight = TypeConverter.StringToInt(this.txtBlockHeight.Text, 0);
|
if (this.BlockHeight <= 0 || (this.WinHeight % this.BlockHeight) != 0)
|
{
|
ShowMessage("Block高不合格");
|
return false;
|
}
|
|
|
this.StrideWidth = TypeConverter.StringToInt(this.txtStrideWidth.Text, 0);
|
if (this.StrideWidth <= 0)
|
{
|
ShowMessage("Stride宽不合格");
|
return false;
|
}
|
|
this.StrideHeight = TypeConverter.StringToInt(this.txtStrideHeight.Text, 0);
|
if (this.StrideHeight <= 0)
|
{
|
ShowMessage("Stride高不合格");
|
return false;
|
}
|
|
|
this.CellWidth = TypeConverter.StringToInt(this.txtCellWidth.Text, 0);
|
if (this.CellWidth <= 0 || this.CellWidth * 2 != this.BlockWidth)
|
{
|
ShowMessage("Cell宽不合格");
|
return false;
|
}
|
|
|
this.CellHeight = TypeConverter.StringToInt(this.txtCellHeight.Text, 0);
|
if (this.CellHeight <= 0 || this.CellHeight * 2 != this.BlockHeight)
|
{
|
ShowMessage("Cell高不合格");
|
return false;
|
}
|
|
this.ChannelNumber = TypeConverter.StringToInt(this.txtChannelNumber.Text, 0);
|
if (this.ChannelNumber <= 0)
|
{
|
ShowMessage("channel不合格");
|
return false;
|
}
|
|
return true;
|
}
|
|
|
private void ShowMessage(string msg)
|
{
|
MessageBox.Show(msg);
|
}
|
|
/// <summary>
|
/// 打开训练工具
|
/// </summary>
|
/// <param name="sender"></param>
|
/// <param name="e"></param>
|
private void Train_Click(object sender, RoutedEventArgs e)
|
{
|
Trainor trainor = new Trainor();
|
trainor.ShowDialog();
|
}
|
|
/// <summary>
|
/// 绘制图像和区域
|
/// </summary>
|
/// <param name="imagePath"></param>
|
/// <param name="regions"></param>
|
private void DrawImageAndRegion(string imagePath, ZTRect[] regions)
|
{
|
byte[] imageData = File.ReadAllBytes(imagePath);
|
MemoryStream ms = new MemoryStream(imageData);
|
this.Dispatcher.BeginInvoke((Action)(() =>
|
{
|
this.container.Children.Clear();
|
////画图
|
Image imgControl = new Image();
|
imgControl.Source = BitmapFrame.Create(ms);
|
this.container.Children.Add(imgControl);
|
|
//画区域
|
for (int i = 0; i < regions.Length; i++)
|
{
|
CreatePathControl(this.container, regions[i]);
|
}
|
}));
|
}
|
|
/// <summary>
|
/// 生成矩形框
|
/// </summary>
|
/// <param name="rect"></param>
|
/// <returns></returns>
|
private System.Windows.Shapes.Rectangle CreatePathControl(Canvas container, ZTRect rect)
|
{
|
|
Grid grid = new Grid();
|
|
System.Windows.Shapes.Rectangle rectControl = new System.Windows.Shapes.Rectangle();
|
grid.Children.Add(rectControl);
|
rectControl.StrokeThickness = 1;
|
rectControl.Stroke = new SolidColorBrush(Colors.Red);
|
rectControl.Width = rect.Rect.Width;
|
rectControl.Height = rect.Rect.Height;
|
|
|
TextBlock tb = new TextBlock();
|
grid.Children.Add(tb);
|
|
tb.Text = "top:" + rect.Rect.Top + "\r\nleft:" + rect.Rect.Left + "\r\nwidth:" + rect.Rect.Width + "\r\nheight:" + rect.Rect.Height + "\r\nscore:" + rect.Score;
|
tb.HorizontalAlignment = HorizontalAlignment.Left;
|
tb.Foreground = new SolidColorBrush(Colors.White);
|
|
|
|
container.Children.Add(grid);
|
Canvas.SetTop(grid, rect.Rect.Top);
|
Canvas.SetLeft(grid, rect.Rect.Left);
|
|
return rectControl;
|
}
|
|
private void Split_Click(object sender, RoutedEventArgs e)
|
{
|
Splitor splitor = new Splitor();
|
splitor.ShowDialog();
|
}
|
|
/// <summary>
|
/// 选择Xml文件
|
/// </summary>
|
/// <param name="sender"></param>
|
/// <param name="e"></param>
|
private void SelXmlFile_Click(object sender, RoutedEventArgs e)
|
{
|
Microsoft.Win32.OpenFileDialog ofd = new Microsoft.Win32.OpenFileDialog();
|
ofd.DefaultExt = ".xml";
|
ofd.Filter = "xml file|*.xml";
|
ofd.Multiselect = false;
|
if (ofd.ShowDialog() == true)
|
{
|
this.txtXmlFile.Text = ofd.FileName;
|
}
|
}
|
|
/// <summary>
|
/// 选择图片
|
/// </summary>
|
/// <param name="sender"></param>
|
/// <param name="e"></param>
|
private void SelImgFile_Click(object sender, RoutedEventArgs e)
|
{
|
Microsoft.Win32.OpenFileDialog ofd = new Microsoft.Win32.OpenFileDialog();
|
ofd.DefaultExt = ".jpg";
|
ofd.Filter = "jpg file|*.jpg";
|
ofd.Multiselect = false;
|
if (ofd.ShowDialog() == true)
|
{
|
this.txtImgFile.Text = ofd.FileName;
|
}
|
}
|
|
}
|
}
|
