From 5a34a5a7bdb1c491661193f392c206cf36851ade Mon Sep 17 00:00:00 2001 From: "li.zhenye" Date: Thu, 4 Aug 2022 15:13:16 +0800 Subject: [PATCH] =?UTF-8?q?[ext]=20=E6=B7=BB=E5=8A=A0=E4=BA=86mask?= =?UTF-8?q?=E5=A4=A7=E5=B0=8F=E7=9A=84=E6=9B=B4=E6=94=B9=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- valve_test.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/valve_test.py b/valve_test.py index 66eb6a7..ce37d7c 100755 --- a/valve_test.py +++ b/valve_test.py @@ -1,5 +1,6 @@ import socket +import cv2 import numpy as np @@ -102,6 +103,7 @@ m. 模式切换：测下一个喷阀还是重发？ elif value == 'g': # g.发个da和db完全重叠的mask mask_a, mask_b = np.eye(256, dtype=np.uint8), np.eye(256, 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 @@ -110,6 +112,7 @@ m. 模式切换：测下一个喷阀还是重发？ elif value == 'h': # h.发个da和db呈现出X形的mask mask_a, mask_b = np.eye(256, dtype=np.uint8), np.eye(256, 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 @@ -185,6 +188,7 @@ if __name__ == '__main__': parser.add_argument('-m', default='192.168.10.8', help='指定master主机名') parser.add_argument('-p', default=13452, help='指定端口') args = parser.parse_args() + mask_size = (1024, 256) # size of cv (Width, Height) if args.c: print("运行客户机") virtual_valve = VirtualValve(host=args.m, port=args.p)