NanjingForestryUniversity/supermachine--tomato-passion_fruit
3
0
Fork 0
mirror of https://github.com/NanjingForestryUniversity/supermachine--tomato-passion_fruit.git synced 2025-11-09 14:54:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat:新增qt_test的预热数据部分，测试成功，预热之后全流程总用时172.5ms

Browse Source
This commit is contained in:
TG 2024-06-17 18:04:53 +08:00
parent 28d403fba7
commit 74e141bb8f
2 changed files with 33 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
20240529RGBtest3/main.py
View File

@ -89,7 +89,7 @@ def main(is_debug=False):
_ = detector.predict(np.ones((30, 30, 224), dtype=np.uint16))
_, _, _, _, _ =dp.analyze_tomato(cv2.imread(r'D:\project\supermachine--tomato-passion_fruit\20240529RGBtest3\data\tomato_img\bad\71.bmp'))
_, _, _, _, _ = dp.analyze_passion_fruit(cv2.imread(r'D:\project\supermachine--tomato-passion_fruit\20240529RGBtest3\data\passion_fruit_img\38.bmp'))
# print('初始化完成')
print('初始化完成')
rgb_receive_name = r'\\.\pipe\rgb_receive'
rgb_send_name = r'\\.\pipe\rgb_send'
@ -98,9 +98,12 @@ def main(is_debug=False):
rgb_receive, rgb_send, spec_receive = pipe.create_pipes(rgb_receive_name, rgb_send_name, spec_receive_name)
# 预热循环，只处理cmd为'YR'的数据
while True:
start_time00 = time.time()
data = pipe.receive_rgb_data(rgb_receive)
cmd, _ = pipe.parse_img(data)
if cmd != 'YR':
end_time00 = time.time()
print(f'接收预热数据时间：{end_time00 - start_time00}秒')
if cmd == 'YR':
break # 当接收到的不是预热命令时，结束预热循环
while True:
@ -108,15 +111,15 @@ def main(is_debug=False):
images = []
cmd = None
for _ in range(5):
# start_time1 = time.time()
start_time1 = time.time()
data = pipe.receive_rgb_data(rgb_receive)
# end_time10 = time.time()
# print(f'接收一份数据时间：{end_time10 - start_time1}秒')
# start_time11 = time.time()
end_time10 = time.time()
print(f'接收一份数据时间：{end_time10 - start_time1}秒')
start_time11 = time.time()
cmd, img = pipe.parse_img(data)
# end_time1 = time.time()
# print(f'处理一份数据时间：{end_time1 - start_time11}秒')
# print(f'接收1张图时间：{end_time1 - start_time1}秒')
end_time1 = time.time()
print(f'处理一份数据时间：{end_time1 - start_time11}秒')
print(f'接收1张图时间：{end_time1 - start_time1}秒')
# print(cmd, img.shape)
# #打印img的数据类型
# print(img.dtype)
@ -127,18 +130,18 @@ def main(is_debug=False):
continue
spec = None
if cmd == 'PF':
# start_time2 = time.time()
start_time2 = time.time()
spec_data = pipe.receive_spec_data(spec_receive)
_, spec = pipe.parse_spec(spec_data)
# end_time2 = time.time()
# print(f'接收光谱数据时间：{end_time2 - start_time2}秒')
end_time2 = time.time()
print(f'接收光谱数据时间：{end_time2 - start_time2}秒')
# print(spec.shape)
# start_time3 = time.time()
start_time3 = time.time()
response = process_data(cmd, images, spec, dp, pipe, detector)
# end_time3 = time.time()
# print(f'处理时间：{end_time3 - start_time3}秒')
end_time3 = time.time()
print(f'处理时间：{end_time3 - start_time3}秒')
end_time = time.time()
# print(f'全流程时间：{end_time - start_time}秒')
print(f'全流程时间：{end_time - start_time}秒')
if response:
logging.info(f'处理成功，响应为: {response}')
else:

14
20240529RGBtest3/qt_test.py
View File

@ -71,6 +71,8 @@ class MainWindow(QMainWindow):
rgb_files = [os.path.join(image_dir, f) for f in os.listdir(image_dir) if f.endswith(('.bmp'))][:5]
spec_files = [os.path.join(image_dir, f) for f in os.listdir(image_dir) if f.endswith('.raw')][:1]
self.send_YR()
for image_path in rgb_files:
img = cv2.imread(image_path, cv2.IMREAD_COLOR)
img = np.asarray(img, dtype=np.uint8)
@ -116,6 +118,18 @@ class MainWindow(QMainWindow):
self.receive_result()
def send_YR(self):
length = 2
length = length.to_bytes(4, byteorder='big')
cmd = 'YR'
data_send = length + cmd.upper().encode('ascii')
try:
win32file.WriteFile(self.rgb_send, data_send)
print("发送预热指令成功")
except Exception as e:
print(f"发送预热指令失败. 错误原因: {e}")
def receive_result(self):
try:
# 读取结果数据