总的来说就是使用socket发送base64图片编码到python客户端,需要注意的是默认对发送数据,接受数据大小有限制,大概60kb左右,需要根据图片大小设置socket缓冲区
java:
/**
* 发包工具
*/
private static byte[] socket(String base64) {
String host = "127.0.0.1"; // Python服务器的地址
int port = 8888; // Python服务器的端口号
try {
// 连接到Python服务器
Socket socket = new Socket(host, port);
// 设置发送缓冲区的大小
int sendBufferSize = 2*1024 * 1024; // 2 MB
socket.setSendBufferSize(sendBufferSize);
// 获取发送缓冲区的大小
int actualSendBufferSize = socket.getSendBufferSize();
System.out.println("Send Buffer Size: " + actualSendBufferSize);
System.out.println("Connected to Python server");
OutputStream outputStream = socket.getOutputStream();
String message = base64;
// message += "#";
outputStream.write(message.getBytes());
// 接收Python服务器返回的数据
InputStream inputStream = socket.getInputStream();
BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream));
String response = reader.readLine();
socket.close();
// 关闭套接字
return response.getBytes();
} catch (IOException e) {
e.printStackTrace();
return null;
}
}
python:
HOST = '127.0.0.1' # 监听的地址
PORT = 8888 # 监听的端口号
# 创建套接字并监听
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_socket.bind((HOST, PORT))
server_socket.listen(1)
# 设置接收缓冲区大小为2MB
server_socket.setsockopt(server_socket, 2*1024 * 1024)
while True:
try:
# 等待客户端连接
client_socket, address = server_socket.accept()
client_socket.settimeout(0.5)
# 接收Java客户端发送的数据并返回
result = ""
while True:
data = client_socket.recv(1024)
if len(data) < 1024:
result += data.decode()
break
result += data.decode()
#'NoneType' object has no attribute 'shape'
# 将 Base64 字符串解码为字节数据
image_bytes = base64.b64decode(result)
print(image_bytes) # 输出: 编码前数据
# 返回标签
client_socket.send(image_bytes)
client_socket.close()
except Exception as err:
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_socket.bind((HOST, PORT))
server_socket.listen(1)
print(err)
continue