diff --git a/capture_img_ply_txt._getFiel.py b/capture_img_ply_txt._getFiel.py new file mode 100644 index 0000000..b27dd9c --- /dev/null +++ b/capture_img_ply_txt._getFiel.py @@ -0,0 +1,244 @@ +# -- coding: utf-8 -- +import sys +import time +import os +import paramiko +from scp import SCPClient +import threading +from os import abort +from pathlib import Path +from clog import logger, set_logger_file_handler + +HOST = "192.168.100.254" +PORT = 22 +USERNAME = "root" +PASSWORD = "ebaina" + +# CLOUD_COUNTS = [600000,800000,1000000,1500000,2000000] +CLOUD_COUNTS = [1500000] +DEFAULT_COUNT = 1500000 +EXPS = [20*1000] +# EXPS = [100 * 1000] +DEFAULT_EXP = int(3 * 1000) +PARENT_DIR_NAME = "普通测试_位置1平角度_天气_阴天中午" +# PARENT_DIR_NAME = "test_1" +SAVE_DIR_NAME = "" +TEST_CLOUD = False +TEST_EXP = True + + +def ssh_execute_command(host, port, username, password, command): + ret = False + # 创建SSH客户端 + client = paramiko.SSHClient() + client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) + client.connect(host, port, username, password) + + # 执行命令 + stdin, stdout, stderr = client.exec_command(command) + # 实时读取标准输出 + for line in iter(stdout.readline, ""): + print(line, end="") # 实时打印 + sys.stdout.flush() # 刷新缓冲区,确保立即输出到终端 + + # 实时读取标准错误(如果有) + for line in iter(stderr.readline, ""): + print(f"ERROR: {line}", end="") + sys.stdout.flush() + + # 关闭连接 + client.close() + return True + + +def scp_get_file(host, port, username, password, remote_path, local_path): + ret = False + # 创建SSH客户端 + client = paramiko.Transport((host, port)) + client.connect(username=username, password=password) + + # 创建SCP客户端 + scp_client = SCPClient(client) + + # 拉取文件 + try: + scp_client.get(remote_path, local_path) + except Exception as e: + print(e) + return False + + # 关闭连接 + scp_client.close() + client.close() + + return True + + +def scp_get_multi_file(host, port, username, password, remote_path, local_path): + ret = False + # 创建SSH客户端 + client = paramiko.Transport((host, port)) + client.connect(username=username, password=password) + + # 创建SCP客户端 + scp_client = SCPClient(client) + + # 拉取文件 + try: + for rp in remote_path: + scp_client.get(rp, local_path) + except Exception as e: + logger.error(f"拉取文件失败:{e}") + return False + + # 关闭连接 + scp_client.close() + client.close() + + return True + + +def cap(exp=DEFAULT_EXP, cloud_count=DEFAULT_COUNT): + global SAVE_DIR_NAME + logger.debug(f"===================开处理============================") + # 先删除旧数据 + logger.debug("删除旧数据") + ssh_execute_command( + host=HOST, + port=PORT, + username=USERNAME, + password=PASSWORD, + command="cd /root/app/detect-gui/ \n " + + "rm output.jpg \n" + + "rm output.ply \n" + + "rm roi_conners.txt \n" + ) + logger.debug("删除旧数据完成") + + logger.debug(f"曝光时间设置为:{exp}") + logger.debug(f"点云数量设置为:{cloud_count}") + SAVE_DIR_NAME = f"exp_{exp}_cn_{cloud_count}" + logger.debug(f"保存路径为:{SAVE_DIR_NAME}") + + # 如果文件夹不存在则创建 + if not os.path.exists(Path(__file__).parent / "result" / PARENT_DIR_NAME / SAVE_DIR_NAME): + os.makedirs(Path(__file__).parent / "result" / PARENT_DIR_NAME / SAVE_DIR_NAME) + + # ssh调用远程 borad_image_framework_tmp.py 进行拍摄图片,雷达点云数据采集,角点ROI数据生成 + start_time = time.time() + command = "export LD_LIBRARY_PATH=/root/app/ss928_codec/mpp/out/lib:/root/app/detect-gui/vendors/image-framework/linux_arm64:/root/app/detect-gui/vendors/image-framework/linux_arm64/lib:/root/app/detect-gui/vendors/image-framework/linux_arm64/lib/npu:/root/app/detect-gui/vendors/image-framework/linux_arm64/lib/svp_npu:$LD_LIBRARY_PATH \n" + \ + "cd /root/app/detect-gui/ \n" + \ + f"/root/miniconda3/bin/python borad_image_framework_tmp.py --exp={exp} --count={cloud_count}\n" + logger.debug(f"ssh_execute_command 命令:{command}") + ssh_execute_command( + host=HOST, + port=PORT, + username=USERNAME, + password=PASSWORD, + command=command, + ) + end_time = time.time() # 记录结束时间 + execution_time = end_time - start_time # 计算执行时间 + logger.debug(f"borad_image_framework_tmp.py 程序执行时间:{execution_time}秒") + time.sleep(1) # 等待操作系统保存数据完毕 + + + + +def cap_wide(): + logger.debug(f"===================广角开始处理============================") + + # 先删除旧数据 + logger.debug("删除旧数据") + ssh_execute_command( + host=HOST, + port=PORT, + username=USERNAME, + password=PASSWORD, + command="cd /root/app/detect-gui/ \n " + + "rm wide_image_processed.png \n" + + "rm wide_image.png \n" + + "rm wide_image_roi.json \n" + ) + logger.debug("删除旧数据完成") + + start_time = time.time() + command = "export LD_LIBRARY_PATH=/root/app/ss928_codec/mpp/out/lib:/root/app/detect-gui/vendors/image-framework/linux_arm64:/root/app/detect-gui/vendors/image-framework/linux_arm64/lib:/root/app/detect-gui/vendors/image-framework/linux_arm64/lib/npu:/root/app/detect-gui/vendors/image-framework/linux_arm64/lib/svp_npu:$LD_LIBRARY_PATH \n" + \ + "cd /root/app/detect-gui/ \n" + \ + f"/root/miniconda3/bin/python cap_wide_camera.py \n" + logger.debug(f"ssh_execute_command 命令:{command}") + ssh_execute_command( + host=HOST, + port=PORT, + username=USERNAME, + password=PASSWORD, + command=command, + ) + end_time = time.time() # 记录结束时间 + execution_time = end_time - start_time # 计算执行时间 + logger.debug(f"cap_wide_camera.py 程序执行时间:{execution_time}秒") + time.sleep(1) # 等待操作系统保存数据完毕 + + logger.debug("开始保存文件到本地") + start_time2 = time.time() + local_path = Path(__file__).parent / "result" / PARENT_DIR_NAME + logger.debug(f"本地文件保存路径为:{local_path}") + scp_get_multi_file( + host=HOST, + port=PORT, + username=USERNAME, + password=PASSWORD, + remote_path=[ + "/root/app/detect-gui/wide_image_processed.png", + "/root/app/detect-gui/wide_image.png", + "/root/app/detect-gui/wide_image_roi.json", + ], + local_path=local_path + ) + end_time2 = time.time() # 记录结束时间 + execution_time2 = end_time2 - start_time2 + logger.debug("文件到本地结束") + logger.debug(f"scp_get_file获取文件到本地 程序执行时间:{execution_time2}秒") + logger.debug(f"===================广角开处理结束============================") + + +if __name__ == '__main__': + logger.debug("开始保存文件到本地") + start_time2 = time.time() + # 从index.txt 读取索引,就一个数字文字 + index = open(Path(__file__).parent / "index.txt", "r").read() + local_path = Path(__file__).parent / "result_r" / index + logger.debug(f"本地文件保存路径为:{local_path}") + # 如果文件夹不存在则创建 + if not os.path.exists(local_path): + os.makedirs(local_path) + scp_get_multi_file( + host=HOST, + port=PORT, + username=USERNAME, + password=PASSWORD, + remote_path=[ + "/root/app/detect-gui/output.jpg", + "/root/app/detect-gui/output.ply", + ], + local_path=local_path + ) + end_time2 = time.time() # 记录结束时间 + execution_time2 = end_time2 - start_time2 + logger.debug("文件到本地结束") + logger.debug(f"scp_get_file获取文件到本地 程序执行时间:{execution_time2}秒") + logger.debug(f"===================开处理结束============================") + # index加一 + open(Path(__file__).parent / "index.txt", "w").write(str(int(index) + 1)) + # 删除旧数据 + logger.debug("删除旧数据") + ssh_execute_command( + host=HOST, + port=PORT, + username=USERNAME, + password=PASSWORD, + command="cd /root/app/detect-gui/ \n " + + "rm output.jpg \n" + + "rm output.ply \n" + ) + logger.debug("删除旧数据完成") diff --git a/capture_img_ply_txt.py b/capture_img_ply_txt.py index 6c72286..b600c3a 100644 --- a/capture_img_ply_txt.py +++ b/capture_img_ply_txt.py @@ -17,10 +17,10 @@ PASSWORD = "ebaina" # CLOUD_COUNTS = [600000,800000,1000000,1500000,2000000] CLOUD_COUNTS = [1500000] DEFAULT_COUNT = 1500000 -EXPS = [2*1000,3*1000] +EXPS = [20*1000] # EXPS = [100 * 1000] DEFAULT_EXP = int(3 * 1000) -PARENT_DIR_NAME = "拼接测试_位置3_1平角度_天气_晴天大太阳下午_driverV2_om3" +PARENT_DIR_NAME = "普通测试_位置1平角度_天气_阴天中午" # PARENT_DIR_NAME = "test_1" SAVE_DIR_NAME = "" TEST_CLOUD = False diff --git a/index.txt b/index.txt new file mode 100644 index 0000000..9d60796 --- /dev/null +++ b/index.txt @@ -0,0 +1 @@ +11 \ No newline at end of file