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 System.Windows.Input;
|
using System.Windows.Media;
|
using System.Windows.Media.Imaging;
|
using System.Windows.Shapes;
|
using Emgu.CV;
|
using Emgu.CV.Structure;
|
using ZTImage;
|
using forms = System.Windows.Forms;
|
|
namespace RichCreator.Editor.Tools.CV
|
{
|
/// <summary>
|
/// Splitor.xaml 的交互逻辑
|
/// </summary>
|
public partial class Splitor : Window
|
{
|
public Splitor()
|
{
|
InitializeComponent();
|
}
|
|
|
|
public string ExportDir { get; set; }
|
|
public int WinSizeWidth { get; set; }
|
|
|
public int WinSizeHeight { get; set; }
|
|
public int StrideSizeWidth { get; set; }
|
|
public int StrideSizeHeight { get; set; }
|
|
|
|
|
/// <summary>
|
/// 打开背景图
|
/// </summary>
|
/// <param name="sender"></param>
|
/// <param name="e"></param>
|
private void OpenPicture_Click(object sender, RoutedEventArgs e)
|
{
|
imagePath = string.Empty;
|
Microsoft.Win32.OpenFileDialog ofd = new Microsoft.Win32.OpenFileDialog();
|
ofd.DefaultExt = ".jpg";
|
ofd.Filter = "jpg file|*.jpg";
|
ofd.Multiselect = false;
|
if (ofd.ShowDialog() == true)
|
{
|
imagePath = ofd.FileName;
|
}
|
else
|
{
|
return;
|
}
|
|
|
this.container.Children.Clear();
|
|
byte[] imageData = System.IO.File.ReadAllBytes(imagePath);
|
var ms = new System.IO.MemoryStream(imageData);
|
|
//画图
|
Image imgControl = new Image();
|
imgControl.Source = BitmapFrame.Create(ms);
|
this.container.Children.Add(imgControl);
|
}
|
|
string imagePath;
|
|
private void AutoGenericRect_Click(object sender, RoutedEventArgs e)
|
{
|
|
}
|
|
/// <summary>
|
/// 开始生成
|
/// </summary>
|
/// <param name="sender"></param>
|
/// <param name="e"></param>
|
private void GenericPicture_Click(object sender, RoutedEventArgs e)
|
{
|
FillParameter();
|
|
if (string.IsNullOrEmpty(this.imagePath))
|
{
|
ShowMessage("请选择要分割的图片");
|
return;
|
}
|
|
if (!System.IO.File.Exists(this.imagePath))
|
{
|
ShowMessage("请选择要分割的图片");
|
return;
|
}
|
|
if (string.IsNullOrEmpty(this.ExportDir))
|
{
|
ShowMessage("请选择要导出到的目录");
|
return;
|
}
|
|
if (!System.IO.Directory.Exists(this.ExportDir))
|
{
|
ShowMessage("请选择要导出到的目录");
|
return;
|
}
|
|
|
Mat mat = CvInvoke.Imread(this.imagePath);
|
if (this.WinSizeWidth<=0||this.WinSizeHeight<=0||this.WinSizeWidth > mat.Width||this.WinSizeHeight>mat.Height)
|
{
|
ShowMessage("请设置合适的win size大小");
|
return;
|
|
}
|
|
if (this.StrideSizeWidth <= 0 || this.StrideSizeHeight <= 0 || this.StrideSizeWidth > mat.Width || this.StrideSizeHeight > mat.Height)
|
{
|
ShowMessage("请设置合适的stride size大小");
|
return;
|
|
}
|
List<Rect> rects = GetRects(mat.Width, mat.Height);
|
Image<Gray, byte> image = mat.ToImage<Gray, byte>();
|
|
Image<Gray, byte> im = null;
|
Random r = new Random();
|
for (int h = 0; h <= (mat.Height-WinSizeHeight); h+=this.StrideSizeHeight)
|
{
|
for (int w = 0; w <= (mat.Width - WinSizeWidth); w += this.StrideSizeWidth)
|
{
|
bool isin = IsInRect(w, h, this.WinSizeWidth, this.WinSizeHeight, rects);
|
if (isin)
|
{
|
continue;
|
}
|
im = image.GetSubRect(new System.Drawing.Rectangle(w, h, this.WinSizeWidth, this.WinSizeHeight));
|
im.Save(System.IO.Path.Combine(this.ExportDir, Guid.NewGuid().ToString() + ".jpg"));
|
}
|
}
|
|
MessageBox.Show("splitor complete");
|
}
|
|
/// <summary>
|
/// 添加排除区域
|
/// </summary>
|
/// <param name="sender"></param>
|
/// <param name="e"></param>
|
private void AddExcArea_Click(object sender, RoutedEventArgs e)
|
{
|
FillParameter();
|
|
SplitArea area = new SplitArea();
|
area.OnClose += Area_OnClose;
|
this.container.Children.Add(area);
|
area.MinWidth = this.WinSizeWidth;
|
area.MinHeight = this.WinSizeHeight;
|
}
|
|
/// <summary>
|
/// 区域删除事件
|
/// </summary>
|
/// <param name="obj"></param>
|
private void Area_OnClose(SplitArea obj)
|
{
|
this.container.Children.Remove(obj);
|
}
|
|
/// <summary>
|
/// 选择导出文件夹
|
/// </summary>
|
/// <param name="sender"></param>
|
/// <param name="e"></param>
|
private void SelExportDir_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.txtExportDir.Text = m_Dir;
|
}
|
|
/// <summary>
|
/// 填充参数
|
/// </summary>
|
private void FillParameter()
|
{
|
this.ExportDir = this.txtExportDir.Text;
|
this.WinSizeWidth = TypeConverter.StringToInt(this.txtWinWidth.Text,0);
|
this.WinSizeHeight = TypeConverter.StringToInt(this.txtWinHeight.Text, 0);
|
this.StrideSizeWidth = TypeConverter.StringToInt(this.txtStrideWidth.Text, 0);
|
this.StrideSizeHeight = TypeConverter.StringToInt(this.txtStrideHeight.Text, 0);
|
}
|
|
private void ShowMessage(string msg)
|
{
|
MessageBox.Show(msg);
|
}
|
|
/// <summary>
|
/// 得到限制区域
|
/// </summary>
|
/// <param name="maxWidth"></param>
|
/// <param name="maxHeight"></param>
|
/// <returns></returns>
|
private List<Rect> GetRects(Int32 maxWidth,Int32 maxHeight)
|
{
|
List<Rect> rects = new List<Rect>();
|
|
foreach (var item in this.container.Children)
|
{
|
SplitArea area = item as SplitArea;
|
if (area == null)
|
{
|
continue;
|
}
|
Rect rect = area.GetRect();
|
if (rect.X < 0 || rect.Y < 0 ||
|
(rect.X + rect.Width) > MaxWidth || (rect.Y + rect.Height) > MaxHeight)
|
{
|
continue;
|
}
|
|
rects.Add(rect);
|
}
|
return rects;
|
}
|
|
private bool IsInRect(Int32 x, Int32 y, Int32 width, Int32 height, List<Rect> rects)
|
{
|
if (rects == null || rects.Count <= 0)
|
{
|
return false;
|
}
|
Int32 xe = x + width;
|
Int32 ye = y + height;
|
Rect r;
|
for (int i = 0; i < rects.Count; i++)
|
{
|
r = rects[i];
|
if (x >= r.X && x <= (r.X + r.Width) && y >= r.Y && y <= (r.Y + r.Height))
|
{
|
return true;
|
}
|
|
if (xe >= r.X && xe <= (r.X + r.Width) && ye >= r.Y && ye <= (r.Y + r.Height))
|
{
|
return true;
|
}
|
}
|
|
return false;
|
}
|
|
|
}
|
}
|
