Making sure visualization image is set even when no face is detected.

wchen-casia · Aug 3, 2018 · 65b2b7c · 65b2b7c
1 parent eec7403
commit 65b2b7c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/gui/OpenFaceOffline/MainWindow.xaml.cs b/gui/OpenFaceOffline/MainWindow.xaml.cs
@@ -362,6 +362,8 @@ private void ProcessIndividualImages(ImageReader reader)
 
                 RecorderOpenFace recorder = new RecorderOpenFace(reader.GetName(), rec_params, record_root);
 
+                visualizer_of.SetImage(frame, reader.GetFx(), reader.GetFy(), reader.GetCx(), reader.GetCy());
+
                 // Detect faces here and return bounding boxes
                 List<Rect> face_detections = new List<Rect>();
                 List<float> confidences = new List<float>();
@@ -401,6 +403,8 @@ private void ProcessIndividualImages(ImageReader reader)
 
                 }
 
+                recorder.SetObservationVisualization(visualizer_of.GetVisImage());
+
                 frame = new RawImage(reader.GetNextImage());
                 gray_frame = new RawImage(reader.GetCurrentFrameGray());