100 changed files with 1184 additions and 2043 deletions
@ -1,30 +0,0 @@ |
|||
<?xml version="1.0" encoding="UTF-8"?> |
|||
<project xmlns="http://maven.apache.org/POM/4.0.0" |
|||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" |
|||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> |
|||
<modelVersion>4.0.0</modelVersion> |
|||
|
|||
<groupId>com.visual.open.anpr</groupId> |
|||
<artifactId>open-anpr-client</artifactId> |
|||
<version>1.1.0</version> |
|||
|
|||
<properties> |
|||
<maven.compiler.source>8</maven.compiler.source> |
|||
<maven.compiler.target>8</maven.compiler.target> |
|||
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding> |
|||
</properties> |
|||
|
|||
<dependencies> |
|||
<dependency> |
|||
<groupId>com.alibaba</groupId> |
|||
<artifactId>fastjson</artifactId> |
|||
<version>1.2.58</version> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>org.apache.httpcomponents</groupId> |
|||
<artifactId>httpclient</artifactId> |
|||
<version>4.5</version> |
|||
</dependency> |
|||
</dependencies> |
|||
|
|||
</project> |
|||
@ -1,47 +0,0 @@ |
|||
package com.visual.open.anpr; |
|||
|
|||
import java.util.Map; |
|||
import java.util.concurrent.ConcurrentHashMap; |
|||
import com.visual.open.anpr.handle.RecognitionHandler; |
|||
|
|||
public class PlateRecognition { |
|||
/**服务地址**/ |
|||
private String serverHost; |
|||
/**实例对象**/ |
|||
private final static Map<String, PlateRecognition> ins = new ConcurrentHashMap<>(); |
|||
|
|||
/** |
|||
* 构建集合对象 |
|||
* @param serverHost 服务地址 |
|||
* @return |
|||
*/ |
|||
private PlateRecognition(String serverHost){ |
|||
this.serverHost = serverHost; |
|||
} |
|||
|
|||
/** |
|||
* 构建集合对象 |
|||
* @param serverHost 服务地址 |
|||
* @return |
|||
*/ |
|||
public static PlateRecognition build (String serverHost){ |
|||
String key = serverHost; |
|||
if(!ins.containsKey(key)){ |
|||
synchronized (PlateRecognition.class){ |
|||
if(!ins.containsKey(key)){ |
|||
ins.put(key, new PlateRecognition(serverHost)); |
|||
} |
|||
} |
|||
} |
|||
return ins.get(key); |
|||
} |
|||
/** |
|||
* 车牌检测操作对象 |
|||
* @return CollectHandler |
|||
*/ |
|||
public RecognitionHandler detection(){ |
|||
return RecognitionHandler.build(serverHost); |
|||
} |
|||
|
|||
|
|||
} |
|||
@ -1,12 +0,0 @@ |
|||
package com.visual.open.anpr.common; |
|||
|
|||
public class Api { |
|||
|
|||
public static final String plate_recognition = "/visual/plate/recognition"; |
|||
|
|||
public static String getUrl(String host, String uri){ |
|||
host = host.replaceAll ("/+$", ""); |
|||
return host + uri; |
|||
} |
|||
|
|||
} |
|||
@ -1,63 +0,0 @@ |
|||
package com.visual.open.anpr.handle; |
|||
|
|||
import com.alibaba.fastjson.TypeReference; |
|||
import com.visual.open.anpr.common.Api; |
|||
import com.visual.open.anpr.http.HttpClient; |
|||
import com.visual.open.anpr.model.Recognition; |
|||
import com.visual.open.anpr.model.RecognitionRep; |
|||
import com.visual.open.anpr.model.RecognitionReq; |
|||
import com.visual.open.anpr.model.Response; |
|||
|
|||
import java.util.List; |
|||
import java.util.Map; |
|||
import java.util.concurrent.ConcurrentHashMap; |
|||
|
|||
public class RecognitionHandler { |
|||
|
|||
/**服务地址**/ |
|||
protected String serverHost; |
|||
/**实例对象**/ |
|||
private final static Map<String, RecognitionHandler> ins = new ConcurrentHashMap<>(); |
|||
|
|||
public String getServerHost() { |
|||
return serverHost; |
|||
} |
|||
|
|||
public RecognitionHandler setServerHost(String serverHost) { |
|||
this.serverHost = serverHost; |
|||
return this; |
|||
} |
|||
|
|||
/** |
|||
* 构建车牌检测对象 |
|||
* @param serverHost 服务地址 |
|||
* @return |
|||
*/ |
|||
public static RecognitionHandler build(String serverHost){ |
|||
String key = serverHost; |
|||
if(!ins.containsKey(key)){ |
|||
synchronized (RecognitionHandler.class){ |
|||
if(!ins.containsKey(key)){ |
|||
ins.put(key, new RecognitionHandler().setServerHost(serverHost)); |
|||
} |
|||
} |
|||
} |
|||
return ins.get(key); |
|||
} |
|||
|
|||
/** |
|||
* 车牌检测 |
|||
* @param recognition 待检测的数据信息 |
|||
* @return 车牌检测结果 |
|||
*/ |
|||
public Response<List<RecognitionRep>> recognition(Recognition recognition){ |
|||
RecognitionReq compareReq = RecognitionReq.build() |
|||
.setImage(recognition.getImage()) |
|||
.setLimit(recognition.getLimit()); |
|||
return HttpClient.post( |
|||
Api.getUrl(this.serverHost, Api.plate_recognition), |
|||
compareReq, |
|||
new TypeReference<Response<List<RecognitionRep>>>() {}); |
|||
} |
|||
|
|||
} |
|||
@ -1,207 +0,0 @@ |
|||
package com.visual.open.anpr.http; |
|||
|
|||
import com.alibaba.fastjson.TypeReference; |
|||
|
|||
import com.visual.open.anpr.model.MapParam; |
|||
import com.visual.open.anpr.model.Response; |
|||
import com.visual.open.anpr.utils.JsonUtil; |
|||
import org.apache.http.NameValuePair; |
|||
import org.apache.http.client.config.RequestConfig; |
|||
import org.apache.http.client.methods.CloseableHttpResponse; |
|||
import org.apache.http.client.methods.HttpGet; |
|||
import org.apache.http.client.methods.HttpPost; |
|||
import org.apache.http.client.utils.URIBuilder; |
|||
import org.apache.http.config.Registry; |
|||
import org.apache.http.config.RegistryBuilder; |
|||
import org.apache.http.conn.socket.ConnectionSocketFactory; |
|||
import org.apache.http.conn.socket.PlainConnectionSocketFactory; |
|||
import org.apache.http.conn.ssl.SSLConnectionSocketFactory; |
|||
import org.apache.http.entity.StringEntity; |
|||
import org.apache.http.impl.client.BasicCookieStore; |
|||
import org.apache.http.impl.client.CloseableHttpClient; |
|||
import org.apache.http.impl.client.HttpClientBuilder; |
|||
import org.apache.http.impl.conn.PoolingHttpClientConnectionManager; |
|||
import org.apache.http.message.BasicNameValuePair; |
|||
import org.apache.http.util.EntityUtils; |
|||
|
|||
import javax.net.ssl.SSLContext; |
|||
import javax.net.ssl.TrustManager; |
|||
import javax.net.ssl.X509TrustManager; |
|||
import java.io.IOException; |
|||
import java.security.cert.CertificateException; |
|||
import java.security.cert.X509Certificate; |
|||
import java.util.LinkedList; |
|||
import java.util.List; |
|||
|
|||
public class HttpClient { |
|||
|
|||
/**编码方式*/ |
|||
private static final String ENCODING = "UTF-8"; |
|||
/**连接超时时间，10秒*/ |
|||
public static final int DEFAULT_CONNECT_TIMEOUT = 10 * 1000; |
|||
/**socket连接超时时间，10秒*/ |
|||
public static final int DEFAULT_READ_TIMEOUT = 10 * 000; |
|||
/**请求超时时间，60秒*/ |
|||
public static final int DEFAULT_CONNECT_REQUEST_TIMEOUT = 30 * 000; |
|||
/**最大连接数,默认为2*/ |
|||
private static final int MAX_TOTAL = 8; |
|||
/**设置指向特定路由的并发连接总数，默认为2*/ |
|||
private static final int MAX_PER_ROUTE = 4; |
|||
|
|||
private static RequestConfig requestConfig; |
|||
private static PoolingHttpClientConnectionManager connectionManager; |
|||
private static BasicCookieStore cookieStore; |
|||
private static HttpClientBuilder httpBuilder; |
|||
private static CloseableHttpClient httpClient; |
|||
private static CloseableHttpClient httpsClient; |
|||
private static SSLContext sslContext; |
|||
|
|||
/** |
|||
* 创建SSLContext对象，用来绕过https证书认证实现访问。 |
|||
*/ |
|||
static { |
|||
try { |
|||
sslContext = SSLContext.getInstance("TLS"); |
|||
// 实现一个X509TrustManager接口，用于绕过验证，不用修改里面的方法
|
|||
X509TrustManager tm = new X509TrustManager() { |
|||
@Override |
|||
public void checkClientTrusted(X509Certificate[] chain, String authType) |
|||
throws CertificateException { |
|||
} |
|||
@Override |
|||
public void checkServerTrusted(X509Certificate[] chain, String authType) |
|||
throws CertificateException { |
|||
} |
|||
@Override |
|||
public X509Certificate[] getAcceptedIssuers() { |
|||
return null; |
|||
} |
|||
}; |
|||
sslContext.init(null, new TrustManager[] {tm}, null); |
|||
} catch (Exception e) { |
|||
e.printStackTrace(); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 初始化httpclient对象，以及在创建httpclient对象之前的一些自定义配置。 |
|||
*/ |
|||
static { |
|||
// 自定义配置信息
|
|||
requestConfig = RequestConfig.custom() |
|||
.setSocketTimeout(DEFAULT_READ_TIMEOUT) |
|||
.setConnectTimeout(DEFAULT_CONNECT_TIMEOUT) |
|||
.setConnectionRequestTimeout(DEFAULT_CONNECT_REQUEST_TIMEOUT) |
|||
.build(); |
|||
//设置协议http和https对应的处理socket链接工厂的对象
|
|||
Registry<ConnectionSocketFactory> socketFactoryRegistry = RegistryBuilder.<ConnectionSocketFactory> create() |
|||
.register("http", new PlainConnectionSocketFactory()) |
|||
.register("https", new SSLConnectionSocketFactory(sslContext)) |
|||
.build(); |
|||
connectionManager = new PoolingHttpClientConnectionManager(socketFactoryRegistry); |
|||
// 设置cookie存储对像，在需要获取cookie信息时，可以使用这个对象。
|
|||
cookieStore = new BasicCookieStore(); |
|||
// 设置最大连接数
|
|||
connectionManager.setMaxTotal(MAX_TOTAL); |
|||
// 设置路由并发数
|
|||
connectionManager.setDefaultMaxPerRoute(MAX_PER_ROUTE); |
|||
httpBuilder = HttpClientBuilder.create(); |
|||
httpBuilder.setDefaultRequestConfig(requestConfig); |
|||
httpBuilder.setConnectionManager(connectionManager); |
|||
httpBuilder.setDefaultCookieStore(cookieStore); |
|||
// 实例化http 和 https的对象。
|
|||
httpClient = httpBuilder.build(); |
|||
httpsClient = httpBuilder.build(); |
|||
} |
|||
|
|||
/** |
|||
* post请求 |
|||
* @param url |
|||
* @param data |
|||
* @param <T> |
|||
* @return |
|||
*/ |
|||
public static <T> Response<T> post(String url, Object data) { |
|||
return post(url, data, new TypeReference<Response<T>>() {}); |
|||
} |
|||
|
|||
/** |
|||
* post请求 |
|||
* @param url |
|||
* @param data |
|||
* @param <T> |
|||
* @return |
|||
*/ |
|||
public static <T> Response<T> post(String url, Object data, TypeReference<Response<T>> type) { |
|||
// 创建HTTP对象
|
|||
HttpPost httpPost = new HttpPost(url); |
|||
httpPost.setConfig(requestConfig); |
|||
httpPost.addHeader("Content-Type", "application/json;charset=UTF-8"); |
|||
// 设置请求头
|
|||
if(null != data){ |
|||
String json = data instanceof String ? String.valueOf(data) : JsonUtil.toString(data); |
|||
StringEntity stringEntity = new StringEntity(json, ENCODING); |
|||
stringEntity.setContentEncoding(ENCODING); |
|||
httpPost.setEntity(stringEntity); |
|||
} |
|||
// 创建httpResponse对象
|
|||
CloseableHttpClient client = url.toLowerCase().startsWith("https") ? httpsClient : httpClient; |
|||
try { |
|||
CloseableHttpResponse httpResponse = client.execute(httpPost); |
|||
String content = EntityUtils.toString(httpResponse.getEntity(), ENCODING); |
|||
return JsonUtil.toEntity(content, type); |
|||
} catch (IOException e) { |
|||
throw new RuntimeException(e); |
|||
} |
|||
} |
|||
|
|||
|
|||
/** |
|||
* get |
|||
* @param url |
|||
* @param param |
|||
* @param <T> |
|||
* @return |
|||
*/ |
|||
public static <T> Response<T> get(String url, MapParam param) { |
|||
return get(url, param, new TypeReference<Response<T>>() {}); |
|||
} |
|||
|
|||
/** |
|||
* get |
|||
* @param url |
|||
* @param param |
|||
* @param <T> |
|||
* @return |
|||
*/ |
|||
public static <T> Response<T> get(String url, MapParam param, TypeReference<Response<T>> type) { |
|||
try { |
|||
//参数构建
|
|||
URIBuilder uriBuilder = new URIBuilder(url); |
|||
if(null != param && !param.isEmpty()){ |
|||
List<NameValuePair> list = new LinkedList<>(); |
|||
for(String key : param.keySet()){ |
|||
Object value = param.get(key); |
|||
if(null == value){ |
|||
list.add(new BasicNameValuePair(key, null)); |
|||
}else{ |
|||
list.add(new BasicNameValuePair(key, String.valueOf(value))); |
|||
} |
|||
} |
|||
uriBuilder.setParameters(list); |
|||
} |
|||
//构建请求
|
|||
HttpGet httpGet = new HttpGet(uriBuilder.build()); |
|||
httpGet.setConfig(requestConfig); |
|||
httpGet.addHeader("Content-Type", "application/json;charset=UTF-8"); |
|||
// 创建httpResponse对象
|
|||
CloseableHttpClient client = url.toLowerCase().startsWith("https") ? httpsClient : httpClient; |
|||
CloseableHttpResponse httpResponse = client.execute(httpGet); |
|||
String content = EntityUtils.toString(httpResponse.getEntity(), ENCODING); |
|||
return JsonUtil.toEntity(content, type); |
|||
} catch (Exception e) { |
|||
throw new RuntimeException(e); |
|||
} |
|||
} |
|||
|
|||
} |
|||
@ -1,4 +0,0 @@ |
|||
package com.visual.open.anpr.http; |
|||
|
|||
public class HttpClientResult { |
|||
} |
|||
@ -1,455 +0,0 @@ |
|||
package com.visual.open.anpr.http; |
|||
|
|||
import org.apache.http.NameValuePair; |
|||
import org.apache.http.client.config.RequestConfig; |
|||
import org.apache.http.client.entity.UrlEncodedFormEntity; |
|||
import org.apache.http.client.methods.*; |
|||
import org.apache.http.client.utils.URIBuilder; |
|||
import org.apache.http.config.Registry; |
|||
import org.apache.http.config.RegistryBuilder; |
|||
import org.apache.http.conn.socket.ConnectionSocketFactory; |
|||
import org.apache.http.conn.socket.PlainConnectionSocketFactory; |
|||
import org.apache.http.conn.ssl.SSLConnectionSocketFactory; |
|||
import org.apache.http.cookie.Cookie; |
|||
import org.apache.http.entity.StringEntity; |
|||
import org.apache.http.impl.client.BasicCookieStore; |
|||
import org.apache.http.impl.client.CloseableHttpClient; |
|||
import org.apache.http.impl.client.HttpClientBuilder; |
|||
import org.apache.http.impl.conn.PoolingHttpClientConnectionManager; |
|||
import org.apache.http.message.BasicNameValuePair; |
|||
import org.apache.http.util.EntityUtils; |
|||
|
|||
import javax.net.ssl.SSLContext; |
|||
import javax.net.ssl.TrustManager; |
|||
import javax.net.ssl.X509TrustManager; |
|||
import java.io.IOException; |
|||
import java.io.UnsupportedEncodingException; |
|||
import java.security.cert.CertificateException; |
|||
import java.security.cert.X509Certificate; |
|||
import java.util.*; |
|||
|
|||
/** |
|||
* @Description: httpclient常用方法封装, |
|||
* @Author: ggf |
|||
* @Date: 2020/06/06 |
|||
*/ |
|||
public class HttpClientUtils { |
|||
/** |
|||
* 编码方式 |
|||
*/ |
|||
private static final String ENCODING = "UTF-8"; |
|||
/** |
|||
* 连接超时时间，60秒 |
|||
*/ |
|||
public static final int DEFAULT_CONNECT_TIMEOUT = 6000; |
|||
/** |
|||
* socket连接超时时间，60秒 |
|||
*/ |
|||
public static final int DEFAULT_READ_TIMEOUT = 6000; |
|||
/** |
|||
* 请求超时时间，60秒 |
|||
*/ |
|||
public static final int DEFAULT_CONNECT_REQUEST_TIMEOUT = 6000; |
|||
/** |
|||
* 最大连接数,默认为2 |
|||
*/ |
|||
private static final int MAX_TOTAL = 64; |
|||
/** |
|||
* 设置指向特定路由的并发连接总数，默认为2 |
|||
*/ |
|||
private static final int MAX_PER_ROUTE = 32; |
|||
|
|||
private static RequestConfig requestConfig; |
|||
private static PoolingHttpClientConnectionManager connectionManager; |
|||
private static BasicCookieStore cookieStore; |
|||
private static HttpClientBuilder httpBuilder; |
|||
private static CloseableHttpClient httpClient; |
|||
private static CloseableHttpClient httpsClient; |
|||
private static SSLContext sslContext; |
|||
|
|||
/** |
|||
* 创建SSLContext对象，用来绕过https证书认证实现访问。 |
|||
*/ |
|||
static { |
|||
try { |
|||
sslContext = SSLContext.getInstance("TLS"); |
|||
// 实现一个X509TrustManager接口，用于绕过验证，不用修改里面的方法
|
|||
X509TrustManager tm = new X509TrustManager() { |
|||
@Override |
|||
public void checkClientTrusted(X509Certificate[] chain, String authType) |
|||
throws CertificateException { |
|||
} |
|||
@Override |
|||
public void checkServerTrusted(X509Certificate[] chain, String authType) |
|||
throws CertificateException { |
|||
} |
|||
@Override |
|||
public X509Certificate[] getAcceptedIssuers() { |
|||
return null; |
|||
} |
|||
}; |
|||
sslContext.init(null, new TrustManager[] {tm}, null); |
|||
} catch (Exception e) { |
|||
e.printStackTrace(); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 初始化httpclient对象，以及在创建httpclient对象之前的一些自定义配置。 |
|||
*/ |
|||
static { |
|||
// 自定义配置信息
|
|||
requestConfig = RequestConfig.custom() |
|||
.setSocketTimeout(DEFAULT_READ_TIMEOUT) |
|||
.setConnectTimeout(DEFAULT_CONNECT_TIMEOUT) |
|||
.setConnectionRequestTimeout(DEFAULT_CONNECT_REQUEST_TIMEOUT) |
|||
.build(); |
|||
//设置协议http和https对应的处理socket链接工厂的对象
|
|||
Registry<ConnectionSocketFactory> socketFactoryRegistry = RegistryBuilder.<ConnectionSocketFactory> create() |
|||
.register("http", new PlainConnectionSocketFactory()) |
|||
.register("https", new SSLConnectionSocketFactory(sslContext)) |
|||
.build(); |
|||
connectionManager = new PoolingHttpClientConnectionManager(socketFactoryRegistry); |
|||
// 设置cookie存储对像，在需要获取cookie信息时，可以使用这个对象。
|
|||
cookieStore = new BasicCookieStore(); |
|||
// 设置最大连接数
|
|||
connectionManager.setMaxTotal(MAX_TOTAL); |
|||
// 设置路由并发数
|
|||
connectionManager.setDefaultMaxPerRoute(MAX_PER_ROUTE); |
|||
httpBuilder = HttpClientBuilder.create(); |
|||
httpBuilder.setDefaultRequestConfig(requestConfig); |
|||
httpBuilder.setConnectionManager(connectionManager); |
|||
httpBuilder.setDefaultCookieStore(cookieStore); |
|||
// 实例化http 和 https的对象。
|
|||
httpClient = httpBuilder.build(); |
|||
httpsClient = httpBuilder.build(); |
|||
} |
|||
|
|||
/** |
|||
* 封装无参数的get请求（http） |
|||
* @param url 请求url |
|||
* @return 返回对象HttpClientResult |
|||
*/ |
|||
public static HttpClientResult doGet(String url) { |
|||
return doGet(url, false); |
|||
} |
|||
|
|||
/** |
|||
* 封装无参get请求，支持https协议 |
|||
* @param url 请求url |
|||
* @param https 请求的是否是https协议，是：true 否false |
|||
* @return |
|||
*/ |
|||
public static HttpClientResult doGet(String url, boolean https){ |
|||
return doGet(url, null, null, https); |
|||
} |
|||
|
|||
/** |
|||
* 封装带参数的get请求，支持https协议 |
|||
* @param url 请求url |
|||
* @param params 请求参数 |
|||
* @param https 是否是https协议 |
|||
*/ |
|||
public static HttpClientResult doGet(String url, Map<String, String> params, boolean https){ |
|||
return doGet(url, null, params, https); |
|||
} |
|||
|
|||
/** |
|||
* 封装带参数和带请求头信息的GET方法，支持https协议请求 |
|||
* @param url 请求url |
|||
* @param headers 请求头信息 |
|||
* @param params 请求参数 |
|||
* @param https 是否使用https协议 |
|||
*/ |
|||
public static HttpClientResult doGet(String url, Map<String, String> headers, Map<String, String> params, boolean https){ |
|||
// 创建HttpGet
|
|||
HttpGet httpGet = null; |
|||
// 创建httpResponse对象
|
|||
CloseableHttpResponse httpResponse = null; |
|||
try { |
|||
// 创建访问的地址
|
|||
URIBuilder uriBuilder = new URIBuilder(url); |
|||
if (params != null) { |
|||
Set<Map.Entry<String, String>> entrySet = params.entrySet(); |
|||
for (Map.Entry<String, String> entry : entrySet) { |
|||
uriBuilder.setParameter(entry.getKey(), entry.getValue()); |
|||
} |
|||
} |
|||
// 创建HTTP对象
|
|||
httpGet = new HttpGet(uriBuilder.build()); |
|||
httpGet.setConfig(requestConfig); |
|||
// 设置请求头
|
|||
setHeader(headers, httpGet); |
|||
// 使用不同的协议进行请求，返回自定义的响应对象
|
|||
if (https) { |
|||
return getHttpClientResult(httpResponse, httpsClient, httpGet); |
|||
} else { |
|||
return getHttpClientResult(httpResponse, httpClient, httpGet); |
|||
} |
|||
} catch (Exception e) { |
|||
e.printStackTrace(); |
|||
} finally { |
|||
// 释放资源
|
|||
if (httpGet != null) { |
|||
httpGet.releaseConnection(); |
|||
} |
|||
release(httpResponse); |
|||
} |
|||
|
|||
return null; |
|||
} |
|||
|
|||
/** |
|||
* POST不带参数，只支持http协议 |
|||
* @param url 请求url |
|||
*/ |
|||
public static HttpClientResult doPost(String url) { |
|||
return doPost(url, Boolean.FALSE); |
|||
} |
|||
|
|||
/** |
|||
* 封装不带参数的post请求，支持https协议 |
|||
* @param url 请求url |
|||
* @param https 是否是https协议 |
|||
*/ |
|||
public static HttpClientResult doPost(String url, boolean https) { |
|||
return doPost(url, null, (Map<String, String>)null, https); |
|||
} |
|||
|
|||
/** |
|||
* 带参数的post请求，支持https协议 |
|||
* @param url 请求url |
|||
* @param params 请求参数 |
|||
* @param https 是否是https协议 |
|||
*/ |
|||
public static HttpClientResult doPost(String url, Map<String, String> params, boolean https) { |
|||
return doPost(url, null, params, https); |
|||
} |
|||
|
|||
/** |
|||
* 带参数和请求头的POST请求，支持https |
|||
* |
|||
* @param url 请求url |
|||
* @param headers 请求头 |
|||
* @param params 请求参数，参数为K=V格式 |
|||
* @param https 是否https协议 |
|||
*/ |
|||
public static HttpClientResult doPost(String url, Map<String, String> headers, Map<String, String> params, boolean https) { |
|||
// 创建HTTP对象
|
|||
HttpPost httpPost = new HttpPost(url); |
|||
httpPost.setConfig(requestConfig); |
|||
// 设置请求头
|
|||
setHeader(headers, httpPost); |
|||
// 封装请求参数
|
|||
setParam(params, httpPost); |
|||
// 创建httpResponse对象
|
|||
CloseableHttpResponse httpResponse = null; |
|||
try { |
|||
if (https) { |
|||
return getHttpClientResult(httpResponse, httpsClient, httpPost); |
|||
} else { |
|||
return getHttpClientResult(httpResponse, httpClient, httpPost); |
|||
} |
|||
} finally { |
|||
httpPost.releaseConnection(); |
|||
release(httpResponse); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 带参数、带请求头的POST请求，支持https协议 |
|||
* |
|||
* @param url 请求url |
|||
* @param headers 请求头 |
|||
* @param json 请求参数为json格式 |
|||
* @param https 是否使用https协议 |
|||
* @throws Exception |
|||
*/ |
|||
public static HttpClientResult doPost(String url, Map<String, String> headers, String json, boolean https) { |
|||
// 创建HTTP对象
|
|||
HttpPost httpPost = new HttpPost(url); |
|||
httpPost.setConfig(requestConfig); |
|||
// 设置请求头
|
|||
setHeader(headers, httpPost); |
|||
StringEntity stringEntity = new StringEntity(json, ENCODING); |
|||
stringEntity.setContentEncoding(ENCODING); |
|||
httpPost.setEntity(stringEntity); |
|||
// 创建httpResponse对象
|
|||
CloseableHttpResponse httpResponse = null; |
|||
try { |
|||
if (https) { |
|||
return getHttpClientResult(httpResponse, httpsClient, httpPost); |
|||
} else { |
|||
return getHttpClientResult(httpResponse, httpClient, httpPost); |
|||
} |
|||
} finally { |
|||
httpPost.releaseConnection(); |
|||
release(httpResponse); |
|||
} |
|||
} |
|||
|
|||
|
|||
/** |
|||
* 发送put请求；不带请求参数 |
|||
* |
|||
* @param url 请求地址 |
|||
* @return |
|||
* @throws Exception |
|||
*/ |
|||
public static HttpClientResult doPut(String url) { |
|||
return doPut(url); |
|||
} |
|||
|
|||
/** |
|||
* 发送put请求；带请求参数 |
|||
* |
|||
* @param url 请求地址 |
|||
* @param params 参数集合 |
|||
* @return |
|||
* @throws Exception |
|||
*/ |
|||
public static HttpClientResult doPut(String url, Map<String, String> params) { |
|||
HttpPut httpPut = new HttpPut(url); |
|||
httpPut.setConfig(requestConfig); |
|||
setParam(params, httpPut); |
|||
CloseableHttpResponse httpResponse = null; |
|||
try { |
|||
return getHttpClientResult(httpResponse, httpClient, httpPut); |
|||
} finally { |
|||
httpPut.releaseConnection(); |
|||
release(httpResponse); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 发送delete请求，不带请求参数 |
|||
* |
|||
* @param url 请求url |
|||
* @return |
|||
* @throws Exception |
|||
*/ |
|||
public static HttpClientResult doDelete(String url) { |
|||
HttpDelete httpDelete = new HttpDelete(url); |
|||
httpDelete.setConfig(requestConfig); |
|||
CloseableHttpResponse httpResponse = null; |
|||
try { |
|||
return getHttpClientResult(httpResponse, httpClient, httpDelete); |
|||
} finally { |
|||
httpDelete.releaseConnection(); |
|||
release(httpResponse); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 发送delete请求，带请求参数， 支持https协议 |
|||
* |
|||
* @param url 请求url |
|||
* @param params 请求参数 |
|||
* @param https 是否https |
|||
* @return |
|||
* @throws Exception |
|||
*/ |
|||
public static HttpClientResult doDelete(String url, Map<String, String> params, boolean https) { |
|||
if (params == null) { |
|||
params = new HashMap<String, String>(); |
|||
} |
|||
params.put("_method", "delete"); |
|||
return doPost(url, params, https); |
|||
} |
|||
|
|||
/** |
|||
* 获取cookie信息 |
|||
* @return 返回所有cookie集合 |
|||
*/ |
|||
public static List<Cookie> getCookies() { |
|||
return cookieStore.getCookies(); |
|||
} |
|||
|
|||
|
|||
/** |
|||
* 设置封装请求头 |
|||
* |
|||
* @param params 头信息 |
|||
* @param httpMethod 请求对象 |
|||
*/ |
|||
public static void setHeader(Map<String, String> params, HttpRequestBase httpMethod) { |
|||
// 封装请求头
|
|||
if (null != params && !params.isEmpty()) { |
|||
Set<Map.Entry<String, String>> entrySet = params.entrySet(); |
|||
for (Map.Entry<String, String> entry : entrySet) { |
|||
// 设置到请求头到HttpRequestBase对象中
|
|||
httpMethod.setHeader(entry.getKey(), entry.getValue()); |
|||
} |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 封装请求参数 |
|||
* |
|||
* @param params 请求参数 |
|||
* @param httpMethod 请求方法 |
|||
*/ |
|||
public static void setParam(Map<String, String> params, HttpEntityEnclosingRequestBase httpMethod) { |
|||
// 封装请求参数
|
|||
if (null != params && !params.isEmpty()) { |
|||
List<NameValuePair> nvps = new ArrayList<NameValuePair>(); |
|||
Set<Map.Entry<String, String>> entrySet = params.entrySet(); |
|||
for (Map.Entry<String, String> entry : entrySet) { |
|||
nvps.add(new BasicNameValuePair(entry.getKey(), entry.getValue())); |
|||
} |
|||
|
|||
UrlEncodedFormEntity entity = null; |
|||
try { |
|||
entity = new UrlEncodedFormEntity(nvps, ENCODING); |
|||
} catch (UnsupportedEncodingException e) { |
|||
e.printStackTrace(); |
|||
} |
|||
// 设置到请求的http对象中
|
|||
httpMethod.setEntity(entity); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 获得响应结果 |
|||
* |
|||
* @param httpResponse 响应对象 |
|||
* @param httpClient httpclient对象 |
|||
* @param httpMethod 请求方法 |
|||
* @return |
|||
* @throws Exception |
|||
*/ |
|||
public static HttpClientResult getHttpClientResult(CloseableHttpResponse httpResponse, CloseableHttpClient httpClient, HttpRequestBase httpMethod) { |
|||
try { |
|||
// 执行请求
|
|||
httpResponse = httpClient.execute(httpMethod); |
|||
// 获取返回结果
|
|||
if (httpResponse != null && httpResponse.getStatusLine() != null) { |
|||
String content = ""; |
|||
if (httpResponse.getEntity() != null) { |
|||
content = EntityUtils.toString(httpResponse.getEntity(), ENCODING); |
|||
} |
|||
// return new HttpClientResult(httpResponse.getStatusLine().getStatusCode(), content);
|
|||
} |
|||
} catch (IOException e) { |
|||
e.printStackTrace(); |
|||
} |
|||
// return new HttpClientResult(HttpStatus.SC_INTERNAL_SERVER_ERROR);
|
|||
return null; |
|||
} |
|||
|
|||
/** |
|||
* 释放资源 |
|||
* |
|||
* @param httpResponse 响应对象 |
|||
*/ |
|||
public static void release(CloseableHttpResponse httpResponse) { |
|||
// 释放资源
|
|||
if (httpResponse != null) { |
|||
try { |
|||
httpResponse.close(); |
|||
} catch (IOException e) { |
|||
e.printStackTrace(); |
|||
} |
|||
} |
|||
} |
|||
} |
|||
@ -1,34 +0,0 @@ |
|||
package com.visual.open.anpr.model; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
public class LocationPoint implements Serializable { |
|||
|
|||
/**坐标X的值**/ |
|||
private int x; |
|||
/**坐标Y的值**/ |
|||
private int y; |
|||
|
|||
public LocationPoint(){} |
|||
|
|||
public LocationPoint(float x, float y) { |
|||
this.x = (int)x; |
|||
this.y = (int)y; |
|||
} |
|||
|
|||
public int getX() { |
|||
return x; |
|||
} |
|||
|
|||
public void setX(int x) { |
|||
this.x = x; |
|||
} |
|||
|
|||
public int getY() { |
|||
return y; |
|||
} |
|||
|
|||
public void setY(int y) { |
|||
this.y = y; |
|||
} |
|||
} |
|||
@ -1,17 +0,0 @@ |
|||
package com.visual.open.anpr.model; |
|||
|
|||
import java.util.HashMap; |
|||
|
|||
public class MapParam extends HashMap<String, Object> { |
|||
|
|||
public static MapParam build(){ |
|||
return new MapParam(); |
|||
} |
|||
|
|||
|
|||
public MapParam put(String key, Object value){ |
|||
super.put(key, value); |
|||
return this; |
|||
} |
|||
|
|||
} |
|||
@ -1,31 +0,0 @@ |
|||
package com.visual.open.anpr.model; |
|||
|
|||
public enum PlateColor { |
|||
|
|||
BLACK("黑色"), |
|||
BLUE("蓝色"), |
|||
GREEN("绿色"), |
|||
WHITE("白色"), |
|||
YELLOW("黄色"), |
|||
UNKNOWN("未知"); |
|||
|
|||
private String name; |
|||
|
|||
PlateColor(String name){ |
|||
this.name = name; |
|||
} |
|||
|
|||
public String getName(){ |
|||
return this.name; |
|||
} |
|||
|
|||
public static PlateColor valueOfName(String name){ |
|||
for (PlateColor item : PlateColor.values()){ |
|||
if(item.name.equals(name)){ |
|||
return item; |
|||
} |
|||
} |
|||
return UNKNOWN; |
|||
} |
|||
|
|||
} |
|||
@ -1,8 +0,0 @@ |
|||
package com.visual.open.anpr.model; |
|||
|
|||
public enum PlateLayout { |
|||
|
|||
SINGLE, //单排
|
|||
DOUBLE, //双排
|
|||
UNKNOWN; //未知
|
|||
} |
|||
@ -1,96 +0,0 @@ |
|||
package com.visual.open.anpr.model; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
public class PlateLocation implements Serializable { |
|||
|
|||
/**左上角坐标值**/ |
|||
private LocationPoint leftTop; |
|||
/**右上角坐标**/ |
|||
private LocationPoint rightTop; |
|||
/**右下角坐标**/ |
|||
private LocationPoint rightBottom; |
|||
/**左下角坐标**/ |
|||
private LocationPoint leftBottom; |
|||
|
|||
public LocationPoint getLeftTop() { |
|||
return leftTop; |
|||
} |
|||
|
|||
public void setLeftTop(LocationPoint leftTop) { |
|||
this.leftTop = leftTop; |
|||
} |
|||
|
|||
public LocationPoint getRightTop() { |
|||
return rightTop; |
|||
} |
|||
|
|||
public void setRightTop(LocationPoint rightTop) { |
|||
this.rightTop = rightTop; |
|||
} |
|||
|
|||
public LocationPoint getRightBottom() { |
|||
return rightBottom; |
|||
} |
|||
|
|||
public void setRightBottom(LocationPoint rightBottom) { |
|||
this.rightBottom = rightBottom; |
|||
} |
|||
|
|||
public LocationPoint getLeftBottom() { |
|||
return leftBottom; |
|||
} |
|||
|
|||
public void setLeftBottom(LocationPoint leftBottom) { |
|||
this.leftBottom = leftBottom; |
|||
} |
|||
|
|||
|
|||
/** |
|||
* x的最小坐标 |
|||
* @return |
|||
*/ |
|||
public int minX(){ |
|||
return Math.min(Math.min(Math.min(leftTop.getX(), rightTop.getX()), rightBottom.getX()), leftBottom.getX()); |
|||
} |
|||
|
|||
/** |
|||
* y的最小坐标 |
|||
* @return |
|||
*/ |
|||
public int minY(){ |
|||
return Math.min(Math.min(Math.min(leftTop.getY(), rightTop.getY()), rightBottom.getY()), leftBottom.getY()); |
|||
} |
|||
|
|||
/** |
|||
* x的最大坐标 |
|||
* @return |
|||
*/ |
|||
public int maxX(){ |
|||
return Math.max(Math.max(Math.max(leftTop.getX(), rightTop.getX()), rightBottom.getX()), leftBottom.getX()); |
|||
} |
|||
|
|||
/** |
|||
* y的最大坐标 |
|||
* @return |
|||
*/ |
|||
public int maxY(){ |
|||
return Math.max(Math.max(Math.max(leftTop.getY(), rightTop.getY()), rightBottom.getY()), leftBottom.getY()); |
|||
} |
|||
|
|||
/** |
|||
* 获取宽度 |
|||
* @return |
|||
*/ |
|||
public float width(){ |
|||
return (float) Math.sqrt(Math.pow((rightTop.getX()-leftTop.getX()), 2)+Math.pow((rightTop.getY()-leftTop.getY()), 2)); |
|||
} |
|||
|
|||
/** |
|||
* 获取高度 |
|||
* @return |
|||
*/ |
|||
public float height(){ |
|||
return (float) Math.sqrt(Math.pow((rightTop.getX()-rightBottom.getX()), 2)+Math.pow((rightTop.getY()-rightBottom.getY()), 2)); |
|||
} |
|||
} |
|||
@ -1,40 +0,0 @@ |
|||
package com.visual.open.anpr.model; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
public class Recognition<ExtendsVo extends Recognition<ExtendsVo>> implements Serializable { |
|||
|
|||
/**图像Base64编码值**/ |
|||
private String image; |
|||
|
|||
/**搜索条数：默认5**/ |
|||
private Integer limit = 5; |
|||
|
|||
/** |
|||
* 构建比对对象 |
|||
* @return |
|||
*/ |
|||
public static Recognition build(){ |
|||
return new Recognition(); |
|||
} |
|||
|
|||
|
|||
public String getImage() { |
|||
return image; |
|||
} |
|||
|
|||
public ExtendsVo setImage(String image) { |
|||
this.image = image; |
|||
return (ExtendsVo) this; |
|||
} |
|||
|
|||
public Integer getLimit() { |
|||
return limit; |
|||
} |
|||
|
|||
public ExtendsVo setLimit(Integer limit) { |
|||
this.limit = limit; |
|||
return (ExtendsVo) this; |
|||
} |
|||
|
|||
} |
|||
@ -1,36 +0,0 @@ |
|||
package com.visual.open.anpr.model; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
public class RecognitionInfo implements Serializable { |
|||
/**车牌布局，单排还是双排**/ |
|||
private PlateLayout layout; |
|||
/**车牌文本信息**/ |
|||
private String plateNo; |
|||
/**车牌的颜色信息**/ |
|||
private PlateColor plateColor; |
|||
|
|||
public PlateLayout getLayout() { |
|||
return layout; |
|||
} |
|||
|
|||
public void setLayout(PlateLayout layout) { |
|||
this.layout = layout; |
|||
} |
|||
|
|||
public String getPlateNo() { |
|||
return plateNo; |
|||
} |
|||
|
|||
public void setPlateNo(String plateNo) { |
|||
this.plateNo = plateNo; |
|||
} |
|||
|
|||
public PlateColor getPlateColor() { |
|||
return plateColor; |
|||
} |
|||
|
|||
public void setPlateColor(PlateColor plateColor) { |
|||
this.plateColor = plateColor; |
|||
} |
|||
} |
|||
@ -1,40 +0,0 @@ |
|||
package com.visual.open.anpr.model; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
public class RecognitionRep implements Serializable { |
|||
|
|||
/**车牌置信分数**/ |
|||
private Float score; |
|||
|
|||
/**车牌位置信息**/ |
|||
private PlateLocation location; |
|||
|
|||
/**车牌识别信息**/ |
|||
private RecognitionInfo recognition; |
|||
|
|||
public Float getScore() { |
|||
return score; |
|||
} |
|||
|
|||
public void setScore(Float score) { |
|||
this.score = score; |
|||
} |
|||
|
|||
public PlateLocation getLocation() { |
|||
return location; |
|||
} |
|||
|
|||
public void setLocation(PlateLocation location) { |
|||
this.location = location; |
|||
} |
|||
|
|||
public RecognitionInfo getRecognition() { |
|||
return recognition; |
|||
} |
|||
|
|||
public void setRecognition(RecognitionInfo recognition) { |
|||
this.recognition = recognition; |
|||
} |
|||
|
|||
} |
|||
@ -1,13 +0,0 @@ |
|||
package com.visual.open.anpr.model; |
|||
|
|||
public class RecognitionReq extends Recognition<RecognitionReq>{ |
|||
|
|||
/** |
|||
* 构建比对对象 |
|||
* @return |
|||
*/ |
|||
public static RecognitionReq build(){ |
|||
return new RecognitionReq(); |
|||
} |
|||
|
|||
} |
|||
@ -1,62 +0,0 @@ |
|||
package com.visual.open.anpr.model; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
/** |
|||
* des:接口返回对象 |
|||
* @author diven |
|||
* @date 上午9:34 2018/7/12 |
|||
*/ |
|||
public class Response<T> implements Serializable{ |
|||
|
|||
private static final long serialVersionUID = -6919611972884058300L; |
|||
|
|||
private Integer code = -1; |
|||
private String message; |
|||
private T data; |
|||
|
|||
public Response(){} |
|||
|
|||
public Response(Integer code, String message, T data) { |
|||
if(null != code) { |
|||
this.code = code; |
|||
} |
|||
this.message = message; |
|||
this.data = data; |
|||
} |
|||
|
|||
public Integer getCode() { |
|||
return code; |
|||
} |
|||
|
|||
public void setCode(Integer code) { |
|||
if(null != code){ |
|||
this.code = code; |
|||
} |
|||
} |
|||
|
|||
public String getMessage() { |
|||
return message; |
|||
} |
|||
|
|||
public void setMessage(String message) { |
|||
this.message = message; |
|||
} |
|||
|
|||
public T getData() { |
|||
return data; |
|||
} |
|||
|
|||
public void setData(T data) { |
|||
this.data = data; |
|||
} |
|||
|
|||
public boolean ok(){ |
|||
return new Integer(0).equals(code); |
|||
} |
|||
|
|||
@Override |
|||
public String toString() { |
|||
return "Response{" + "code=" + code + ", message='" + message + '\'' + ", data=" + data + '}'; |
|||
} |
|||
} |
|||
@ -1,48 +0,0 @@ |
|||
package com.visual.open.anpr.utils; |
|||
|
|||
import org.apache.commons.codec.binary.Base64; |
|||
|
|||
import java.io.ByteArrayOutputStream; |
|||
import java.io.FileInputStream; |
|||
import java.io.IOException; |
|||
import java.io.InputStream; |
|||
|
|||
public class Base64Util { |
|||
|
|||
public static String encode(byte[] binaryData) { |
|||
byte[] bytes = Base64.encodeBase64(binaryData); |
|||
return new String(bytes); |
|||
} |
|||
|
|||
public static String encode(InputStream in) { |
|||
// 读取图片字节数组
|
|||
try { |
|||
ByteArrayOutputStream swapStream = new ByteArrayOutputStream(); |
|||
byte[] buff = new byte[100]; |
|||
int rc; |
|||
while ((rc = in.read(buff, 0, 100)) > 0) { |
|||
swapStream.write(buff, 0, rc); |
|||
} |
|||
return encode(swapStream.toByteArray()); |
|||
} catch (Exception e) { |
|||
throw new RuntimeException(e); |
|||
} finally { |
|||
if (in != null) { |
|||
try { |
|||
in.close(); |
|||
} catch (IOException e) { |
|||
e.printStackTrace(); |
|||
} |
|||
} |
|||
} |
|||
} |
|||
|
|||
public static String encode(String filePath){ |
|||
try { |
|||
return encode(new FileInputStream(filePath)); |
|||
} catch (Exception e) { |
|||
throw new RuntimeException(e); |
|||
} |
|||
} |
|||
|
|||
} |
|||
@ -1,40 +0,0 @@ |
|||
<?xml version="1.0" encoding="UTF-8"?> |
|||
<project xmlns="http://maven.apache.org/POM/4.0.0" |
|||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" |
|||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> |
|||
<parent> |
|||
<artifactId>open-anpr</artifactId> |
|||
<groupId>com.visual.open.anpr</groupId> |
|||
<version>1.1.0</version> |
|||
</parent> |
|||
<modelVersion>4.0.0</modelVersion> |
|||
|
|||
<artifactId>open-anpr-core</artifactId> |
|||
|
|||
<dependencies> |
|||
<dependency> |
|||
<groupId>org.openpnp</groupId> |
|||
<artifactId>opencv</artifactId> |
|||
</dependency> |
|||
|
|||
<dependency> |
|||
<groupId>com.microsoft.onnxruntime</groupId> |
|||
<artifactId>onnxruntime</artifactId> |
|||
</dependency> |
|||
|
|||
<dependency> |
|||
<groupId>com.alibaba</groupId> |
|||
<artifactId>fastjson</artifactId> |
|||
</dependency> |
|||
|
|||
<dependency> |
|||
<groupId>org.apache.commons</groupId> |
|||
<artifactId>commons-collections4</artifactId> |
|||
</dependency> |
|||
|
|||
<dependency> |
|||
<groupId>org.apache.commons</groupId> |
|||
<artifactId>commons-math3</artifactId> |
|||
</dependency> |
|||
</dependencies> |
|||
</project> |
|||
@ -1,10 +0,0 @@ |
|||
package com.visual.open.anpr.core.base; |
|||
|
|||
import java.util.Map; |
|||
import com.visual.open.anpr.core.domain.ImageMat; |
|||
import com.visual.open.anpr.core.domain.PlateInfo.ParseInfo; |
|||
|
|||
public interface PlateRecognition { |
|||
|
|||
ParseInfo inference(ImageMat image, Boolean single, Map<String, Object> params); |
|||
} |
|||
@ -1,131 +0,0 @@ |
|||
package com.visual.open.anpr.core.domain; |
|||
|
|||
import org.opencv.core.CvType; |
|||
import org.opencv.core.Mat; |
|||
|
|||
import javax.imageio.ImageIO; |
|||
import java.awt.*; |
|||
import java.awt.image.BufferedImage; |
|||
import java.awt.image.DataBufferByte; |
|||
import java.io.File; |
|||
import java.io.IOException; |
|||
|
|||
|
|||
public class DrawImage { |
|||
|
|||
private BufferedImage image; |
|||
|
|||
private DrawImage(BufferedImage image){ |
|||
this.image = image; |
|||
} |
|||
|
|||
public static DrawImage build(String image){ |
|||
try { |
|||
return build(ImageIO.read(new File(image))); |
|||
} catch (IOException e) { |
|||
throw new RuntimeException(e); |
|||
} |
|||
} |
|||
|
|||
public static DrawImage build(BufferedImage image){ |
|||
return new DrawImage(image); |
|||
} |
|||
|
|||
private static int getLength(String text) { |
|||
int textLength = text.length(); |
|||
int length = textLength; |
|||
for (int i = 0; i < textLength; i++) { |
|||
if (String.valueOf(text.charAt(i)).getBytes().length > 1) { |
|||
length++; |
|||
} |
|||
} |
|||
return (length % 2 == 0) ? length / 2 : length / 2 + 1; |
|||
} |
|||
|
|||
public DrawImage drawRect(Rect rect, int lineWidth, Color color){ |
|||
Graphics2D g = (Graphics2D)image.getGraphics(); |
|||
g.setColor(color); |
|||
g.setStroke(new BasicStroke(lineWidth)); |
|||
g.drawRect(rect.x, rect.y, rect.width, rect.height); |
|||
return this; |
|||
} |
|||
|
|||
public DrawImage drawLine(Point point1, Point point2, int lineWidth, Color color){ |
|||
Graphics2D g = (Graphics2D)image.getGraphics(); |
|||
g.setColor(color); |
|||
g.setStroke(new BasicStroke(lineWidth)); |
|||
g.drawLine(point1.x, point1.y, point2.x, point2.y); |
|||
return this; |
|||
} |
|||
|
|||
public DrawImage drawText(String text, Point point, int fontSize, Color color){ |
|||
Graphics2D g = image.createGraphics(); |
|||
g.setColor(color); |
|||
g.setFont(new Font("微软雅黑", Font.PLAIN, fontSize)); |
|||
g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, 1)); |
|||
int width_1 = fontSize * getLength(text); |
|||
int height_1 = fontSize; |
|||
int widthDiff = image.getWidth() - width_1; |
|||
int heightDiff = image.getHeight() - height_1; |
|||
if(point.x < 0){ |
|||
point.x = widthDiff / 2; |
|||
}else if(point.x > widthDiff){ |
|||
point.x = widthDiff; |
|||
} |
|||
if(point.y < 0){ |
|||
point.y = heightDiff / 2; |
|||
}else if(point.y > heightDiff){ |
|||
point.y = heightDiff; |
|||
} |
|||
g.drawString(text, point.x, point.y + height_1); |
|||
return this; |
|||
} |
|||
|
|||
public Mat toMat(){ |
|||
try { |
|||
if(image.getType() != BufferedImage.TYPE_3BYTE_BGR){ |
|||
BufferedImage temp = new BufferedImage(image.getWidth(), image.getHeight(), BufferedImage.TYPE_3BYTE_BGR); |
|||
Graphics2D g = temp.createGraphics(); |
|||
try { |
|||
g.setComposite(AlphaComposite.Src); |
|||
g.drawImage(image, 0, 0, null); |
|||
} finally { |
|||
g.dispose(); |
|||
} |
|||
image = temp; |
|||
} |
|||
byte[] pixels = ((DataBufferByte) image.getRaster().getDataBuffer()).getData(); |
|||
Mat mat = Mat.eye(image.getHeight(), image.getWidth(), CvType.CV_8UC3); |
|||
mat.put(0, 0, pixels); |
|||
return mat; |
|||
}catch (Exception e){ |
|||
throw new RuntimeException(e); |
|||
} |
|||
} |
|||
|
|||
public static class Rect{ |
|||
public int x; |
|||
public int y; |
|||
public int width; |
|||
public int height; |
|||
|
|||
public Rect(int x, int y, int width, int height) { |
|||
this.x = x; |
|||
this.y = y; |
|||
this.width = width; |
|||
this.height = height; |
|||
} |
|||
|
|||
} |
|||
|
|||
public static class Point{ |
|||
public int x; |
|||
public int y; |
|||
|
|||
public Point(int x, int y) { |
|||
this.x = x; |
|||
this.y = y; |
|||
} |
|||
} |
|||
|
|||
} |
|||
@ -1,128 +0,0 @@ |
|||
package com.visual.open.anpr.core.utils; |
|||
|
|||
import com.alibaba.fastjson.JSON; |
|||
import com.alibaba.fastjson.JSONArray; |
|||
import com.alibaba.fastjson.JSONObject; |
|||
import com.alibaba.fastjson.TypeReference; |
|||
import com.alibaba.fastjson.serializer.SerializerFeature; |
|||
|
|||
import java.util.List; |
|||
import java.util.Map; |
|||
|
|||
public class JsonUtil { |
|||
|
|||
/** |
|||
* 将Bean转化为json字符串 |
|||
* |
|||
* @param obj bean对象 |
|||
* @return json |
|||
*/ |
|||
public static String toString(Object obj) { |
|||
return toString(obj, false, false); |
|||
} |
|||
|
|||
public static String toSimpleString(Object obj) { |
|||
return toString(obj, false, true); |
|||
} |
|||
|
|||
/** |
|||
* 将Bean转化为json字符串 |
|||
* |
|||
* @param obj bean对象 |
|||
* @param prettyFormat 是否格式化 |
|||
* @return json |
|||
*/ |
|||
public static String toString(Object obj, boolean prettyFormat, boolean noNull) { |
|||
if (prettyFormat) { |
|||
if (noNull) { |
|||
return JSON.toJSONString(obj, SerializerFeature.DisableCircularReferenceDetect, SerializerFeature.PrettyFormat); |
|||
} else { |
|||
return JSON.toJSONString(obj, SerializerFeature.WriteMapNullValue, SerializerFeature.WriteNullListAsEmpty, SerializerFeature.DisableCircularReferenceDetect, SerializerFeature.PrettyFormat); |
|||
} |
|||
} else { |
|||
if (noNull) { |
|||
return JSON.toJSONString(obj, SerializerFeature.DisableCircularReferenceDetect); |
|||
} else { |
|||
return JSON.toJSONString(obj, SerializerFeature.WriteMapNullValue, SerializerFeature.WriteNullListAsEmpty, SerializerFeature.DisableCircularReferenceDetect); |
|||
} |
|||
} |
|||
} |
|||
|
|||
|
|||
/** |
|||
* 将字符串转换为Entity |
|||
* |
|||
* @param json 数据字符串 |
|||
* @param clazz Entity class |
|||
* @return |
|||
*/ |
|||
public static <T> T toEntity(String json, Class<T> clazz) { |
|||
return JSON.parseObject(json, clazz); |
|||
} |
|||
|
|||
/** |
|||
* 将字符串转换为Entity |
|||
* |
|||
* @param json 数据字符串 |
|||
* @param typeReference Entity class |
|||
* @return |
|||
*/ |
|||
public static <T> T toEntity(String json, TypeReference<T> typeReference) { |
|||
return JSON.parseObject(json, typeReference); |
|||
} |
|||
|
|||
/** |
|||
* 将字符串转换为Map |
|||
* |
|||
* @param json 数据字符串 |
|||
* @return Map |
|||
*/ |
|||
public static Map<String, Object> toMap(String json) { |
|||
return JSON.parseObject(json, new TypeReference<Map<String, Object>>() { |
|||
}); |
|||
} |
|||
|
|||
/** |
|||
* 将字符串转换为List<T> |
|||
* |
|||
* @param json 数据字符串 |
|||
* @param collectionClass 泛型 |
|||
* @return list<T> |
|||
*/ |
|||
public static <T> List<T> toList(String json, Class<T> collectionClass) { |
|||
return JSON.parseArray(json, collectionClass); |
|||
} |
|||
|
|||
/** |
|||
* 将字符串转换为List<Map<String, Object>> |
|||
* |
|||
* @param json 数据字符串 |
|||
* @return list<map> |
|||
*/ |
|||
public static List<Map<String, Object>> toListMap(String json) { |
|||
return JSON.parseObject(json, new TypeReference<List<Map<String, Object>>>() { |
|||
}); |
|||
} |
|||
|
|||
/** |
|||
* 将字符串转换为Object |
|||
* |
|||
* @param json 数据字符串 |
|||
* @return list<map> |
|||
*/ |
|||
public static JSONObject toJsonObject(String json) { |
|||
return JSON.parseObject(json); |
|||
} |
|||
|
|||
/** |
|||
* 将字符串转换为Array |
|||
* |
|||
* @param json 数据字符串 |
|||
* @return list<map> |
|||
*/ |
|||
public static JSONArray toJsonArray(String json) { |
|||
return JSON.parseArray(json); |
|||
} |
|||
|
|||
} |
|||
|
|||
@ -1,28 +0,0 @@ |
|||
package com.visual.open.anpr.core.base; |
|||
|
|||
import ai.onnxruntime.OrtEnvironment; |
|||
|
|||
import java.io.File; |
|||
import java.util.Map; |
|||
import java.util.TreeMap; |
|||
|
|||
public abstract class BaseTest { |
|||
|
|||
//静态加载动态链接库
|
|||
static{ nu.pattern.OpenCV.loadShared(); } |
|||
private OrtEnvironment env = OrtEnvironment.getEnvironment(); |
|||
|
|||
public static Map<String, String> getImagePathMap(String imagePath){ |
|||
Map<String, String> map = new TreeMap<>(); |
|||
File file = new File(imagePath); |
|||
if(file.isFile()){ |
|||
map.put(file.getName(), file.getAbsolutePath()); |
|||
}else if(file.isDirectory()){ |
|||
for(File tmpFile : file.listFiles()){ |
|||
map.putAll(getImagePathMap(tmpFile.getPath())); |
|||
} |
|||
} |
|||
return map; |
|||
} |
|||
|
|||
} |
|||
@ -1,77 +0,0 @@ |
|||
package com.visual.open.anpr.core.extract; |
|||
|
|||
import com.visual.open.anpr.core.base.BaseTest; |
|||
import com.visual.open.anpr.core.domain.*; |
|||
import com.visual.open.anpr.core.models.TorchPlateDetection; |
|||
import com.visual.open.anpr.core.models.TorchPlateRecognition; |
|||
import org.opencv.core.Mat; |
|||
import org.opencv.core.Point; |
|||
import org.opencv.core.Scalar; |
|||
import org.opencv.highgui.HighGui; |
|||
import org.opencv.imgcodecs.Imgcodecs; |
|||
import org.opencv.imgproc.Imgproc; |
|||
|
|||
import java.awt.*; |
|||
import java.util.HashMap; |
|||
import java.util.List; |
|||
import java.util.Map; |
|||
|
|||
public class PlateExtractorTest extends BaseTest { |
|||
|
|||
private static String plateDetectionPath = "open-anpr-core/src/main/resources/models/plate_detect.onnx"; |
|||
private static String plateRecognitionPath = "open-anpr-core/src/main/resources/models/plate_rec_color.onnx"; |
|||
|
|||
public static void main(String[] args) { |
|||
TorchPlateDetection torchPlateDetection = new TorchPlateDetection(plateDetectionPath, 1); |
|||
TorchPlateRecognition torchPlateRecognition = new TorchPlateRecognition(plateRecognitionPath, 1); |
|||
PlateExtractor extractor = new PlateExtractorImpl(torchPlateDetection, torchPlateRecognition); |
|||
|
|||
String imagePath = "open-anpr-core/src/test/resources/images"; |
|||
Map<String, String> map = getImagePathMap(imagePath); |
|||
for(String fileName : map.keySet()) { |
|||
String imageFilePath = map.get(fileName); |
|||
System.out.println(imageFilePath); |
|||
Mat image = Imgcodecs.imread(imageFilePath); |
|||
long startTime = System.currentTimeMillis(); |
|||
ExtParam extParam = ExtParam.build() |
|||
.setTopK(20) |
|||
.setScoreTh(0.3f) |
|||
.setIouTh(0.5f); |
|||
//推理
|
|||
PlateImage plateImage = extractor.extract(ImageMat.fromCVMat(image), extParam, new HashMap<>()); |
|||
System.out.println("cost:" + (System.currentTimeMillis()-startTime)); |
|||
List<PlateInfo> plateInfos = plateImage.PlateInfos(); |
|||
//可视化
|
|||
DrawImage drawImage = DrawImage.build(imageFilePath); |
|||
for(PlateInfo plateInfo: plateInfos){ |
|||
//画框
|
|||
PlateInfo.Point [] points = plateInfo.box.toArray(); |
|||
for(int i =0; i< points.length; i++){ |
|||
if(i+1 == points.length){ |
|||
drawImage.drawLine( |
|||
new DrawImage.Point((int)points[i].x, (int)points[i].y), |
|||
new DrawImage.Point((int)points[0].x, (int)points[0].y), |
|||
2, Color.RED |
|||
); |
|||
}else{ |
|||
drawImage.drawLine( |
|||
new DrawImage.Point((int)points[i].x, (int)points[i].y), |
|||
new DrawImage.Point((int)points[i+1].x, (int)points[i+1].y), |
|||
2, Color.RED |
|||
); |
|||
} |
|||
} |
|||
//添加文本
|
|||
PlateInfo.ParseInfo parseInfo = plateInfo.parseInfo; |
|||
int fonSize = Float.valueOf(plateInfo.box.width() / parseInfo.plateNo.length() * 1.4f).intValue(); |
|||
drawImage.drawText(parseInfo.plateNo, |
|||
new DrawImage.Point((int)points[0].x, (int)points[0].y-(int)(fonSize*2.2)), fonSize, Color.RED); |
|||
drawImage.drawText((plateInfo.single ? "单排" : "双排") + ":" + parseInfo.plateColor, |
|||
new DrawImage.Point((int)points[0].x, (int)points[0].y-(int)(fonSize*1.2)), fonSize, Color.RED); |
|||
} |
|||
//show
|
|||
ImageMat.fromCVMat(drawImage.toMat()).imShow(); |
|||
image.release(); |
|||
} |
|||
} |
|||
} |
|||
@ -1,42 +0,0 @@ |
|||
package com.visual.open.anpr.core.models; |
|||
|
|||
import com.visual.open.anpr.core.domain.DrawImage; |
|||
import com.visual.open.anpr.core.domain.ImageMat; |
|||
import com.visual.open.anpr.core.domain.PlateInfo; |
|||
|
|||
import java.awt.*; |
|||
import java.util.HashMap; |
|||
import java.util.List; |
|||
|
|||
public class TorchPlateDetectionTest { |
|||
public static void main(String[] args) { |
|||
TorchPlateDetection torchPlateDetection = new TorchPlateDetection("open-anpr-core/src/main/resources/models/plate_detect.onnx", 1); |
|||
|
|||
String imagePath = "open-anpr-core/src/test/resources/images/image001.jpg"; |
|||
ImageMat imageMat = ImageMat.fromImage(imagePath); |
|||
List<PlateInfo> plateInfos = torchPlateDetection.inference(imageMat, 0.3f,0.5f, new HashMap<>()); |
|||
System.out.println(plateInfos); |
|||
|
|||
DrawImage drawImage = DrawImage.build(imagePath); |
|||
for(PlateInfo plateInfo : plateInfos){ |
|||
PlateInfo.Point [] points = plateInfo.box.toArray(); |
|||
for(int i =0; i< points.length; i++){ |
|||
if(i+1 == points.length){ |
|||
drawImage.drawLine( |
|||
new DrawImage.Point((int)points[i].x, (int)points[i].y), |
|||
new DrawImage.Point((int)points[0].x, (int)points[0].y), |
|||
2, Color.RED |
|||
); |
|||
}else{ |
|||
drawImage.drawLine( |
|||
new DrawImage.Point((int)points[i].x, (int)points[i].y), |
|||
new DrawImage.Point((int)points[i+1].x, (int)points[i+1].y), |
|||
2, Color.RED |
|||
); |
|||
} |
|||
} |
|||
} |
|||
ImageMat.fromCVMat(drawImage.toMat()).imShow(); |
|||
} |
|||
|
|||
} |
|||
@ -1,28 +0,0 @@ |
|||
package com.visual.open.anpr.core.models; |
|||
|
|||
import com.visual.open.anpr.core.domain.ImageMat; |
|||
import com.visual.open.anpr.core.domain.PlateInfo; |
|||
import com.visual.open.anpr.core.utils.CropUtil; |
|||
import org.opencv.core.Mat; |
|||
|
|||
import java.util.HashMap; |
|||
import java.util.List; |
|||
|
|||
public class TorchPlateRecognitionTest { |
|||
|
|||
public static void main(String[] args) { |
|||
TorchPlateDetection torchPlateDetection = new TorchPlateDetection("open-anpr-core/src/main/resources/models/plate_detect.onnx", 1); |
|||
TorchPlateRecognition torchPlateRecognition = new TorchPlateRecognition("open-anpr-core/src/main/resources/models/plate_rec_color.onnx", 1); |
|||
|
|||
String imagePath = "open-anpr-core/src/test/resources/images"; |
|||
ImageMat imageMat = ImageMat.fromImage(imagePath); |
|||
List<PlateInfo> plateInfos = torchPlateDetection.inference(imageMat, 0.3f,0.5f, new HashMap<>()); |
|||
System.out.println(plateInfos); |
|||
for(PlateInfo plateInfo : plateInfos){ |
|||
Mat crop = CropUtil.crop(imageMat.toCvMat(), plateInfo.box); |
|||
// ImageMat.fromCVMat(crop).imShow();
|
|||
PlateInfo.ParseInfo parseInfo = torchPlateRecognition.inference(ImageMat.fromCVMat(crop), plateInfo.single, new HashMap<>()); |
|||
System.out.println(parseInfo); |
|||
} |
|||
} |
|||
} |
|||
|
Before Width: | Height: | Size: 107 KiB |
|
Before Width: | Height: | Size: 64 KiB |
|
Before Width: | Height: | Size: 40 KiB |
|
Before Width: | Height: | Size: 390 KiB |
|
Before Width: | Height: | Size: 47 KiB |
|
Before Width: | Height: | Size: 64 KiB |
|
Before Width: | Height: | Size: 671 KiB |
|
Before Width: | Height: | Size: 328 KiB |
|
Before Width: | Height: | Size: 382 KiB |
|
Before Width: | Height: | Size: 85 KiB |
|
Before Width: | Height: | Size: 29 KiB |
|
Before Width: | Height: | Size: 999 KiB |
|
Before Width: | Height: | Size: 34 KiB |
|
Before Width: | Height: | Size: 400 KiB |
|
Before Width: | Height: | Size: 116 KiB |
@ -1,10 +1,10 @@ |
|||
package com.visual.open.anpr.core.base; |
|||
package com.visual.open.anpr.server.base; |
|||
|
|||
import ai.onnxruntime.OrtEnvironment; |
|||
import ai.onnxruntime.OrtLoggingLevel; |
|||
import ai.onnxruntime.OrtSession; |
|||
|
|||
public abstract class BaseOnnxInfer extends OpenCVLoader{ |
|||
public abstract class BaseOnnxInfer extends OpenCVLoader { |
|||
|
|||
private OrtEnvironment env; |
|||
private String[] inputNames; |
|||
@ -1,4 +1,4 @@ |
|||
package com.visual.open.anpr.core.base; |
|||
package com.visual.open.anpr.server.base; |
|||
|
|||
public abstract class OpenCVLoader { |
|||
|
|||
@ -1,7 +1,7 @@ |
|||
package com.visual.open.anpr.core.base; |
|||
package com.visual.open.anpr.server.base; |
|||
|
|||
import com.visual.open.anpr.core.domain.ImageMat; |
|||
import com.visual.open.anpr.core.domain.PlateInfo; |
|||
import com.visual.open.anpr.server.domain.ImageMat; |
|||
import com.visual.open.anpr.server.domain.PlateInfo; |
|||
|
|||
import java.util.List; |
|||
import java.util.Map; |
|||
@ -0,0 +1,11 @@ |
|||
package com.visual.open.anpr.server.base; |
|||
|
|||
import com.visual.open.anpr.server.domain.ImageMat; |
|||
import com.visual.open.anpr.server.domain.PlateInfo.ParseInfo; |
|||
|
|||
import java.util.Map; |
|||
|
|||
public interface PlateRecognition { |
|||
|
|||
ParseInfo inference(ImageMat image, Boolean single, Map<String, Object> params); |
|||
} |
|||
@ -1,4 +1,4 @@ |
|||
package com.visual.open.anpr.core.domain; |
|||
package com.visual.open.anpr.server.domain; |
|||
|
|||
import org.opencv.core.Mat; |
|||
|
|||
@ -1,4 +1,4 @@ |
|||
package com.visual.open.anpr.utils; |
|||
package com.visual.open.anpr.server.domain; |
|||
|
|||
import org.opencv.core.CvType; |
|||
import org.opencv.core.Mat; |
|||
@ -1,4 +1,4 @@ |
|||
package com.visual.open.anpr.core.domain; |
|||
package com.visual.open.anpr.server.domain; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
@ -1,8 +1,8 @@ |
|||
package com.visual.open.anpr.core.domain; |
|||
package com.visual.open.anpr.server.domain; |
|||
|
|||
import ai.onnxruntime.OnnxTensor; |
|||
import ai.onnxruntime.OrtEnvironment; |
|||
import com.visual.open.anpr.core.utils.MatUtil; |
|||
import com.visual.open.anpr.server.utils.MatUtil; |
|||
import org.opencv.core.Point; |
|||
import org.opencv.core.*; |
|||
import org.opencv.dnn.Dnn; |
|||
@ -1,4 +1,4 @@ |
|||
package com.visual.open.anpr.core.domain; |
|||
package com.visual.open.anpr.server.domain; |
|||
|
|||
import java.io.Serializable; |
|||
import java.util.ArrayList; |
|||
@ -1,4 +1,4 @@ |
|||
package com.visual.open.anpr.core.domain; |
|||
package com.visual.open.anpr.server.domain; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
@ -1,9 +1,10 @@ |
|||
package com.visual.open.anpr.core.extract; |
|||
package com.visual.open.anpr.server.extract; |
|||
|
|||
import com.visual.open.anpr.server.domain.ExtParam; |
|||
import com.visual.open.anpr.server.domain.ImageMat; |
|||
import com.visual.open.anpr.server.domain.PlateImage; |
|||
|
|||
import java.util.Map; |
|||
import com.visual.open.anpr.core.domain.ExtParam; |
|||
import com.visual.open.anpr.core.domain.ImageMat; |
|||
import com.visual.open.anpr.core.domain.PlateImage; |
|||
|
|||
public interface PlateExtractor { |
|||
|
|||
@ -1,12 +1,12 @@ |
|||
package com.visual.open.anpr.core.extract; |
|||
package com.visual.open.anpr.server.extract; |
|||
|
|||
import com.visual.open.anpr.core.base.PlateDetection; |
|||
import com.visual.open.anpr.core.base.PlateRecognition; |
|||
import com.visual.open.anpr.core.domain.ExtParam; |
|||
import com.visual.open.anpr.core.domain.ImageMat; |
|||
import com.visual.open.anpr.core.domain.PlateImage; |
|||
import com.visual.open.anpr.core.domain.PlateInfo; |
|||
import com.visual.open.anpr.core.utils.CropUtil; |
|||
import com.visual.open.anpr.server.base.PlateDetection; |
|||
import com.visual.open.anpr.server.base.PlateRecognition; |
|||
import com.visual.open.anpr.server.domain.ExtParam; |
|||
import com.visual.open.anpr.server.domain.ImageMat; |
|||
import com.visual.open.anpr.server.domain.PlateImage; |
|||
import com.visual.open.anpr.server.domain.PlateInfo; |
|||
import com.visual.open.anpr.server.utils.CropUtil; |
|||
import org.opencv.core.Core; |
|||
import org.opencv.core.Mat; |
|||
|
|||
@ -1,14 +1,17 @@ |
|||
package com.visual.open.anpr.core.models; |
|||
package com.visual.open.anpr.server.models; |
|||
|
|||
import ai.onnxruntime.OnnxTensor; |
|||
import ai.onnxruntime.OrtSession; |
|||
import com.visual.open.anpr.core.base.BaseOnnxInfer; |
|||
import com.visual.open.anpr.core.base.PlateDetection; |
|||
import com.visual.open.anpr.core.domain.ImageMat; |
|||
import com.visual.open.anpr.core.domain.BorderMat; |
|||
import com.visual.open.anpr.core.domain.PlateInfo; |
|||
import com.visual.open.anpr.core.utils.ReleaseUtil; |
|||
import org.opencv.core.*; |
|||
import com.visual.open.anpr.server.base.BaseOnnxInfer; |
|||
import com.visual.open.anpr.server.base.PlateDetection; |
|||
import com.visual.open.anpr.server.domain.BorderMat; |
|||
import com.visual.open.anpr.server.domain.ImageMat; |
|||
import com.visual.open.anpr.server.domain.PlateInfo; |
|||
import com.visual.open.anpr.server.utils.ReleaseUtil; |
|||
import org.opencv.core.Core; |
|||
import org.opencv.core.Mat; |
|||
import org.opencv.core.Scalar; |
|||
import org.opencv.core.Size; |
|||
import org.opencv.imgproc.Imgproc; |
|||
|
|||
import java.util.*; |
|||
@ -1,20 +1,21 @@ |
|||
package com.visual.open.anpr.core.models; |
|||
package com.visual.open.anpr.server.models; |
|||
|
|||
import ai.onnxruntime.OnnxTensor; |
|||
import ai.onnxruntime.OrtSession; |
|||
import com.visual.open.anpr.core.base.BaseOnnxInfer; |
|||
import com.visual.open.anpr.core.base.PlateRecognition; |
|||
import com.visual.open.anpr.core.domain.ImageMat; |
|||
import com.visual.open.anpr.core.domain.PlateInfo.ParseInfo; |
|||
import com.visual.open.anpr.core.utils.ArrayUtil; |
|||
import com.visual.open.anpr.core.utils.MatUtil; |
|||
import com.visual.open.anpr.core.utils.ReleaseUtil; |
|||
import com.visual.open.anpr.core.utils.SoftMaxUtil; |
|||
import com.visual.open.anpr.server.base.BaseOnnxInfer; |
|||
import com.visual.open.anpr.server.base.PlateRecognition; |
|||
import com.visual.open.anpr.server.domain.ImageMat; |
|||
import com.visual.open.anpr.server.domain.PlateInfo.ParseInfo; |
|||
import com.visual.open.anpr.server.utils.ArrayUtil; |
|||
import com.visual.open.anpr.server.utils.MatUtil; |
|||
import com.visual.open.anpr.server.utils.ReleaseUtil; |
|||
import com.visual.open.anpr.server.utils.SoftMaxUtil; |
|||
import org.opencv.core.Mat; |
|||
import org.opencv.core.Rect; |
|||
import org.opencv.core.Scalar; |
|||
import org.opencv.core.Size; |
|||
import org.opencv.imgproc.Imgproc; |
|||
|
|||
import java.util.Collections; |
|||
import java.util.Map; |
|||
|
|||
@ -1,4 +1,4 @@ |
|||
package com.visual.open.anpr.core.utils; |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
import org.apache.commons.math3.linear.RealMatrix; |
|||
import org.apache.commons.math3.linear.RealVector; |
|||
@ -1,4 +1,4 @@ |
|||
package com.visual.open.anpr.core.utils; |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
public class ArrayUtil { |
|||
|
|||
@ -1,6 +1,6 @@ |
|||
package com.visual.open.anpr.core.utils; |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
import com.visual.open.anpr.core.domain.PlateInfo; |
|||
import com.visual.open.anpr.server.domain.PlateInfo; |
|||
import org.opencv.core.CvType; |
|||
import org.opencv.core.Mat; |
|||
import org.opencv.core.Point; |
|||
@ -0,0 +1,324 @@ |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
import ws.schild.jave.Encoder; |
|||
import ws.schild.jave.EncoderException; |
|||
import ws.schild.jave.MultimediaObject; |
|||
import ws.schild.jave.encode.AudioAttributes; |
|||
import ws.schild.jave.encode.EncodingAttributes; |
|||
import ws.schild.jave.encode.VideoAttributes; |
|||
import ws.schild.jave.info.MultimediaInfo; |
|||
import ws.schild.jave.process.ProcessWrapper; |
|||
import ws.schild.jave.process.ffmpeg.DefaultFFMPEGLocator; |
|||
|
|||
import java.io.BufferedReader; |
|||
import java.io.File; |
|||
import java.io.IOException; |
|||
import java.io.InputStreamReader; |
|||
import java.net.URL; |
|||
import java.text.SimpleDateFormat; |
|||
import java.util.Date; |
|||
|
|||
/** |
|||
* @author ruibo.duan, <1573434995@qq.com> |
|||
* @since 2021/7/22 |
|||
*/ |
|||
public class FfmpegUtil { |
|||
|
|||
/** |
|||
* 通过本地路径获取多媒体文件信息(宽，高，时长，编码等) |
|||
* |
|||
* @param localPath 本地路径 |
|||
* @return MultimediaInfo 对象,包含 (宽，高，时长，编码等) |
|||
* @throws EncoderException |
|||
*/ |
|||
public static MultimediaInfo getMultimediaInfo(String localPath) { |
|||
MultimediaInfo multimediaInfo = null; |
|||
try { |
|||
multimediaInfo = new MultimediaObject(new File(localPath)).getInfo(); |
|||
} catch (EncoderException e) { |
|||
System.out.println("获取多媒体文件信息异常"); |
|||
e.printStackTrace(); |
|||
} |
|||
return multimediaInfo; |
|||
} |
|||
|
|||
/** |
|||
// * 通过URL获取多媒体文件信息
|
|||
* |
|||
* @param url 网络url |
|||
* @return MultimediaInfo 对象,包含 (宽，高，时长，编码等) |
|||
* @throws EncoderException |
|||
*/ |
|||
public static MultimediaInfo getMultimediaInfoFromUrl(String url) { |
|||
MultimediaInfo multimediaInfo = null; |
|||
try { |
|||
multimediaInfo = new MultimediaObject(new URL(url)).getInfo(); |
|||
} catch (Exception e) { |
|||
System.out.println("获取多媒体文件信息异常"); |
|||
e.printStackTrace(); |
|||
} |
|||
return multimediaInfo; |
|||
} |
|||
|
|||
private static final int SAMPLING_RATE = 16000; |
|||
private static final int SINGLE_CHANNEL = 1; |
|||
|
|||
/** |
|||
* 音频格式化为wav,并设置单声道和采样率 |
|||
* |
|||
* @param url 需要转格式的音频 |
|||
* @param targetPath 格式化后要保存的目标路径 |
|||
*/ |
|||
public static boolean formatAudio(String url, String targetPath) { |
|||
File target = new File(targetPath); |
|||
MultimediaObject multimediaObject; |
|||
try { |
|||
// 若是本地文件： multimediaObject = new MultimediaObject(new File("你的本地路径"));
|
|||
multimediaObject = new MultimediaObject(new URL(url)); |
|||
// 音频参数
|
|||
// TODO: 2023/1/31 此处按需自定义音频参数
|
|||
AudioAttributes audio = new AudioAttributes(); |
|||
// 采样率
|
|||
audio.setSamplingRate(SAMPLING_RATE); |
|||
// 单声道
|
|||
audio.setChannels(SINGLE_CHANNEL); |
|||
Encoder encoder = new Encoder(); |
|||
EncodingAttributes attrs = new EncodingAttributes(); |
|||
// 输出格式
|
|||
attrs.setOutputFormat("wav"); |
|||
attrs.setAudioAttributes(audio); |
|||
encoder.encode(multimediaObject, target, attrs); |
|||
return true; |
|||
} catch (Exception e) { |
|||
System.out.println("格式化音频异常"); |
|||
return false; |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 视频格式化为mp4 |
|||
* |
|||
* @param url |
|||
* @param targetPath |
|||
* @return |
|||
*/ |
|||
public static boolean formatToMp4(String url, String targetPath) { |
|||
File target = new File(targetPath); |
|||
MultimediaObject multimediaObject; |
|||
try { |
|||
// 若是本地文件： multimediaObject = new MultimediaObject(new File("你的本地路径"));
|
|||
multimediaObject = new MultimediaObject(new URL(url)); |
|||
EncodingAttributes attributes = new EncodingAttributes(); |
|||
// 设置视频的音频参数
|
|||
AudioAttributes audioAttributes = new AudioAttributes(); |
|||
attributes.setAudioAttributes(audioAttributes); |
|||
// 设置视频的视频参数
|
|||
VideoAttributes videoAttributes = new VideoAttributes(); |
|||
// 设置帧率
|
|||
videoAttributes.setFrameRate(25); |
|||
attributes.setVideoAttributes(videoAttributes); |
|||
// 设置输出格式
|
|||
attributes.setOutputFormat("mp4"); |
|||
Encoder encoder = new Encoder(); |
|||
encoder.encode(multimediaObject, target, attributes); |
|||
return true; |
|||
} catch (Exception e) { |
|||
System.out.println("格式化视频异常"); |
|||
e.printStackTrace(); |
|||
return false; |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 获取视频缩略图 获取视频第0秒的第一帧图片 |
|||
* |
|||
* <p>执行的ffmpeg 命令为： ffmpeg -i 你的视频文件路径 -ss 指定的秒数 生成文件的全路径地址 |
|||
* |
|||
* @param localPath 本地路径 |
|||
* @param targetPath 存放的目标路径 |
|||
* @return |
|||
*/ |
|||
public static boolean getTargetThumbnail(String localPath, String targetPath) { |
|||
// FIXME: 2023/1/31 该方法基本可作为执行ffmpeg命令的模板方法，之后的几个方法与此类似
|
|||
try { |
|||
ProcessWrapper ffmpeg = new DefaultFFMPEGLocator().createExecutor(); |
|||
ffmpeg.addArgument("-i"); |
|||
ffmpeg.addArgument(localPath); |
|||
ffmpeg.addArgument("-ss"); |
|||
// 此处可自定义视频的秒数
|
|||
ffmpeg.addArgument("0"); |
|||
ffmpeg.addArgument(targetPath); |
|||
ffmpeg.execute(); |
|||
try (BufferedReader br = new BufferedReader(new InputStreamReader(ffmpeg.getErrorStream()))) { |
|||
blockFfmpeg(br); |
|||
} |
|||
} catch (IOException e) { |
|||
System.out.println("获取视频缩略图失败"); |
|||
e.printStackTrace(); |
|||
return false; |
|||
} |
|||
return true; |
|||
} |
|||
|
|||
/** |
|||
* 抓拍实况 |
|||
* @param localPath 实况地址 |
|||
* @param targetPath 存放的目标路径 |
|||
* @return 返回抓拍图片的文件名 |
|||
*/ |
|||
// 获取实况抓拍图片
|
|||
public static String getLiveCapture(String localPath, String targetPath) { |
|||
|
|||
String formattedDate =""; |
|||
|
|||
try { |
|||
ProcessWrapper ffmpeg = new DefaultFFMPEGLocator().createExecutor(); |
|||
ffmpeg.addArgument("-i"); |
|||
ffmpeg.addArgument(localPath); |
|||
ffmpeg.addArgument("-vf"); |
|||
// 加上 "select=eq(pict_type\,I)" -frames:v 1
|
|||
ffmpeg.addArgument("select='eq(pict_type,I)'"); |
|||
ffmpeg.addArgument("-frames:v"); |
|||
ffmpeg.addArgument("1"); |
|||
|
|||
// 生成一个根据时间的文件名
|
|||
SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddHHmmss"); |
|||
// 获取当前时间
|
|||
Date currentDate = new Date(); |
|||
// 格式化时间为字符串
|
|||
formattedDate = sdf.format(currentDate)+ ".jpg"; |
|||
|
|||
ffmpeg.addArgument(formattedDate); |
|||
|
|||
ffmpeg.addArgument(targetPath + formattedDate); |
|||
|
|||
ffmpeg.execute(); |
|||
try (BufferedReader br = new BufferedReader(new InputStreamReader(ffmpeg.getErrorStream()))) { |
|||
blockFfmpeg(br); |
|||
} |
|||
} catch (IOException e) { |
|||
formattedDate = "获取实况抓拍图片失败"; |
|||
System.out.println(formattedDate); |
|||
e.printStackTrace(); |
|||
return formattedDate; |
|||
} return formattedDate; |
|||
} |
|||
|
|||
/** |
|||
* 等待命令执行成功，退出 |
|||
* |
|||
* @param br |
|||
* @throws IOException |
|||
*/ |
|||
private static void blockFfmpeg(BufferedReader br) throws IOException { |
|||
String line; |
|||
// 该方法阻塞线程，直至合成成功
|
|||
while ((line = br.readLine()) != null) { |
|||
doNothing(line); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 打印日志 |
|||
* |
|||
* @param line |
|||
*/ |
|||
private static void doNothing(String line) { |
|||
// FIXME: 2023/1/31 正式使用时注释掉此行，仅用于观察日志
|
|||
System.out.println(line); |
|||
} |
|||
|
|||
/** |
|||
* 视频增加字幕 |
|||
* |
|||
* @param originVideoPath 原视频地址 |
|||
* @param targetVideoPath 目标视频地址 |
|||
* @param srtPath 固定格式的srt文件地址或存储位置，字母文件名： xxx.srt，样例看博客 |
|||
* @return |
|||
* @throws Exception |
|||
*/ |
|||
public static boolean addSubtitle( |
|||
String originVideoPath, String srtPath, String targetVideoPath) { |
|||
try { |
|||
ProcessWrapper ffmpeg = new DefaultFFMPEGLocator().createExecutor(); |
|||
ffmpeg.addArgument("-i"); |
|||
ffmpeg.addArgument(originVideoPath); |
|||
ffmpeg.addArgument("-i"); |
|||
ffmpeg.addArgument(srtPath); |
|||
ffmpeg.addArgument("-c"); |
|||
ffmpeg.addArgument("copy"); |
|||
ffmpeg.addArgument(targetVideoPath); |
|||
ffmpeg.execute(); |
|||
try (BufferedReader br = new BufferedReader(new InputStreamReader(ffmpeg.getErrorStream()))) { |
|||
blockFfmpeg(br); |
|||
} |
|||
} catch (IOException e) { |
|||
System.out.println("字幕增加失败"); |
|||
e.printStackTrace(); |
|||
} |
|||
return true; |
|||
} |
|||
|
|||
/** |
|||
* 常用命令 |
|||
* |
|||
* @return |
|||
*/ |
|||
public static void cmd() { |
|||
// FIXME: 2023/1/31 还有很多类似命令 不再一一列举 ，附上命令,具体写法参考 getTargetThumbnail或addSubtitle方法
|
|||
// FIXME: 2023/1/31 ffmpeg命令网上搜索即可
|
|||
|
|||
// 剪切视频
|
|||
// ffmpeg -ss 00:00:00 -t 00:00:30 -i test.mp4 -vcodec copy -acodec copy output.mp4
|
|||
// * -ss 指定从什么时间开始
|
|||
// * -t 指定需要截取多长时间
|
|||
// * -i 指定输入文件
|
|||
|
|||
// ffmpeg -ss 10 -t 15 -accurate_seek -i test.mp4 -codec copy cut.mp4
|
|||
// ffmpeg -ss 10 -t 15 -accurate_seek -i test.mp4 -codec copy -avoid_negative_ts 1 cut.mp4
|
|||
|
|||
// 拼接MP4
|
|||
// 第一种方法：
|
|||
// ffmpeg -i "concat:1.mp4|2.mp4|3.mp4" -codec copy out_mp4.mp4
|
|||
// 1.mp4 第一个视频文件的全路径
|
|||
// 2.mp4 第二个视频文件的全路径
|
|||
|
|||
// 提取视频中的音频
|
|||
// ffmpeg -i input.mp4 -acodec copy -vn output.mp3
|
|||
// -vn: 去掉视频；-acodec: 音频选项， 一般后面加copy表示拷贝
|
|||
|
|||
// 音视频合成
|
|||
// ffmpeg -y –i input.mp4 –i input.mp3 –vcodec copy –acodec copy output.mp4
|
|||
// -y 覆盖输出文件
|
|||
|
|||
// 剪切视频
|
|||
// ffmpeg -ss 0:1:30 -t 0:0:20 -i input.mp4 -vcodec copy -acodec copy output.mp4
|
|||
// -ss 开始时间; -t 持续时间
|
|||
|
|||
// 视频截图
|
|||
// ffmpeg –i test.mp4 –f image2 -t 0.001 -s 320x240 image-%3d.jpg
|
|||
// -s 设置分辨率; -f 强迫采用格式fmt;
|
|||
|
|||
// 视频分解为图片
|
|||
// ffmpeg –i test.mp4 –r 1 –f image2 image-%3d.jpg
|
|||
// -r 指定截屏频率
|
|||
|
|||
// 将图片合成视频
|
|||
// ffmpeg -f image2 -i image%d.jpg output.mp4
|
|||
|
|||
// 视频拼接
|
|||
// ffmpeg -f concat -i filelist.txt -c copy output.mp4
|
|||
|
|||
// 将视频转为gif
|
|||
// ffmpeg -i input.mp4 -ss 0:0:30 -t 10 -s 320x240 -pix_fmt rgb24 output.gif
|
|||
// -pix_fmt 指定编码
|
|||
|
|||
// 视频添加水印
|
|||
// ffmpeg -i input.mp4 -i logo.jpg
|
|||
// -filter_complex[0:v][1:v]overlay=main_w-overlay_w-10:main_h-overlay_h-10[out] -map [out] -map
|
|||
// 0:a -codec:a copy output.mp4
|
|||
// main_w-overlay_w-10 视频的宽度-水印的宽度-水印边距；
|
|||
|
|||
} |
|||
} |
|||
|
|||
@ -0,0 +1,173 @@ |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.core.io.ClassPathResource; |
|||
import org.springframework.core.io.Resource; |
|||
import org.springframework.core.io.support.PathMatchingResourcePatternResolver; |
|||
import org.springframework.core.io.support.ResourcePatternResolver; |
|||
|
|||
import java.io.*; |
|||
|
|||
|
|||
@Slf4j |
|||
public class FileCopyUtil { |
|||
|
|||
|
|||
/** |
|||
* 复制path目录下所有文件 |
|||
* @param path 文件目录 不能以/开头 |
|||
* @param newpath 新文件目录 |
|||
*/ |
|||
public static void BatCopyFileFromJar(String path,String newpath) { |
|||
if (!new File(newpath).exists()){ |
|||
new File(newpath).mkdir(); |
|||
} |
|||
ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver(); |
|||
try { |
|||
//获取所有匹配的文件
|
|||
Resource[] resources = resolver.getResources(path+"/*"); |
|||
//打印有多少文件
|
|||
for(int i=0;i<resources.length;i++) { |
|||
Resource resource=resources[i]; |
|||
try { |
|||
//以jar运行时，resource.getFile().isFile() 无法获取文件类型，会报异常，抓取异常后直接生成新的文件即可；以非jar运行时，需要判断文件类型，避免如果是目录会复制错误，将目录写成文件。
|
|||
if(resource.getFile().isFile()) { |
|||
makeFile(newpath+"/"+resource.getFilename()); |
|||
InputStream stream = resource.getInputStream(); |
|||
write2File(stream, newpath+"/"+resource.getFilename()); |
|||
} |
|||
}catch (Exception e) { |
|||
makeFile(newpath+"/"+resource.getFilename()); |
|||
InputStream stream = resource.getInputStream(); |
|||
write2File(stream, newpath+"/"+resource.getFilename()); |
|||
} |
|||
} |
|||
} catch (Exception e) { |
|||
e.printStackTrace(); |
|||
} |
|||
} |
|||
/** |
|||
* 创建文件 |
|||
* @param path 全路径 指向文件 |
|||
* @return |
|||
*/ |
|||
public static boolean makeFile(String path) { |
|||
File file = new File(path); |
|||
if(file.exists()) { |
|||
return false; |
|||
} |
|||
if (path.endsWith(File.separator)) { |
|||
return false; |
|||
} |
|||
if(!file.getParentFile().exists()) { |
|||
if(!file.getParentFile().mkdirs()) { |
|||
return false; |
|||
} |
|||
} |
|||
try { |
|||
if (file.createNewFile()) { |
|||
return true; |
|||
} else { |
|||
return false; |
|||
} |
|||
} catch (IOException e) { |
|||
e.printStackTrace(); |
|||
return false; |
|||
} |
|||
} |
|||
/** |
|||
* 输入流写入文件 |
|||
* |
|||
* @param is |
|||
* 输入流 |
|||
* @param filePath |
|||
* 文件保存目录路径 |
|||
* @throws IOException |
|||
*/ |
|||
public static void write2File(InputStream is, String filePath) throws IOException { |
|||
OutputStream os = new FileOutputStream(filePath); |
|||
int len = 8192; |
|||
byte[] buffer = new byte[len]; |
|||
while ((len = is.read(buffer, 0, len)) != -1) { |
|||
os.write(buffer, 0, len); |
|||
} |
|||
os.close(); |
|||
is.close(); |
|||
} |
|||
/** |
|||
*处理异常报错（springboot读取classpath里的文件，解决打jar包java.io.FileNotFoundException: class path resource cannot be opened） |
|||
**/ |
|||
public static File getFileFromClassPath(String path){ |
|||
File targetFile = new File(path); |
|||
if(!targetFile.exists()){ |
|||
if(targetFile.getParent()!=null){ |
|||
File parent=new File(targetFile.getParent()); |
|||
if(!parent.exists()){ |
|||
parent.mkdirs(); |
|||
} |
|||
} |
|||
InputStream initialStream=null; |
|||
OutputStream outStream =null; |
|||
try { |
|||
Resource resource=new ClassPathResource(path); |
|||
//注意通过getInputStream，不能用getFile
|
|||
initialStream=resource.getInputStream(); |
|||
byte[] buffer = new byte[initialStream.available()]; |
|||
initialStream.read(buffer); |
|||
outStream = new FileOutputStream(targetFile); |
|||
outStream.write(buffer); |
|||
} catch (IOException e) { |
|||
} finally { |
|||
if (initialStream != null) { |
|||
try { |
|||
initialStream.close(); // 关闭流
|
|||
} catch (IOException e) { |
|||
} |
|||
} |
|||
if (outStream != null) { |
|||
try { |
|||
outStream.close(); // 关闭流
|
|||
} catch (IOException e) { |
|||
} |
|||
} |
|||
} |
|||
} |
|||
return targetFile; |
|||
} |
|||
private static String userDir = System.getProperty("user.dir"); |
|||
public static void copyHCNetSDKFromLinux(){ |
|||
log.info("开始复制文件"); |
|||
String linuxSOFolder1 = userDir + "/lib/hk/linux/"; |
|||
String linuxSOFolder2 = userDir + "/lib/hk/linux/HKNetSDKCom/"; |
|||
|
|||
File fileFromClassPath1 = FileCopyUtil.getFileFromClassPath("/lib/hk/linux64/"); |
|||
File fileFromClassPath2 = FileCopyUtil.getFileFromClassPath("/lib/hk/linux64/HCNetSDKCom/"); |
|||
|
|||
BatCopyFileFromJar(fileFromClassPath1.toString(),linuxSOFolder1); |
|||
BatCopyFileFromJar(fileFromClassPath2.toString(),linuxSOFolder2); |
|||
log.info("文件复制结束"); |
|||
} |
|||
|
|||
public static void copyHCNetSdkFromWindows(){ |
|||
|
|||
log.info("开始复制文件"); |
|||
String WindowsDllFolder1 = userDir + "/models/"; |
|||
|
|||
File fileFromClassPath1 = FileCopyUtil.getFileFromClassPath("/models/"); |
|||
|
|||
BatCopyFileFromJar(fileFromClassPath1.toString(), WindowsDllFolder1); |
|||
log.info("文件复制结束"); |
|||
|
|||
} |
|||
|
|||
public static <T> boolean isStartupFromJar() { |
|||
String protocol = FileCopyUtil.class.getResource("").getProtocol(); |
|||
if ("jar".equals(protocol)){ |
|||
return true; |
|||
}else { |
|||
return false; |
|||
} |
|||
} |
|||
|
|||
} |
|||
|
|||
@ -0,0 +1,140 @@ |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
import org.springframework.core.io.ClassPathResource; |
|||
import org.springframework.core.io.Resource; |
|||
import org.springframework.core.io.support.PathMatchingResourcePatternResolver; |
|||
import org.springframework.core.io.support.ResourcePatternResolver; |
|||
|
|||
import java.io.*; |
|||
|
|||
/** |
|||
* @author:hahaha |
|||
* @creattime:2021-12-02 10:33 |
|||
*/ |
|||
|
|||
public class FreeMarkerUtil { |
|||
|
|||
|
|||
/** |
|||
* 复制path目录下所有文件 |
|||
* @param path 文件目录 不能以/开头 |
|||
* @param newpath 新文件目录 |
|||
*/ |
|||
public static void BatCopyFileFromJar(String path,String newpath) { |
|||
if (!new File(newpath).exists()){ |
|||
new File(newpath).mkdir(); |
|||
} |
|||
ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver(); |
|||
try { |
|||
//获取所有匹配的文件
|
|||
Resource[] resources = resolver.getResources(path+"/*"); |
|||
//打印有多少文件
|
|||
for(int i=0;i<resources.length;i++) { |
|||
Resource resource=resources[i]; |
|||
try { |
|||
//以jar运行时，resource.getFile().isFile() 无法获取文件类型，会报异常，抓取异常后直接生成新的文件即可；以非jar运行时，需要判断文件类型，避免如果是目录会复制错误，将目录写成文件。
|
|||
if(resource.getFile().isFile()) { |
|||
makeFile(newpath+"/"+resource.getFilename()); |
|||
InputStream stream = resource.getInputStream(); |
|||
write2File(stream, newpath+"/"+resource.getFilename()); |
|||
} |
|||
}catch (Exception e) { |
|||
makeFile(newpath+"/"+resource.getFilename()); |
|||
InputStream stream = resource.getInputStream(); |
|||
write2File(stream, newpath+"/"+resource.getFilename()); |
|||
} |
|||
} |
|||
} catch (Exception e) { |
|||
e.printStackTrace(); |
|||
} |
|||
} |
|||
/** |
|||
* 创建文件 |
|||
* @param path 全路径 指向文件 |
|||
* @return |
|||
*/ |
|||
public static boolean makeFile(String path) { |
|||
File file = new File(path); |
|||
if(file.exists()) { |
|||
return false; |
|||
} |
|||
if (path.endsWith(File.separator)) { |
|||
return false; |
|||
} |
|||
if(!file.getParentFile().exists()) { |
|||
if(!file.getParentFile().mkdirs()) { |
|||
return false; |
|||
} |
|||
} |
|||
try { |
|||
if (file.createNewFile()) { |
|||
return true; |
|||
} else { |
|||
return false; |
|||
} |
|||
} catch (IOException e) { |
|||
e.printStackTrace(); |
|||
return false; |
|||
} |
|||
} |
|||
/** |
|||
* 输入流写入文件 |
|||
* |
|||
* @param is |
|||
* 输入流 |
|||
* @param filePath |
|||
* 文件保存目录路径 |
|||
* @throws IOException |
|||
*/ |
|||
public static void write2File(InputStream is, String filePath) throws IOException { |
|||
OutputStream os = new FileOutputStream(filePath); |
|||
int len = 8192; |
|||
byte[] buffer = new byte[len]; |
|||
while ((len = is.read(buffer, 0, len)) != -1) { |
|||
os.write(buffer, 0, len); |
|||
} |
|||
os.close(); |
|||
is.close(); |
|||
} |
|||
/** |
|||
*处理异常报错（springboot读取classpath里的文件，解决打jar包java.io.FileNotFoundException: class path resource cannot be opened） |
|||
**/ |
|||
public static File getFileFromClassPath(String path){ |
|||
File targetFile = new File(path); |
|||
if(!targetFile.exists()){ |
|||
if(targetFile.getParent()!=null){ |
|||
File parent=new File(targetFile.getParent()); |
|||
if(!parent.exists()){ |
|||
parent.mkdirs(); |
|||
} |
|||
} |
|||
InputStream initialStream=null; |
|||
OutputStream outStream =null; |
|||
try { |
|||
Resource resource=new ClassPathResource(path); |
|||
//注意通过getInputStream，不能用getFile
|
|||
initialStream=resource.getInputStream(); |
|||
byte[] buffer = new byte[initialStream.available()]; |
|||
initialStream.read(buffer); |
|||
outStream = new FileOutputStream(targetFile); |
|||
outStream.write(buffer); |
|||
} catch (IOException e) { |
|||
} finally { |
|||
if (initialStream != null) { |
|||
try { |
|||
initialStream.close(); // 关闭流
|
|||
} catch (IOException e) { |
|||
} |
|||
} |
|||
if (outStream != null) { |
|||
try { |
|||
outStream.close(); // 关闭流
|
|||
} catch (IOException e) { |
|||
} |
|||
} |
|||
} |
|||
} |
|||
return targetFile; |
|||
} |
|||
} |
|||
|
|||
@ -0,0 +1,82 @@ |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
import javax.imageio.ImageIO; |
|||
import java.awt.image.BufferedImage; |
|||
import java.io.ByteArrayInputStream; |
|||
import java.io.ByteArrayOutputStream; |
|||
import java.io.File; |
|||
import java.io.IOException; |
|||
import java.util.Base64; |
|||
|
|||
public class ImageUtil { |
|||
|
|||
/** |
|||
* 图片转Base64字符串 |
|||
* @param imageFileName |
|||
* @return |
|||
*/ |
|||
public static String convertImageToBase64Str(String imageFileName) { |
|||
ByteArrayOutputStream baos = null; |
|||
try { |
|||
//获取图片类型
|
|||
String suffix = imageFileName.substring(imageFileName.lastIndexOf(".") + 1); |
|||
//构建文件
|
|||
File imageFile = new File(imageFileName); |
|||
//通过ImageIO把文件读取成BufferedImage对象
|
|||
BufferedImage bufferedImage = ImageIO.read(imageFile); |
|||
//构建字节数组输出流
|
|||
baos = new ByteArrayOutputStream(); |
|||
//写入流
|
|||
ImageIO.write(bufferedImage, suffix, baos); |
|||
//通过字节数组流获取字节数组
|
|||
byte[] bytes = baos.toByteArray(); |
|||
//获取JDK8里的编码器Base64.Encoder转为base64字符
|
|||
return Base64.getEncoder().encodeToString(bytes); |
|||
} catch (Exception e) { |
|||
e.printStackTrace(); |
|||
} finally { |
|||
try { |
|||
if (baos != null) { |
|||
baos.close(); |
|||
} |
|||
} catch (IOException e) { |
|||
e.printStackTrace(); |
|||
} |
|||
} |
|||
return null; |
|||
} |
|||
|
|||
/** |
|||
* Base64字符串转图片 |
|||
* @param base64String |
|||
* @param imageFileName |
|||
*/ |
|||
public static void convertBase64StrToImage(String base64String, String imageFileName) { |
|||
ByteArrayInputStream bais = null; |
|||
try { |
|||
//获取图片类型
|
|||
String suffix = imageFileName.substring(imageFileName.lastIndexOf(".") + 1); |
|||
//获取JDK8里的解码器Base64.Decoder,将base64字符串转为字节数组
|
|||
byte[] bytes = Base64.getDecoder().decode(base64String); |
|||
//构建字节数组输入流
|
|||
bais = new ByteArrayInputStream(bytes); |
|||
//通过ImageIO把字节数组输入流转为BufferedImage
|
|||
BufferedImage bufferedImage = ImageIO.read(bais); |
|||
//构建文件
|
|||
File imageFile = new File(imageFileName); |
|||
//写入生成文件
|
|||
ImageIO.write(bufferedImage, suffix, imageFile); |
|||
} catch (Exception e) { |
|||
e.printStackTrace(); |
|||
} finally { |
|||
try { |
|||
if (bais != null) { |
|||
bais.close(); |
|||
} |
|||
} catch (IOException e) { |
|||
e.printStackTrace(); |
|||
} |
|||
} |
|||
} |
|||
|
|||
} |
|||
@ -1,4 +1,4 @@ |
|||
package com.visual.open.anpr.utils; |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
import com.alibaba.fastjson.JSON; |
|||
import com.alibaba.fastjson.JSONArray; |
|||
@ -1,4 +1,4 @@ |
|||
package com.visual.open.anpr.core.utils; |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
import org.opencv.core.Mat; |
|||
import org.opencv.core.Range; |
|||
@ -1,4 +1,4 @@ |
|||
package com.visual.open.anpr.core.utils; |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
import org.apache.commons.math3.linear.Array2DRowRealMatrix; |
|||
import org.apache.commons.math3.linear.ArrayRealVector; |
|||
@ -1,9 +1,9 @@ |
|||
package com.visual.open.anpr.core.utils; |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
import ai.onnxruntime.OnnxTensor; |
|||
import ai.onnxruntime.OrtSession; |
|||
import com.visual.open.anpr.core.domain.BorderMat; |
|||
import com.visual.open.anpr.core.domain.ImageMat; |
|||
import com.visual.open.anpr.server.domain.BorderMat; |
|||
import com.visual.open.anpr.server.domain.ImageMat; |
|||
import org.opencv.core.Mat; |
|||
|
|||
public class ReleaseUtil { |
|||
@ -1,4 +1,4 @@ |
|||
package com.visual.open.anpr.core.utils; |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
import java.util.function.DoubleUnaryOperator; |
|||
|
|||
@ -1,9 +1,7 @@ |
|||
package com.visual.open.anpr.core.utils; |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
import org.apache.commons.math3.linear.RealMatrix; |
|||
|
|||
import com.visual.open.anpr.core.utils.ArrayUtil; |
|||
|
|||
public class Similarity { |
|||
|
|||
/** |
|||
@ -1,4 +1,4 @@ |
|||
package com.visual.open.anpr.core.utils; |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
import java.util.Arrays; |
|||
import java.util.NoSuchElementException; |
|||
@ -1,4 +1,4 @@ |
|||
package com.visual.open.anpr.core.utils; |
|||
package com.visual.open.anpr.server.utils; |
|||
|
|||
public class ThreadUtil { |
|||
|
|||
@ -0,0 +1,170 @@ |
|||
//package com.visual.open.anpr.server.utils;
|
|||
//
|
|||
//import org.bytedeco.ffmpeg.avcodec.AVCodecParameters;
|
|||
//import org.bytedeco.ffmpeg.avformat.AVFormatContext;
|
|||
//import org.bytedeco.ffmpeg.avformat.AVStream;
|
|||
//import org.bytedeco.ffmpeg.global.avcodec;
|
|||
//import org.bytedeco.javacv.*;
|
|||
//import org.springframework.beans.factory.annotation.Value;
|
|||
//
|
|||
//import java.util.logging.Logger;
|
|||
//
|
|||
//
|
|||
//public class VideoUtil {
|
|||
//
|
|||
// @Value("3")
|
|||
// private Integer retryCount;//重试次数，默认值
|
|||
// @Value("100")
|
|||
// private Integer sleepTime;//休眠时间，默认值
|
|||
//
|
|||
// private static final Logger logger = Logger.getLogger(VideoUtil.class.getName());
|
|||
// /**
|
|||
// * 视频转码函数(仅转码)
|
|||
// *
|
|||
// * @param inputfile 原始视频文件完整路径
|
|||
// * @param outputfile 目标视频文件完整保存路径（必须完整文件名，即包含格式后缀，推荐格式后缀为.mp4）
|
|||
// * @throws Exception 异常
|
|||
// */
|
|||
// public static void videoConvert(String inputfile, String outputfile, Integer retryCount, Integer sleepTime) throws Exception {
|
|||
// if (outputfile.lastIndexOf('.') < 0) {
|
|||
// throw new Exception("Error! Output file format undetected!");
|
|||
// }
|
|||
// String format = outputfile.substring(outputfile.lastIndexOf('.'));
|
|||
//
|
|||
// FFmpegLogCallback.set();
|
|||
// Frame frame;
|
|||
// FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(inputfile);
|
|||
// FFmpegFrameRecorder recorder = null;
|
|||
//
|
|||
// try {
|
|||
// long start = System.currentTimeMillis();
|
|||
// logger.info("视频存储开始："+ DateUtil.getCurrentDateTime());
|
|||
// logger.info("开始初始化帧抓取器");
|
|||
// // 初始化帧抓取器，例如数据结构（时间戳、编码器上下文、帧对象等），
|
|||
// // 如果入参等于true，还会调用avformat_find_stream_info方法获取流的信息，放入AVFormatContext类型的成员变量oc中
|
|||
// grabber.start(true);
|
|||
//
|
|||
// logger.info("帧抓取器初始化完成");
|
|||
//
|
|||
// // grabber.start方法中，初始化的解码器信息存在放在grabber的成员变量oc中
|
|||
// AVFormatContext avformatcontext = grabber.getFormatContext();
|
|||
//
|
|||
// // 文件内有几个媒体流（一般是视频流+音频流）
|
|||
// int streamNum = avformatcontext.nb_streams();
|
|||
//
|
|||
// // 没有媒体流就不用继续了
|
|||
// if (streamNum < 1)
|
|||
// {
|
|||
// logger.info("文件内不存在媒体流");
|
|||
// throw new Exception("Error! There is no media stream in the file!");
|
|||
// }
|
|||
//
|
|||
// // 取得视频的帧率
|
|||
// double framerate = grabber.getVideoFrameRate();
|
|||
//
|
|||
// System.out.printf("视频帧率[%f]，视频时长[%d]秒，媒体流数量[%d]\r\n", framerate, avformatcontext.duration() / 1000000,
|
|||
// avformatcontext.nb_streams());
|
|||
//
|
|||
// // 遍历每一个流，检查其类型
|
|||
// for (int i = 0; i < streamNum; i++) {
|
|||
// AVStream avstream = avformatcontext.streams(i);
|
|||
// AVCodecParameters avcodecparameters = avstream.codecpar();
|
|||
// System.out.printf("流的索引[%d]，编码器类型[%d]，编码器ID[%d]\r\n", i, avcodecparameters.codec_type(),
|
|||
// avcodecparameters.codec_id());
|
|||
// }
|
|||
//
|
|||
// // 视频宽度
|
|||
// int frameWidth = grabber.getImageWidth();
|
|||
// // 视频高度
|
|||
// int frameHeight = grabber.getImageHeight();
|
|||
// // 音频通道数量
|
|||
// int audiochannels = grabber.getAudioChannels();
|
|||
//
|
|||
// System.out.printf("视频宽度[%d]，视频高度[%d]，音频通道数[%d]\r\n", frameWidth, frameHeight, audiochannels);
|
|||
//
|
|||
// recorder = new FFmpegFrameRecorder(outputfile, frameWidth, frameHeight, audiochannels);
|
|||
// recorder.setVideoCodec(avcodec.AV_CODEC_ID_H264);
|
|||
//
|
|||
// recorder.setFormat(format);
|
|||
// // 使用原始视频的码率，若需要则自行修改码率
|
|||
// recorder.setVideoBitrate(grabber.getVideoBitrate());
|
|||
//
|
|||
// // 一秒内的帧数，帧率
|
|||
// recorder.setFrameRate(framerate);
|
|||
//
|
|||
// // 两个关键帧之间的帧数
|
|||
// recorder.setGopSize((int)framerate);
|
|||
//
|
|||
// // 设置音频通道数，与视频源的通道数相等
|
|||
// recorder.setAudioChannels(grabber.getAudioChannels());
|
|||
//
|
|||
// recorder.start();
|
|||
//
|
|||
// int videoframenum = 0;
|
|||
// int audioframenum = 0;
|
|||
// int dataframenum = 0;
|
|||
//
|
|||
// logger.info("开始录制\r\n");
|
|||
//
|
|||
// // 循环条件
|
|||
// int count = 0;
|
|||
//
|
|||
// // 持续从视频源取帧
|
|||
// while (true) {
|
|||
// frame = grabber.grab();
|
|||
// if (frame != null) {
|
|||
// // 有图像，就把视频帧加一
|
|||
// if (null != frame.image) {
|
|||
// videoframenum++;
|
|||
// // 取出的每一帧，都保存到视频
|
|||
// recorder.record(frame);
|
|||
// }
|
|||
// // 有声音，就把音频帧加一
|
|||
// if (null != frame.samples) {
|
|||
// audioframenum++;
|
|||
// // 取出的每一帧，都保存到视频
|
|||
// recorder.record(frame);
|
|||
// }
|
|||
// // 有数据，就把数据帧加一
|
|||
// if (null != frame.data) {
|
|||
// dataframenum++;
|
|||
// }
|
|||
// count = 0;
|
|||
// }else {
|
|||
// if (retryCount == null || retryCount == 0) {
|
|||
// break;
|
|||
// }
|
|||
// if (count > retryCount) {
|
|||
// break;
|
|||
// }
|
|||
// if (sleepTime != null && sleepTime > 0) {
|
|||
// // 休眠一段时间后重试
|
|||
// Thread.sleep(sleepTime);
|
|||
// }
|
|||
// count ++;
|
|||
// }
|
|||
// }
|
|||
// long end = System.currentTimeMillis();
|
|||
// long second = (end - start) / 1000;
|
|||
// logger.info("视频存储结束："+ DateUtil.getCurrentDateTime());
|
|||
// logger.info("视频存储总耗时(秒)："+ second);
|
|||
// System.out.printf("转码完成，视频帧[%d]，音频帧[%d]，数据帧[%d]\r\n", videoframenum, audioframenum, dataframenum);
|
|||
// logger.info("转码完成，视频帧"+videoframenum+"，音频帧" + audioframenum + "，数据帧" + dataframenum);
|
|||
// } catch (Exception e) {
|
|||
// e.printStackTrace();
|
|||
// } finally {
|
|||
// if (recorder != null) {
|
|||
// try {
|
|||
// recorder.close();
|
|||
// } catch (Exception e) {
|
|||
// e.printStackTrace();
|
|||
// }
|
|||
// }
|
|||
// try {
|
|||
// grabber.close();
|
|||
// } catch (FrameGrabber.Exception e) {
|
|||
// e.printStackTrace();
|
|||
// }
|
|||
// }
|
|||
// }
|
|||
//}
|
|||
@ -1,32 +0,0 @@ |
|||
<?xml version="1.0" encoding="UTF-8"?> |
|||
<project xmlns="http://maven.apache.org/POM/4.0.0" |
|||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" |
|||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> |
|||
<modelVersion>4.0.0</modelVersion> |
|||
|
|||
<groupId>com.visual.open.anpr</groupId> |
|||
<artifactId>open-anpr-test</artifactId> |
|||
<version>1.1.0</version> |
|||
|
|||
<properties> |
|||
<java.version>1.8</java.version> |
|||
<maven.compiler.source>8</maven.compiler.source> |
|||
<maven.compiler.target>8</maven.compiler.target> |
|||
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding> |
|||
</properties> |
|||
|
|||
<dependencies> |
|||
<dependency> |
|||
<groupId>com.visual.open.anpr</groupId> |
|||
<artifactId>open-anpr-client</artifactId> |
|||
<version>${project.version}</version> |
|||
</dependency> |
|||
|
|||
<dependency> |
|||
<groupId>org.openpnp</groupId> |
|||
<artifactId>opencv</artifactId> |
|||
<version>4.6.0-0</version> |
|||
</dependency> |
|||
</dependencies> |
|||
|
|||
</project> |
|||
@ -1,84 +0,0 @@ |
|||
package com.visual.open.anpr.exps; |
|||
|
|||
import com.visual.open.anpr.PlateRecognition; |
|||
import com.visual.open.anpr.model.*; |
|||
import com.visual.open.anpr.utils.Base64Util; |
|||
import com.visual.open.anpr.utils.DrawImage; |
|||
import org.opencv.highgui.HighGui; |
|||
|
|||
import java.awt.*; |
|||
import java.io.File; |
|||
import java.util.List; |
|||
import java.util.Objects; |
|||
|
|||
public class PlateRecognitionExample { |
|||
|
|||
static{ nu.pattern.OpenCV.loadShared(); } |
|||
//本地开发模式
|
|||
public static String serverHost = "http://127.0.0.1:8080"; |
|||
//docker部署模式
|
|||
//public static String serverHost = "http://127.0.0.1:56790";
|
|||
//远程测试服务
|
|||
//public static String serverHost = "http://open-anpr.diven.nat300.top";
|
|||
|
|||
public static PlateRecognition plateRecognition = PlateRecognition.build(serverHost); |
|||
|
|||
/**搜索*/ |
|||
public static void recognition() { |
|||
String searchPath = "open-anpr-test/src/main/resources/image"; |
|||
File [] files = Objects.requireNonNull(new File(searchPath).listFiles()); |
|||
System.out.println(files.length); |
|||
for(int i=0; i < files.length; i++){ |
|||
System.out.println(i); |
|||
File imageA = files[i]; |
|||
String imageBase64 = Base64Util.encode(imageA.getAbsolutePath()); |
|||
Response<List<RecognitionRep>> response = plateRecognition.detection().recognition( |
|||
Recognition.build().setImage(imageBase64).setLimit(5) |
|||
); |
|||
DrawImage drawImage = DrawImage.build(imageA.getAbsolutePath()); |
|||
if(response.ok()){ |
|||
for(RecognitionRep recognitionRep : response.getData()){ |
|||
//画位置信息
|
|||
PlateLocation location = recognitionRep.getLocation(); |
|||
drawImage.drawLine( |
|||
new DrawImage.Point(location.getLeftTop().getX(), location.getLeftTop().getY()), |
|||
new DrawImage.Point(location.getRightTop().getX(), location.getRightTop().getY()), |
|||
2, Color.RED |
|||
); |
|||
drawImage.drawLine( |
|||
new DrawImage.Point(location.getRightTop().getX(), location.getRightTop().getY()), |
|||
new DrawImage.Point(location.getRightBottom().getX(), location.getRightBottom().getY()), |
|||
2, Color.RED |
|||
); |
|||
drawImage.drawLine( |
|||
new DrawImage.Point(location.getRightBottom().getX(), location.getRightBottom().getY()), |
|||
new DrawImage.Point(location.getLeftBottom().getX(), location.getLeftBottom().getY()), |
|||
2, Color.RED |
|||
); |
|||
drawImage.drawLine( |
|||
new DrawImage.Point(location.getLeftBottom().getX(), location.getLeftBottom().getY()), |
|||
new DrawImage.Point(location.getLeftTop().getX(), location.getLeftTop().getY()), |
|||
2, Color.RED |
|||
); |
|||
//画识别信息
|
|||
RecognitionInfo recognition = recognitionRep.getRecognition(); |
|||
int fonSize = Float.valueOf(1.4f * location.width() / recognition.getPlateNo().length()).intValue(); |
|||
drawImage.drawText(recognition.getPlateNo(), |
|||
new DrawImage.Point(location.minX(), location.minY()-(int)(fonSize*2.2)), fonSize, Color.RED); |
|||
drawImage.drawText((recognition.getLayout() == PlateLayout.SINGLE ? "单排" : "双排") + ":" + recognition.getPlateColor().getName(), |
|||
new DrawImage.Point(location.minX(), location.minY()-(int)(fonSize*1.2)), fonSize, Color.RED); |
|||
} |
|||
} |
|||
HighGui.imshow("image", drawImage.toMat()); |
|||
HighGui.waitKey(); |
|||
} |
|||
//退出程序
|
|||
System.exit(1); |
|||
} |
|||
|
|||
/**main**/ |
|||
public static void main(String[] args) { |
|||
recognition(); |
|||
} |
|||
|
|||
} |
|||
@ -1,107 +0,0 @@ |
|||
package com.visual.open.anpr.utils; |
|||
|
|||
import org.opencv.core.Mat; |
|||
import org.opencv.core.Range; |
|||
|
|||
import javax.imageio.ImageIO; |
|||
import java.awt.image.BufferedImage; |
|||
import java.io.ByteArrayOutputStream; |
|||
import java.util.Base64; |
|||
import java.util.Objects; |
|||
|
|||
public class MatUtil { |
|||
|
|||
/** |
|||
* 将Mat转换为BufferedImage |
|||
* @param mat |
|||
* @return BufferedImage |
|||
*/ |
|||
public static BufferedImage matToBufferedImage(Mat mat) { |
|||
int dataSize = mat.cols() * mat.rows() * (int) mat.elemSize(); |
|||
byte[] data = new byte[dataSize]; |
|||
mat.get(0, 0, data); |
|||
int type = mat.channels() == 1 ? BufferedImage.TYPE_BYTE_GRAY : BufferedImage.TYPE_3BYTE_BGR; |
|||
if (type == BufferedImage.TYPE_3BYTE_BGR) { |
|||
for (int i = 0; i < dataSize; i += 3) { |
|||
byte blue = data[i + 0]; |
|||
data[i + 0] = data[i + 2]; |
|||
data[i + 2] = blue; |
|||
} |
|||
} |
|||
BufferedImage image = new BufferedImage(mat.cols(), mat.rows(), type); |
|||
image.getRaster().setDataElements(0, 0, mat.cols(), mat.rows(), data); |
|||
return image; |
|||
} |
|||
|
|||
/** |
|||
* 将Mat转换为 Base64 |
|||
* @param mat |
|||
* @return Base64 |
|||
*/ |
|||
public static String matToBase64(Mat mat) { |
|||
ByteArrayOutputStream byteArrayOutputStream = null; |
|||
try { |
|||
byteArrayOutputStream = new ByteArrayOutputStream(); |
|||
ImageIO.write(matToBufferedImage(mat), "jpg", byteArrayOutputStream); |
|||
byte[] bytes = byteArrayOutputStream.toByteArray(); |
|||
Base64.Encoder encoder = Base64.getMimeEncoder(); |
|||
return encoder.encodeToString(Objects.requireNonNull(bytes)); |
|||
}catch (Exception e){ |
|||
throw new RuntimeException(e); |
|||
}finally { |
|||
if(null != byteArrayOutputStream){ |
|||
try { |
|||
byteArrayOutputStream.close(); |
|||
} catch (Exception e) {} |
|||
} |
|||
} |
|||
} |
|||
|
|||
public static Mat concat(Mat m1, Mat ...m2){ |
|||
Mat mat = m1; |
|||
if(null != m2){ |
|||
for(Mat m : m2){ |
|||
mat = concat(mat, m1); |
|||
} |
|||
} |
|||
return mat; |
|||
} |
|||
|
|||
/** |
|||
* 横向拼接两个图像的数据（Mat），该两个图像的类型必须是相同的类型，如：均为CvType.CV_8UC3类型 |
|||
* @author bailichun |
|||
* @since 2020.02.20 15:00 |
|||
* @param m1 要合并的图像1（左图） |
|||
* @param m2 要合并的图像2（右图） |
|||
* @return 拼接好的Mat图像数据集。其高度等于两个图像中高度较大者的高度；其宽度等于两个图像的宽度之和。类型与两个输入图像相同。 |
|||
* @throws Exception 当两个图像数据的类型不同时，抛出异常 |
|||
*/ |
|||
public static Mat concat(Mat m1, Mat m2){ |
|||
if(m1.type() != m2.type()){ |
|||
throw new RuntimeException("concat:两个图像数据的类型不同！"); |
|||
} |
|||
long time = System.currentTimeMillis(); |
|||
//宽度为两图的宽度之和
|
|||
double w = m1.size().width + m2.size().width; |
|||
//高度取两个矩阵中的较大者的高度
|
|||
double h = m1.size().height > m2.size().height ? m1.size().height : m2.size().height; |
|||
//创建一个大矩阵对象
|
|||
Mat des = Mat.zeros((int)h, (int)w, m1.type()); |
|||
//在最终的大图上标记一块区域，用于存放复制图1（左图）的数据，大小为从第0列到m1.cols()列
|
|||
Mat rectForM1 = des.colRange(new Range(0, m1.cols())); |
|||
//标记出位于rectForM1的垂直方向上中间位置的区域，高度为图1的高度，此时该区域的大小已经和图1的大小相同。（用于存放复制图1（左图）的数据）
|
|||
int rowOffset1 = (int)(rectForM1.size().height-m1.rows())/2; |
|||
rectForM1 = rectForM1.rowRange(rowOffset1, rowOffset1 + m1.rows()); |
|||
//在最终的大图上标记一块区域，用于存放复制图2（右图）的数据
|
|||
Mat rectForM2 = des.colRange(new Range(m1.cols(), des.cols())); |
|||
//标记出位于rectForM2的垂直方向上中间位置的区域，高度为图2的高度，此时该区域的大小已经和图2的大小相同。（用于存放复制图2（右图）的数据）
|
|||
int rowOffset2 = (int)(rectForM2.size().height-m2.rows())/2; |
|||
rectForM2 = rectForM2.rowRange(rowOffset2, rowOffset2 + m2.rows()); |
|||
//将图1拷贝到des的指定区域 rectForM1
|
|||
m1.copyTo(rectForM1); |
|||
//将图2拷贝到des的指定区域 rectForM2
|
|||
m2.copyTo(rectForM2); |
|||
return des; |
|||
} |
|||
|
|||
} |
|||
|
Before Width: | Height: | Size: 107 KiB |
|
Before Width: | Height: | Size: 64 KiB |
|
Before Width: | Height: | Size: 40 KiB |
|
Before Width: | Height: | Size: 390 KiB |
|
Before Width: | Height: | Size: 47 KiB |
|
Before Width: | Height: | Size: 64 KiB |
|
Before Width: | Height: | Size: 671 KiB |
|
Before Width: | Height: | Size: 328 KiB |
|
Before Width: | Height: | Size: 382 KiB |
|
Before Width: | Height: | Size: 85 KiB |
|
Before Width: | Height: | Size: 29 KiB |
|
Before Width: | Height: | Size: 999 KiB |
|
Before Width: | Height: | Size: 34 KiB |
|
Before Width: | Height: | Size: 400 KiB |
|
Before Width: | Height: | Size: 659 KiB |
|
Before Width: | Height: | Size: 82 KiB |
|
Before Width: | Height: | Size: 979 KiB |
Loading…
Reference in new issue