From 73dcb7578c902eef6af1f6da8a6522e55c01759f Mon Sep 17 00:00:00 2001
From: wrz1-zzzzz <914921336@qq.com>
Date: Mon, 2 Dec 2024 16:38:10 +0800
Subject: [PATCH] socket
---
DPL/CPU条件下预测速度.md | 20 ++++
socket/socket.py | 228 +++++++++++++++++++++++++++++++++++++++
2 files changed, 248 insertions(+)
create mode 100644 DPL/CPU条件下预测速度.md
create mode 100644 socket/socket.py
diff --git a/DPL/CPU条件下预测速度.md b/DPL/CPU条件下预测速度.md
new file mode 100644
index 0000000..d84622f
--- /dev/null
+++ b/DPL/CPU条件下预测速度.md
@@ -0,0 +1,20 @@
+## CPU条件下预测速度
+
+
+
+| imgsiz | batchsz | modeltype | device | preprocess | total | inference | nms | memory |
+| ------ | ------- | --------- | ------ | ---------- | ------------------------------------------------------------ | --------- | ----- | ------------------------------------------------------------ |
+| 640 | 1 | 地膜 | cpu | 0.001 | 45 | 0.045 | 0.001 | CPU
Intel(R) Core(TM) i9-9900K CPU @ 3.60GHz
基准速度: 3.60 GHz
插槽: 1
内核: 8
逻辑处理器: 16
虚拟化: 已启用
L1 缓存: 512 KB
L2 缓存: 2.0 MB
L3 缓存: 16.0 MB
利用率 11%
速度 4.74 GHz
正常运行时间 8:05:55:50
进程 242
线程 3668
句柄 115016 |
+| 1280 | 1 | 地膜 | cpu | 0.003 | 230 | 0.22 | 0.001 | CPU
Intel(R) Core(TM) i9-9900K CPU @ 3.60GHz
基准速度: 3.60 GHz
插槽: 1
内核: 8
逻辑处理器: 16
虚拟化: 已启用
L1 缓存: 512 KB
L2 缓存: 2.0 MB
L3 缓存: 16.0 MB
利用率 100%
速度 4.68 GHz
正常运行时间 8:06:05:35
进程 242
线程 3771
句柄 116048 |
+| 640 | 6 | 地膜 | | | | | | |
+| 1280 | 6 | 地膜 | | | | | | |
+
+## GPU
+
+
+
+| imgsiz | batchsz | modeltype | device | preprocess | total | inference | nms | memory |
+| ------ | ------- | --------- | ------ | ---------- | ----------- | --------- | ---- | ------------------------------------------------------------ |
+| 1280 | 6 | 地膜 | 3060 | 1.8 | 140（每批） | 22.8 | 3.6 | GPU 0
NVIDIA GeForce RTX 3060 Laptop GPU
驱动程序版本: 32.0.15.6070
驱动程序日期: 2024/7/10
DirectX 版本: 12 (FL 12.1)
物理位置： PCI 总线 1、设备 0、功能 0
利用率 84%
专用 GPU 内存 3.9/6.0 GB
共享 GPU 内存 0.1/7.6 GB
GPU 内存 4.1/13.6 GB |
+| 640 | 6 | 地膜 | 3060 | 1.1ms | 40（每批） | 6.7 | 4 | GPU 0
NVIDIA GeForce RTX 3060 Laptop GPU
驱动程序版本: 32.0.15.6070
驱动程序日期: 2024/7/10
DirectX 版本: 12 (FL 12.1)
物理位置： PCI 总线 1、设备 0、功能 0
利用率 100%
专用 GPU 内存 4.7/6.0 GB
共享 GPU 内存 0.1/7.6 GB
GPU 内存 4.8/13.6 GB |
+| | | | | | | | | |
\ No newline at end of file
diff --git a/socket/socket.py b/socket/socket.py
new file mode 100644
index 0000000..ab09692
--- /dev/null
+++ b/socket/socket.py
@@ -0,0 +1,228 @@
+import socket
+import time
+
+import cv2
+import numpy as np
+
+
+class ValveTest:
+ def __init__(self, host=None, port=13452, column_num=64):
+ self.column_num = column_num
+ self.increase_modes = ['测下一个', '重复测试']
+ self.last_cmd = None #上一次阀门测试编号
+ self.increase_mode = 0 #当前模式 modes
+ self.reminder = None #显示提示信息
+ self.update_reminder()
+ self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 创建 socket 对象 用于监听客户端的连接请求。AF_INET表示使用IPv4协议，SOCK_STREAM表示使用TCP协议
+ self.s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+ host = socket.gethostname() if host is None else host # 获取本地主机名
+ print(f"Service Address {host}, {port}.")
+ self.s.bind((host, port)) # 绑定端口
+ self.s.listen(5) # 等待客户端连接
+ self.c = None
+
+
+ def update_reminder(self):
+ self.reminder = f"""======================================================================================
+快，给我个指令😉😉😉︎：
+a. 开始命令 st. e. 设置 光谱(a)相机 的延时，格式 e,500
+b. 停止命令 sp. f. 设置 彩色(b)相机 的延时, 格式 f,500
+c. 设置光谱相机分频系数,4的倍数且>=8, 格式 c,8 g. 发个da和db完全重叠的mask
+d. 阀板的脉冲分频系数,>=2即可 h. 发个da和db呈现出X形的mask
+m. 模式切换：测下一个喷阀还是重发？
+你给我个小于{self.column_num}的数字，我就测试对应的喷阀。如果已经测试过一个，可以直接回车{self.increase_modes[self.increase_mode]}。
+给q指令我就退出。
+======================================================================================\n"""
+
+ def run(self):
+ print("我在等连接...")
+ self.c, addr = self.s.accept() # 建立客户端连接 客户端地址和端口信息
+ print('和它的链接建立成功了：', addr)
+ self.c.settimeout(0.1)
+ while True:
+ data = ''
+ try:
+ data = self.c.recv(1024)
+ except Exception as e:
+ print(f"===================================================================")
+ if len(data) > 0:
+ print("receive data!!!")
+ print(data)
+ value = input(self.reminder)
+ if value == 'q':
+ print("好的，我退出啦")
+ self.s.close()
+ break
+ else:
+ self.process_cmd(value)
+ self.c.close() # 关闭连接
+
+ @staticmethod
+ def cmd_padding(cmd):
+ return b'\xAA' + cmd + b'\xFF\xFF\xBB'
+
+ @staticmethod
+ def param_cmd_parser(cmd, default_value, checker=None):
+ try:
+ value = int(cmd.split(',')[-1])
+ except:
+ print(f'你给的值不对啊，我先给你弄个{default_value}吧')
+ value = default_value
+ if checker is not None:
+ if not checker(value):
+ return None
+ return value
+
+ def process_cmd(self, value):
+ if value == 'a':
+ # a.开始命令
+ cmd = b'\x00\x03' + 'st'.encode('ascii') + b'\xFF'
+ elif value == 'b':
+ # b.停止命令
+ cmd = b'\x00\x03' + 'sp'.encode('ascii') + b'\xFF'
+ elif value.startswith('c'):
+ # c. 设置光谱相机分频，得是4的倍数而且>=8，格式：c,8
+ checker = lambda x: (x % 4 == 0) and (x >= 8)
+ value = self.param_cmd_parser(value, default_value=8, checker=checker)
+ if value is None:
+ print("值需要是4的倍数且大于8")
+ return
+ cmd = b'\x00\x0a' + 'sc'.encode('ascii') + f"{value:08d}".encode('ascii')
+ elif value.startswith('d'):
+ # d. 阀板的脉冲分频系数，>=2即可
+ checker = lambda x: x >= 2
+ value = self.param_cmd_parser(value, default_value=2, checker=checker)
+ if value is None:
+ print("你得大于等于2")
+ return
+ cmd = b'\x00\x0a' + 'sv'.encode('ascii') + f"{value:08d}".encode('ascii')
+ elif value.startswith('e'):
+ # e. 设置 光谱(a)相机 的延时，格式 e,500
+ checker = lambda x: (x >= 0)
+ value = self.param_cmd_parser(value, default_value=2, checker=checker)
+ if value is None:
+ print("你得大于等于0")
+ return
+ cmd = b'\x00\x0a' + 'sa'.encode('ascii') + f"{value:08d}".encode('ascii')
+ elif value.startswith('f'):
+ # f. 设置 RGB(b)相机 的延时，格式 e,500
+ checker = lambda x: (x >= 0)
+ value = self.param_cmd_parser(value, default_value=2, checker=checker)
+ if value is None:
+ print("你得大于等于0")
+ return
+ cmd = b'\x00\x0a' + 'sb'.encode('ascii') + f"{value:08d}".encode('ascii')
+ elif value == 'g':
+ # g.发个da和db完全重叠的mask
+ mask_a, mask_b = np.eye(self.column_num, dtype=np.uint8), np.eye(self.column_num, dtype=np.uint8)
+ mask_a, mask_b = [cv2.resize(mask, mask_size) for mask in [mask_a, mask_b]]
+ len_a, data_a = self.format_data(mask_a)
+ len_b, data_b = self.format_data(mask_b)
+ cmd = len_a + 'da'.encode('ascii') + data_a
+ self.send(cmd)
+ cmd = len_b + 'db'.encode('ascii') + data_b
+ elif value == 'h':
+ # h.发个da和db呈现出X形的mask
+ mask_a, mask_b = np.eye(self.column_num, dtype=np.uint8), np.eye(self.column_num, dtype=np.uint8).T
+ mask_a, mask_b = [cv2.resize(mask, mask_size) for mask in [mask_a, mask_b]]
+ len_a, data_a = self.format_data(mask_a)
+ len_b, data_b = self.format_data(mask_b)
+ cmd = len_a + 'da'.encode('ascii') + data_a
+ self.send(cmd)
+ cmd = len_b + 'db'.encode('ascii') + data_b
+ elif value == 'i':
+ # i. 完全模拟全流程
+ # a.开始命令
+ cmd = b'\x00\x03' + 'st'.encode('ascii') + b'\xFF'
+ self.send(cmd)
+ time.sleep(0.170)
+ for i in range(1000):
+ mask_a, mask_b = np.eye(self.column_num, dtype=np.uint8), np.eye(self.column_num, dtype=np.uint8)
+ mask_a, mask_b = [cv2.resize(mask, mask_size) for mask in [mask_a, mask_b]]
+ len_a, data_a = self.format_data(mask_a)
+ len_b, data_b = self.format_data(mask_b)
+ cmd = len_a + 'da'.encode('ascii') + data_a
+ self.send(cmd)
+ time.sleep(0.150)
+ # cmd = len_b + 'db'.encode('ascii') + data_b
+ elif value == 'm':
+ self.increase_mode = int(1 - self.increase_mode)
+ self.update_reminder()
+ print("模式切换")
+ return
+ elif value == '' and self.last_cmd is not None:
+ if self.increase_mode == 0:
+ self.last_cmd += 1
+ if self.last_cmd > 256:
+ self.last_cmd = 1
+ print(f'自动变化到 喷阀测试 {self.last_cmd}')
+ value = self.last_cmd
+ cmd = b'\x00\x0A' + 'te'.encode('ascii') + f"{value - 1:08d}".encode('ascii')
+ else:
+ try:
+ value = int(value)
+ except Exception as e:
+ print(e)
+ print(f"你给的指令: {value} 咋看都不对")
+ return
+ if (value <= self.column_num) and (value >= 1):
+ cmd = b'\x00\x0A' + 'te'.encode('ascii') + f"{value - 1:08d}".encode('ascii')
+ self.last_cmd = value
+ elif value == (self.column_num+1):
+ cmd = b'\x00\x0A' + 'te'.encode('ascii') + f"{value:08d}".encode('ascii')
+ print(f"恭喜你发现了这个隐藏的{self.column_num+1}号流水灯指令😝😝😝，好厉害。")
+ else:
+ print(f'你给的指令: {value} 值不对，我们有 {self.column_num} 个阀门, 范围是 [1, {self.column_num}]，略大一个好像也可以')
+ return
+ self.send(cmd)
+
+ def send(self, cmd: bytes) -> None:
+ cmd = self.cmd_padding(cmd)
+ print("我要 send 这个了:")
+ print(cmd.hex())
+ try:
+ self.c.send(cmd)
+ except Exception as e:
+ print(f"发失败了, 这是我找到的错误信息:\n{e}")
+ return
+ print("发好了")
+
+ @staticmethod
+ def format_data(array_to_send: np.ndarray) -> (bytes, bytes):
+ data = np.packbits(array_to_send, axis=-1)
+ data = data.tobytes()
+ data_len = (len(data) + 2).to_bytes(2, 'big')
+ return data_len, data
+
+
+class VirtualValve:
+ def __init__(self, host, port):
+ self.client = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 声明socket类型，同时生成链接对象
+ self.client.connect((host, port)) # 建立一个链接，连接到本地的13452端口
+
+ def run(self):
+ while True:
+ # addr = client.accept()
+ # print '连接地址：', addr
+ data = self.client.recv(4096) # 接收一个信息，并指定接收的大小 为1024字节
+ print("原始报文:")
+ print(data.hex()) # 输出我接收的信息
+
+
+if __name__ == '__main__':
+ import argparse
+ parser = argparse.ArgumentParser(description='阀门测程序')
+ parser.add_argument('-c', default=False, action='store_true', help='是否是开个客户端', required=False)
+ parser.add_argument('-m', default='192.168.10.8', help='指定master主机名')
+ parser.add_argument('-p', default=13452, help='指定端口')
+ parser.add_argument('-cnt', default=64, help='指定端口')
+ args = parser.parse_args()
+ mask_size = (1024, args.cnt) # size of cv (Width, Height)
+ if args.c:
+ print("运行客户机")
+ virtual_valve = VirtualValve(host=args.m, port=args.p)
+ virtual_valve.run()
+ else:
+ print("运行主机")
+ valve_tester = ValveTest(host=args.m, port=args.p, column_num=args.cnt)
+ valve_tester.run()