static void Main(string[] args)
{
Vision.Windows.WindowsCore.Init();
OpenFileDialog ofd = new OpenFileDialog()
{
InitialDirectory = Environment.CurrentDirectory
};
if (DialogResult.OK == ofd.ShowDialog())
{
var file = new Vision.FileNode(ofd.FileName, true);
EyeGazeCalibrationLog log = new EyeGazeCalibrationLog(file);
log.Load();
LinearEyeGazeCalibratorEngine e = new LinearEyeGazeCalibratorEngine();
e.SetData(log.Data);
e.Train();
var plot = log.Plot(ScreenProperties.CreatePixelScreen(1920, 1080, 96), new EyeGazeCalibrater()
{
Engine = e
});
Core.Cv.ImgShow("plt", plot, 0, true);
}
}
private void TryMove(FileNode source, FileNode dist, int retry = 0, Exception innerException = null)
{
if (retry > 5)
{
throw new Exception("FileCopy Error", innerException);
}
try
{
source.Move(dist);
}
catch (Exception ex)
{
Logger.Error(this, ex);
Core.Sleep(500);
TryMove(source, dist, retry + 1, ex);
}
}
private void Bt_ChangeSize_Click(object sender, RoutedEventArgs e)
{
double newWidth;
double newHeight;
try
{
string[] spl = Tb_NewSize.Text.Split(',');
newWidth = Convert.ToDouble(spl[0]);
newHeight = Convert.ToDouble(spl[1]);
}
catch (Exception ex)
{
Logger.Error("Cant convert input");
return;
}
Vision.Size oldSize = model.ScreenPixelSize;
Vision.Size newSize = new Vision.Size(newWidth, newHeight);
foreach (EyeGazeModelElement ele in model.Elements)
{
Logger.Log($"{ele.Point} To");
ele.Point = LayoutHelper.ResizePoint(ele.Point, oldSize, newSize, Vision.Stretch.Uniform);
Logger.Log($"{ele.Point}");
FileNode newFile = model.Directory.GetFile(EyeGazeModelElement.GetFileName(ele));
TryMove(ele.File, newFile);
ele.File = newFile;
}
model.ScreenPixelSize = newSize;
FileNode node = model.ModelTxt;
if (node.IsExist)
{
node.Delete();
}
node.Create();
using (Stream stream = node.Open())
model.WriteModelText(stream);
Logger.Log("ModelTXT writed");
}
private void Save()
{
var dir = model.Directory.GetDirectory($"[{DateTime.Now.ToString()}] EyesSubModule");
Storage.FixPathChars(dir);
if (!dir.IsExist)
{
dir.Create();
}
var dirLeft = dir.GetDirectory("left");
if (!dirLeft.IsExist)
{
dirLeft.Create();
}
var dirLeft2 = dir.GetDirectory("left2");
if (!dirLeft2.IsExist)
{
dirLeft2.Create();
}
var dirRight = dir.GetDirectory("right");
if (!dirRight.IsExist)
{
dirRight.Create();
}
var dirRight2 = dir.GetDirectory("right2");
if (!dirRight2.IsExist)
{
dirRight2.Create();
}
var dirFace = dir.GetDirectory("face");
if (!dirFace.IsExist)
{
dirFace.Create();
}
if (dir != null)
{
FileNode text = dir.NewFile("model.txt");
using (var stream = text.Open())
{
model.WriteModelText(stream);
}
List <FaceDetectionProvider> detectors = new List <FaceDetectionProvider>();
for (int i = 0; i < Environment.ProcessorCount; i++)
{
Logger.Log($"Load Detector [{i + 1}/{Environment.ProcessorCount}]");
detectors.Add(new OpenFaceDetector()
{
MaxSize = 640,
Interpolation = InterpolationFlags.Lanczos4
});
}
ScreenProperties = new ScreenProperties()
{
Origin = model.ScreenOrigin,
PixelSize = model.ScreenPixelSize,
Size = model.ScreenSize
};
object countLocker = new object();
int count = 0;
Parallel.ForEach(model.Elements, new ParallelOptions()
{
MaxDegreeOfParallelism = Environment.ProcessorCount
}, (ele) =>
{
int detectorInd;
FaceDetectionProvider detector;
lock (countLocker)
{
count++;
detectorInd = count % Environment.ProcessorCount;
detector = detectors[detectorInd];
}
using (Mat frame = Core.Cv.ImgRead(ele.File))
{
if (frame != null && !frame.IsEmpty)
{
try
{
FaceRect[] faces = detector.Detect(frame);
if (faces != null && faces.Length > 0)
{
foreach (var face in faces)
{
if (Math.Abs(face.LandmarkTransformVector[2]) < 7500 && face.LeftEye != null && face.RightEye != null)
{
var rod = face.SolveLookScreenVector(ele.Point, ScreenProperties).ToArray();
var filename = $"{ele.Index},{rod[0]},{rod[1]},{rod[2]}.jpg";
FileNode eyeFileLeft = dirLeft.GetFile(filename);
FileNode eyeFileLeft2 = dirLeft2.GetFile(filename);
FileNode eyeFileRight = dirRight.GetFile(filename);
FileNode eyeFileRight2 = dirRight2.GetFile(filename);
FileNode eyeFileFace = dirFace.GetFile(filename);
using (Mat roi = face.LeftEye.RoiCropByPercent(frame, 0.25))
Core.Cv.ImgWrite(eyeFileLeft, roi, 92);
using (Mat roi = face.LeftEye.RoiCropByPercent(frame, 0.5))
Core.Cv.ImgWrite(eyeFileLeft2, roi, 92);
using (Mat roi = face.RightEye.RoiCropByPercent(frame, 0.25))
Core.Cv.ImgWrite(eyeFileRight, roi, 92);
using (Mat roi = face.RightEye.RoiCropByPercent(frame, 0.5))
Core.Cv.ImgWrite(eyeFileRight2, roi, 92);
using (Mat roi = face.ROI(frame))
Core.Cv.ImgWrite(eyeFileFace, roi, 92);
}
}
}
}
catch (Exception ex)
{
Logger.Error(ex.ToString());
}
}
}
Logger.Log($"Extracted [{count}/{model.Elements.Count}]");
});
foreach (var d in detectors)
{
d.Dispose();
}
}
else
{
Logger.Error("Cant Create Dir");
MessageBox.Show("CantCreateDir");
}
}
