fix
This commit is contained in:
leon
parent
5ca747feed
commit
ede4803987
244
capture_img_ply_txt._getFiel.py
Normal file
244
capture_img_ply_txt._getFiel.py
Normal file
@ -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("删除旧数据完成")
|
||||||
@ -17,10 +17,10 @@ PASSWORD = "ebaina"
|
|||||||
# CLOUD_COUNTS = [600000,800000,1000000,1500000,2000000]
|
# CLOUD_COUNTS = [600000,800000,1000000,1500000,2000000]
|
||||||
CLOUD_COUNTS = [1500000]
|
CLOUD_COUNTS = [1500000]
|
||||||
DEFAULT_COUNT = 1500000
|
DEFAULT_COUNT = 1500000
|
||||||
EXPS = [2*1000,3*1000]
|
EXPS = [20*1000]
|
||||||
# EXPS = [100 * 1000]
|
# EXPS = [100 * 1000]
|
||||||
DEFAULT_EXP = int(3 * 1000)
|
DEFAULT_EXP = int(3 * 1000)
|
||||||
PARENT_DIR_NAME = "拼接测试_位置3_1平角度_天气_晴天大太阳下午_driverV2_om3"
|
PARENT_DIR_NAME = "普通测试_位置1平角度_天气_阴天中午"
|
||||||
# PARENT_DIR_NAME = "test_1"
|
# PARENT_DIR_NAME = "test_1"
|
||||||
SAVE_DIR_NAME = ""
|
SAVE_DIR_NAME = ""
|
||||||
TEST_CLOUD = False
|
TEST_CLOUD = False
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user