JoeAllen/DetectAnimalsFromCameras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source 
Signed-off-by: JoeAllen <2494136784@qq.com>
This commit is contained in:
JoeAllen
2022-07-29 11:27:57 +08:00
parent d63075814e
commit f5f35bc520
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/cn/heavenlybook/Main.java
+9 -3
View File
@@ -19,13 +19,15 @@ import javax.swing.*;
import java.awt.image.BufferedImage; import java.awt.image.BufferedImage;
import java.io.File; import java.io.File;
import java.io.IOException; import java.io.IOException;
import java.util.Date; import java.nio.file.Paths;
/** /**
* @author ht * @author ht
*/ */
@Log @Log
public class Main { public class Main {
static int i = 0;
public static void main(String[] args) throws IOException, ModelException, TranslateException { public static void main(String[] args) throws IOException, ModelException, TranslateException {
// 开启摄像头，获取图像（得到的图像为frame类型，需要转换为mat类型进行检测和识别） // 开启摄像头，获取图像（得到的图像为frame类型，需要转换为mat类型进行检测和识别）
OpenCVFrameGrabber grabber = new OpenCVFrameGrabber(0); OpenCVFrameGrabber grabber = new OpenCVFrameGrabber(0);
@@ -54,9 +56,13 @@ public class Main {
log.info(classifications.toJson()); log.info(classifications.toJson());
// 显示视频图像 // 显示视频图像
canvas.showImage(frame); canvas.showImage(frame);
if (bestItem.getProbability() == 1.0) { if (bestItem.getProbability() == 1.0 && !"非动物".equals(bestItem.getClassName())) {
File outPutFile = new File(new Date() + ".png"); File outPutFile =
new File(String.valueOf(Paths.get("src/test/resources/" + i++ + "coffee.png")));
assert buffImg != null; assert buffImg != null;
if (!outPutFile.exists()) {
log.info(String.valueOf(outPutFile.createNewFile()));
}
ImageIO.write(buffImg, "png", outPutFile); ImageIO.write(buffImg, "png", outPutFile);
} }
} }