mirror of
https://github.com/Karllzy/cotton_color.git
synced 2025-11-09 11:13:54 +00:00
socket
This commit is contained in:
wrz1-zzzzz
parent
29fe90eda9
commit
73dcb7578c
20
DPL/CPU条件下预测速度.md
Normal file
20
DPL/CPU条件下预测速度.md
Normal file
@ -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<br/><br/> Intel(R) Core(TM) i9-9900K CPU @ 3.60GHz<br/><br/> 基准速度: 3.60 GHz<br/> 插槽: 1<br/> 内核: 8<br/> 逻辑处理器: 16<br/> 虚拟化: 已启用<br/> L1 缓存: 512 KB<br/> L2 缓存: 2.0 MB<br/> L3 缓存: 16.0 MB<br/><br/> 利用率 11%<br/> 速度 4.74 GHz<br/> 正常运行时间 8:05:55:50<br/> 进程 242<br/> 线程 3668<br/> 句柄 115016 |
|
||||
| 1280 | 1 | 地膜 | cpu | 0.003 | 230 | 0.22 | 0.001 | CPU<br/><br/> Intel(R) Core(TM) i9-9900K CPU @ 3.60GHz<br/><br/> 基准速度: 3.60 GHz<br/> 插槽: 1<br/> 内核: 8<br/> 逻辑处理器: 16<br/> 虚拟化: 已启用<br/> L1 缓存: 512 KB<br/> L2 缓存: 2.0 MB<br/> L3 缓存: 16.0 MB<br/><br/> 利用率 100%<br/> 速度 4.68 GHz<br/> 正常运行时间 8:06:05:35<br/> 进程 242<br/> 线程 3771<br/> 句柄 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<br/><br/> NVIDIA GeForce RTX 3060 Laptop GPU<br/><br/> 驱动程序版本: 32.0.15.6070<br/> 驱动程序日期: 2024/7/10<br/> DirectX 版本: 12 (FL 12.1)<br/> 物理位置： PCI 总线 1、设备 0、功能 0<br/><br/> 利用率 84%<br/> 专用 GPU 内存 3.9/6.0 GB<br/> 共享 GPU 内存 0.1/7.6 GB<br/> GPU 内存 4.1/13.6 GB |
|
||||
| 640 | 6 | 地膜 | 3060 | 1.1ms | 40（每批） | 6.7 | 4 | GPU 0<br/><br/> NVIDIA GeForce RTX 3060 Laptop GPU<br/><br/> 驱动程序版本: 32.0.15.6070<br/> 驱动程序日期: 2024/7/10<br/> DirectX 版本: 12 (FL 12.1)<br/> 物理位置： PCI 总线 1、设备 0、功能 0<br/><br/> 利用率 100%<br/> 专用 GPU 内存 4.7/6.0 GB<br/> 共享 GPU 内存 0.1/7.6 GB<br/> GPU 内存 4.8/13.6 GB |
|
||||
| | | | | | | | | |
|
||||
228
socket/socket.py
Normal file
228
socket/socket.py
Normal file
@ -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()
|
||||
Loading…
Reference in New Issue
Block a user