From d63075814eaf8d5e1ff24c0b3a750189454333f3 Mon Sep 17 00:00:00 2001
From: JoeAllen <2494136784@qq.com>
Date: Fri, 29 Jul 2022 11:01:54 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=9D=E5=AD=98=E6=A3=80=E6=B5=8B=E5=88=B0?=
 =?UTF-8?q?=E7=9B=AE=E6=A0=87=E7=9A=84=E6=97=B6=E9=97=B4=E5=92=8C=E5=9B=BE?=
 =?UTF-8?q?=E7=89=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: JoeAllen <2494136784@qq.com>
---
 src/main/java/cn/heavenlybook/Main.java | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/main/java/cn/heavenlybook/Main.java b/src/main/java/cn/heavenlybook/Main.java
index 54a0dbf..9c51fe2 100644
--- a/src/main/java/cn/heavenlybook/Main.java
+++ b/src/main/java/cn/heavenlybook/Main.java
@@ -14,9 +14,12 @@ import org.bytedeco.javacv.OpenCVFrameConverter;
 import org.bytedeco.javacv.OpenCVFrameGrabber;
 import org.bytedeco.opencv.opencv_core.Mat;
 
+import javax.imageio.ImageIO;
 import javax.swing.*;
 import java.awt.image.BufferedImage;
+import java.io.File;
 import java.io.IOException;
+import java.util.Date;
 
 /**
  * @author ht
@@ -51,6 +54,11 @@ public class Main {
       log.info(classifications.toJson());
       // 显示视频图像
       canvas.showImage(frame);
+      if (bestItem.getProbability() == 1.0) {
+        File outPutFile = new File(new Date() + ".png");
+        assert buffImg != null;
+        ImageIO.write(buffImg, "png", outPutFile);
+      }
     }
     canvas.dispose();
     grabber.close();