using RichCreator.Editor.Utils;
|
using System;
|
using System.Collections.Generic;
|
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 forms=System.Windows.Forms;
|
using System.Windows.Input;
|
using System.Windows.Media;
|
using System.Windows.Media.Imaging;
|
using System.Windows.Shapes;
|
using ZTImage;
|
|
namespace RichCreator.Editor.Tools.CV
|
{
|
/// <summary>
|
/// Trainor.xaml 的交互逻辑
|
/// </summary>
|
public partial class Trainor : Window
|
{
|
public Trainor()
|
{
|
InitializeComponent();
|
}
|
|
public string PosDir { get; set; }
|
|
public string NegDir { get; set; }
|
|
public string GenericXmlName { 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; }
|
|
|
private void Button_Click(object sender, RoutedEventArgs e)
|
{
|
if (!CheckAndFillAllParameter())
|
{
|
return;
|
}
|
|
if (System.IO.File.Exists(this.GenericXmlName))
|
{
|
try
|
{
|
System.IO.File.Delete(this.GenericXmlName);
|
}
|
catch(Exception ex)
|
{
|
ShowMessage("已经存在的xml无法移除,"+ex.ToString());
|
return;
|
}
|
}
|
|
|
HOGTrain train = new HOGTrain(
|
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
|
);
|
|
string[] files = null;
|
//训练正样本
|
files = System.IO.Directory.GetFiles(this.PosDir,"*.jpg",System.IO.SearchOption.AllDirectories);
|
if (files.Length <= 0)
|
{
|
ShowMessage("正样本不存在");
|
return;
|
}
|
train.Train(files, 1);
|
|
//训练负样本
|
files = System.IO.Directory.GetFiles(this.NegDir, "*.jpg", System.IO.SearchOption.AllDirectories);
|
if (files.Length <= 0)
|
{
|
ShowMessage("负样本不存在");
|
return;
|
}
|
train.Train(files, -1);
|
|
|
train.Save(this.GenericXmlName);
|
ShowMessage("train complete");
|
}
|
|
|
private Boolean CheckAndFillAllParameter()
|
{
|
this.GenericXmlName = this.txtGenericXmlName.Text.Trim();
|
if (string.IsNullOrEmpty(this.GenericXmlName))
|
{
|
ShowMessage("请填写结果Xml名");
|
return false;
|
}
|
|
if (System.IO.File.Exists(this.GenericXmlName))
|
{
|
if (MessageBox.Show("xml已经存在，是否覆盖?", "提示", MessageBoxButton.YesNo) == MessageBoxResult.No)
|
{
|
return false;
|
}
|
}
|
|
|
this.PosDir = this.txtPosDir.Text.Trim();
|
if (string.IsNullOrEmpty(this.PosDir))
|
{
|
ShowMessage("请选择正样本目录");
|
return false;
|
}
|
|
if (!System.IO.Directory.Exists(this.PosDir))
|
{
|
ShowMessage("选择的正样本目录不存在");
|
return false;
|
}
|
|
this.NegDir = this.txtNegDir.Text.Trim();
|
if (string.IsNullOrEmpty(this.NegDir))
|
{
|
ShowMessage("请选择负样本目录");
|
return false;
|
}
|
|
if (!System.IO.Directory.Exists(this.NegDir))
|
{
|
ShowMessage("选择的负样本目录不存在");
|
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)
|
{
|
System.Windows.MessageBox.Show(msg);
|
}
|
|
private void SelPosDir_Click(object sender, RoutedEventArgs e)
|
{
|
//选择正样本目录
|
forms.FolderBrowserDialog m_Dialog = new forms.FolderBrowserDialog();
|
m_Dialog.SelectedPath = AppDomain.CurrentDomain.BaseDirectory;
|
forms.DialogResult result = m_Dialog.ShowDialog();
|
|
if (result == System.Windows.Forms.DialogResult.Cancel)
|
{
|
return;
|
}
|
string m_Dir = m_Dialog.SelectedPath.Trim();
|
this.txtPosDir.Text = m_Dir;
|
}
|
|
private void SelNegDir_Click(object sender, RoutedEventArgs e)
|
{
|
//选择负样本目录
|
forms.FolderBrowserDialog m_Dialog = new forms.FolderBrowserDialog();
|
m_Dialog.SelectedPath = AppDomain.CurrentDomain.BaseDirectory;
|
forms.DialogResult result = m_Dialog.ShowDialog();
|
|
if (result == System.Windows.Forms.DialogResult.Cancel)
|
{
|
return;
|
}
|
string m_Dir = m_Dialog.SelectedPath.Trim();
|
this.txtNegDir.Text = m_Dir;
|
}
|
}
|
}
|
