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 = [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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user