|
|
@@ -1,15 +1,14 @@
|
|
|
package com.izouma.awesomeadmin.util;
|
|
|
|
|
|
import com.izouma.awesomeadmin.service.OSSFileService;
|
|
|
+import io.netty.buffer.ByteBuf;
|
|
|
import org.apache.commons.lang3.StringUtils;
|
|
|
import org.bytedeco.javacpp.indexer.UByteRawIndexer;
|
|
|
+import org.bytedeco.javacpp.lept;
|
|
|
import org.bytedeco.javacpp.opencv_imgproc;
|
|
|
import org.bytedeco.javacpp.opencv_ml;
|
|
|
import org.bytedeco.javacpp.tesseract;
|
|
|
-import org.bytedeco.javacv.FFmpegFrameGrabber;
|
|
|
-import org.bytedeco.javacv.Frame;
|
|
|
-import org.bytedeco.javacv.FrameGrabber;
|
|
|
-import org.bytedeco.javacv.OpenCVFrameConverter;
|
|
|
+import org.bytedeco.javacv.*;
|
|
|
import org.hibernate.validator.internal.util.privilegedactions.GetResource;
|
|
|
|
|
|
import javax.imageio.ImageIO;
|
|
|
@@ -27,6 +26,7 @@ import java.util.UUID;
|
|
|
import java.util.regex.Pattern;
|
|
|
|
|
|
import static org.bytedeco.javacpp.lept.pixRead;
|
|
|
+import static org.bytedeco.javacpp.lept.pixReadMem;
|
|
|
import static org.bytedeco.javacpp.opencv_core.*;
|
|
|
import static org.bytedeco.javacpp.opencv_imgcodecs.imencode;
|
|
|
import static org.bytedeco.javacpp.opencv_imgcodecs.imwrite;
|
|
|
@@ -275,14 +275,11 @@ public class VideoProcessTool {
|
|
|
}
|
|
|
|
|
|
private String doOCR(Mat img) {
|
|
|
- imwrite("/Users/drew/Desktop/text.jpg", img);
|
|
|
- cvtColor(img, img, COLOR_GRAY2RGBA, 4);
|
|
|
+ imwrite("text.jpg", img);
|
|
|
final tesseract.TessBaseAPI baseApi = new tesseract.TessBaseAPI();
|
|
|
baseApi.Init(GetResource.class.getClassLoader().getResource("trainneddata").getPath(), "pubg");
|
|
|
baseApi.SetPageSegMode(PSM_SINGLE_LINE);
|
|
|
- // Ensure that text is recognized.
|
|
|
- // baseApi.SetImage(pixReadMem(img.data(), img.elemSize()));
|
|
|
- baseApi.SetImage(pixRead("/Users/drew/Desktop/text.jpg"));
|
|
|
+ baseApi.SetImage(pixRead("text.jpg"));
|
|
|
String recognizedText = baseApi.GetUTF8Text().getString();
|
|
|
return recognizedText;
|
|
|
}
|