package com.lihao.client;
import java.io.IOException;
import java.net.URI;
import java.net.URLEncoder;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import java.nio.charset.StandardCharsets;
import java.nio.file.Path;
import java.util.Map;
public class HttpUtils {
/**
*
* @param url API地址
* @param params 参数
* @param contentType 响应类型
* @param data 二进制参数
* @param file 文件参数
* @return
* @throws IOException
* @throws InterruptedException
*/
public static String postRequest(String url, Map<String,String> params, String contentType, byte[] data, Path file) throws IOException, InterruptedException {
//初始化client
HttpClient client = HttpClient.newBuilder().version(HttpClient.Version.HTTP_1_1).build();
//初始化builder
HttpRequest.Builder builder = HttpRequest.newBuilder();
//处理自定义参数
if (params != null) {
int i = 0;
StringBuilder sb = new StringBuilder(url);
sb.append("?");
for (Map.Entry<String, String> entry : params.entrySet()) {
if (i > 0) {
sb.append("&");
}
sb.append(URLEncoder.encode(entry.getKey(), StandardCharsets.UTF_8));
sb.append("=");
sb.append(URLEncoder.encode(entry.getValue(), StandardCharsets.UTF_8));
i++;
}
url = sb.toString();
}
//构造请求url+参数
builder.uri(URI.create(url));
//设置响应类型
if (contentType != null) {
builder.header("Content-Type", contentType);
}
if (data != null) {//处理二进制参数
builder.POST(HttpRequest.BodyPublishers.ofByteArray(data));
} else if (file != null) {//处理文件参数
builder.POST(HttpRequest.BodyPublishers.ofFile(file));
} else {
builder.POST(HttpRequest.BodyPublishers.noBody());
}
//初始化request
HttpRequest request = builder.build();
//发送请求
HttpResponse<byte[]> response = client.send(request, HttpResponse.BodyHandlers.ofByteArray());
//处理响应
String result = new String(response.body(), StandardCharsets.UTF_8);
//结果返回
return result;
}
}
package com.lihao.client;
import java.util.Map;
public class RegisterDemo {
public static void main(String[] args) throws Exception {
// 模型地址
String url = "file:///D://LIHAOWORK//serving-0.23.0//person-test.zip";
//构造参数
Map<String, String> params = Map.of("url", url, "modelName","person-test","engine", "OnnxRuntime");
//注册模型
String response = HttpUtils.postRequest("http://127.0.0.1:8080/models", params, null, null, null);
System.out.println(response);
}
}
运行结果，显示"status": "Model “person_test” registered."模型已经注册
查看模型状态和访问地址
package com.lihao.client;
import java.nio.file.Path;
public class PredictDemo {
public static void main(String[] args) throws Exception {
String url = "http://127.0.0.1:8080/predictions/person_test/";
String response = HttpUtils.postRequest(
url,
null,
"application/octet-stream",
null,
Path.of("D:\\LIHAOWORK\\serving-0.23.0\\ren.jpg"));
System.out.println(response);
}
}
推理结果
通过结果绘制原图片
将PredictDemo更改了一下，增加了随机抽帧一张图片，进行推理，将推理结果进行绘制，最后将绘制后的图片展示出来。
package com.lihao.client;
import ai.djl.modality.cv.BufferedImageFactory;
import ai.djl.modality.cv.Image;
import ai.djl.modality.cv.output.BoundingBox;
import ai.djl.modality.cv.output.DetectedObjects;
import ai.djl.modality.cv.output.Point;
import ai.djl.modality.cv.output.Rectangle;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import org.bytedeco.ffmpeg.global.avutil;
import org.bytedeco.javacv.*;
import javax.swing.*;
import java.awt.image.BufferedImage;
import java.io.*;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;
public class PredictDemo {
public static void main(String[] args) throws Exception {
String path = getImage();
String url = "http://127.0.0.1:8080/predictions/person_test/";
String response = HttpUtils.postRequest(
url,
null,
"application/octet-stream",
null,
Path.of(path));
System.out.println(response);
DetectedObjects result = str2DetectedObjects(response);
Image image = BufferedImageFactory.getInstance().fromFile(Path.of(path));
image.drawBoundingBoxes(result);
CanvasFrame canvasFrame = new CanvasFrame("摄像机");
canvasFrame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
canvasFrame.setAlwaysOnTop(true);
canvasFrame.showImage(image2Frame(image));
}
/**
* 获取一张图片，返回图片路径
* @return
*/
private static String getImage() {
FFmpegFrameGrabber grabber = null;
try {
grabber = FFmpegFrameGrabber.createDefault("rtsp://admin:admin1234@192.168.66.150:554/cam/realmonitor?channel=12&subtype=1");
grabber.setOption("rtsp_transport", "tcp"); // 使用tcp的方式
grabber.setOption("stimeout", "5000000");
grabber.setPixelFormat(avutil.AV_PIX_FMT_RGB24); // 像素格式
grabber.setImageWidth(640);
grabber.setImageHeight(640);
grabber.setFrameRate(30);
grabber.start();
Frame frame = grabber.grabFrame();//抽取一帧
Image image = frame2Image(frame);
String name = UUID.randomUUID().toString();
String path = "D:\\LIHAOWORK\\serving-0.23.0\\"+ name+".jpg";
OutputStream os = new FileOutputStream(new File(path));
image.save(os,"jpg");
return path;
} catch (FFmpegFrameGrabber.Exception e) {
e.printStackTrace();
} catch (FrameGrabber.Exception e) {
e.printStackTrace();
} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
return null;
}
/**
* str2DetectedObjects
*
* @param str
* @return
*/
private static DetectedObjects str2DetectedObjects(String str) {
List<JSONObject> jsons = JSONObject.parseArray(str, JSONObject.class);
List<String> classList = new ArrayList<>();
List<Double> probList = new ArrayList<>();
List<BoundingBox> rectList = new ArrayList<>();
jsons.forEach(item -> {
classList.add(item.getString("className"));
probList.add(item.getDouble("probability"));
JSONObject b = item.getJSONObject("boundingBox");
List<Point> corners = JSON.parseArray(b.getString("corners"), Point.class);
Rectangle newBox = new Rectangle(corners.get(0), b.getDouble("width"), b.getDouble("height"));
rectList.add(newBox);
});
DetectedObjects r = new DetectedObjects(classList, probList, rectList);
return r;
}
/**
* image2Frame
*
* @param image
* @return
*/
private static Frame image2Frame(Image image) {
BufferedImage temp = (BufferedImage) image.getWrappedImage();
Frame frame = Java2DFrameUtils.toFrame(temp);
return frame;
}
/**
* frame2Image
*
* @param frame
* @return
*/
private static Image frame2Image(Frame frame) {
BufferedImage temp = Java2DFrameUtils.toBufferedImage(frame);
Image image = BufferedImageFactory.getInstance().fromImage(temp);
return image;
}
}
package com.lihao.client;
import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import java.nio.charset.StandardCharsets;
public class UnregisterDemo {
public static void main(String[] args) throws Exception {
String url = "http://127.0.0.1:8080/models/person_test/";
HttpClient client = HttpClient.newBuilder().version(HttpClient.Version.HTTP_1_1).build();
HttpRequest.Builder builder = HttpRequest.newBuilder();
builder.uri(URI.create(url));
builder.DELETE();
HttpRequest request = builder.build();
HttpResponse<byte[]> response =
client.send(request, HttpResponse.BodyHandlers.ofByteArray());
System.out.println(new String(response.body(), StandardCharsets.UTF_8));
}
}
