保存检测到目标的时间和图片

Signed-off-by: JoeAllen <2494136784@qq.com>
2022-07-29 14:17:33 +08:00
parent 25b7c56a48
commit 1d2704ce5e
1 changed files with 6 additions and 6 deletions
@@ -57,13 +57,13 @@ public class Main {
      // 显示视频图像
      canvas.showImage(frame);
      if (bestItem.getProbability() == 1.0 && !"非动物".equals(bestItem.getClassName())) {
-        File outPutFile =
-            new File(String.valueOf(Paths.get("src/test/resources/" + i++ + "coffee.png")));
-        assert buffImg != null;
-        if (!outPutFile.exists()) {
-          log.info(String.valueOf(outPutFile.createNewFile()));
+        File outPutFile = Paths.get("src/test/resources/coffee" + i++ + ".png").toFile();
+        if (buffImg != null) {
+          if (!outPutFile.exists()) {
+            log.info(String.valueOf(outPutFile.createNewFile()));
+          }
+          ImageIO.write(buffImg, "png", outPutFile);
        }
-        ImageIO.write(buffImg, "png", outPutFile);
      }
    }
    canvas.dispose();