dongchangxi
1 year ago
36 changed files with 712 additions and 3566 deletions
Binary file not shown.
@ -1,125 +0,0 @@ |
|||||||
import os, sys, time, shlex, subprocess, shutil, requests, cv2, numpy as np |
|
||||||
from PIL import Image |
|
||||||
import platform,socket,atexit |
|
||||||
if platform.system() == 'Windows': |
|
||||||
sys.path.append('e:\\libs\\') |
|
||||||
#sys.path.append('libs') |
|
||||||
else: |
|
||||||
sys.path.append('/data/deploy/make3d/make2/libs/') |
|
||||||
import config, libs, libs_db,main_service_db,common |
|
||||||
|
|
||||||
arrPids = [184219,184215,184209,184206,184201,184199,184164,184087,184076,184069,184059,184029,184013,183993,183952,183888,183882,183868,183824,183817,183815,183812,183791,183777,183715,183705,183670,183613,183573,182974,182983,182985,182989,182997,183007,183051,183060,183054,183053,183097,183119,183108,183139,183141,183155,183163,183169,183180,183191,183196,183214,183244,183266,183278,183275,183309,183402,183422,183440,182974,182964,182958,182956,182916,182855,182854,182820,182787,182800,182809,182780,182772,182770,182766,182754,182742,182740,182723,182707,182647,182643,182614,182595,182563,182557,182547,182546,182538,182530,182517,182498,182478,182452,182386,182352,182285,182266,182260,182230,182224] |
|
||||||
|
|
||||||
r12Pids = [184219,184215,184209,184069,184059,183868,183817,183791,183613,183573,182974,182983,182985,182989,182997,183007,183051,183060,183054,183053,183097,183119,183108,183139,183141,183155,183163,183169,183180,183191,183196,183214,183244,183266,183278,183275,183309,183402,183422,183440,182974,182964,182958,182855,182854] |
|
||||||
print(f"处理的数据长度--{len(arrPids)}") |
|
||||||
i=0 |
|
||||||
noPids = '' |
|
||||||
for pid in r12Pids: |
|
||||||
pid = str(pid) |
|
||||||
path = f"D://{pid}/mask" |
|
||||||
#判断是否存在文件夹 |
|
||||||
if not os.path.exists(path): |
|
||||||
continue |
|
||||||
|
|
||||||
#判断文件夹是否为空 |
|
||||||
if len(os.listdir(path)) == 0: |
|
||||||
continue |
|
||||||
|
|
||||||
i+=1 |
|
||||||
|
|
||||||
|
|
||||||
#检查 depth 和 mask 的文件创建时间是否差异比较大 |
|
||||||
fileDepth = f"D://{pid}/depth" |
|
||||||
fileMask = f"D://{pid}/mask" |
|
||||||
|
|
||||||
diffDepthTime = [] |
|
||||||
dfffMaskTime = [] |
|
||||||
|
|
||||||
arrTimeDepth = [] |
|
||||||
arrTimeMask = [] |
|
||||||
for file in os.listdir(fileDepth): |
|
||||||
createTime = os.path.getctime(f"{fileDepth}/{file}") |
|
||||||
updateTime = os.path.getmtime(f"{fileDepth}/{file}") |
|
||||||
# print(updateTime - createTime) |
|
||||||
diffDepthTime.append(updateTime - createTime) |
|
||||||
arrTimeDepth.append(os.path.getctime(f"{fileDepth}/{file}")) |
|
||||||
|
|
||||||
for file in os.listdir(fileMask): |
|
||||||
createTime = os.path.getctime(f"{fileMask}/{file}") |
|
||||||
updateTime = os.path.getmtime(f"{fileMask}/{file}") |
|
||||||
dfffMaskTime.append(updateTime - createTime) |
|
||||||
|
|
||||||
arrTimeMask.append(os.path.getctime(f"{fileMask}/{file}")) |
|
||||||
|
|
||||||
#打印出 两者的 最大 和最小的时间信息 |
|
||||||
print(f"pid: {pid} depth: { (max(arrTimeDepth) - min(arrTimeDepth))/60 }, diffdepth: {max(diffDepthTime)}, mask: { (max(arrTimeMask) - min(arrTimeMask))/60},diffmask: {max(dfffMaskTime)}") |
|
||||||
# print(f"pid: {pid} diffdepth: {max(arrTimeDepth)} diffmask: {max(arrTimeMask)}") |
|
||||||
|
|
||||||
diffDepth = (max(arrTimeDepth) - min(arrTimeDepth))/60 |
|
||||||
diffMask = (max(arrTimeMask) - min(arrTimeMask))/60 |
|
||||||
|
|
||||||
if diffDepth > 5 or diffMask > 5: |
|
||||||
print(f"异常数据 pid: {pid} depth: {diffDepth} mask: {diffMask}") |
|
||||||
break |
|
||||||
|
|
||||||
#复制文件到指定目录 |
|
||||||
maskPath = f"D://{pid}/mask" |
|
||||||
depthPath = f"D://{pid}/depth" |
|
||||||
regPath = f"D://{pid}/reg" |
|
||||||
imageList = f"D://{pid}/{pid}_imageList.txt" |
|
||||||
outFila = f"D://{pid}/reg/{pid}_registration.out" |
|
||||||
|
|
||||||
sourcePath = f"D://model_info/{pid}" |
|
||||||
if os.path.exists(sourcePath): |
|
||||||
shutil.rmtree(sourcePath) |
|
||||||
|
|
||||||
if not os.path.exists(sourcePath): |
|
||||||
os.makedirs(sourcePath) |
|
||||||
|
|
||||||
#复制文件 |
|
||||||
shutil.copytree(maskPath, f"D://model_info/{pid}/mask") |
|
||||||
shutil.copytree(depthPath, f"D://model_info/{pid}/depth") |
|
||||||
shutil.copy(imageList, f"D://model_info/{pid}/{pid}_imageList.txt") |
|
||||||
shutil.copy(outFila, f"D://model_info/{pid}/{pid}_registration.out") |
|
||||||
|
|
||||||
#遍历reg文件夹 |
|
||||||
for file in os.listdir(regPath): |
|
||||||
if file == f"{pid}_registration.out": |
|
||||||
continue |
|
||||||
|
|
||||||
if "_1.jpg" in file: |
|
||||||
sourceTempPath = f"D://model_info/{pid}/reg/photo1" |
|
||||||
if not os.path.exists(sourceTempPath): |
|
||||||
os.makedirs(sourceTempPath) |
|
||||||
shutil.copy(f"{regPath}/{file}", f"{sourceTempPath}/{file}") |
|
||||||
|
|
||||||
if "_8.jpg" in file: |
|
||||||
sourceTempPath = f"D://model_info/{pid}/reg/photo2" |
|
||||||
if not os.path.exists(sourceTempPath): |
|
||||||
os.makedirs(sourceTempPath) |
|
||||||
shutil.copy(f"{regPath}/{file}", f"{sourceTempPath}/{file}") |
|
||||||
|
|
||||||
#shutil.copy(f"{regPath}/{file}", f"D://model_info/{pid}/reg/{file}") |
|
||||||
|
|
||||||
#移除 reg 里的 {pid}_registration.out |
|
||||||
|
|
||||||
|
|
||||||
print(f"处理完成 pid: {pid}") |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# shutil.copytree(f"D://{pid}", f"D://{pid}_bak") |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# cmd = f'{config.rcbin} {config.r2["init"]} -setInstanceName {pid} \ |
|
||||||
# -load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}"'# -quit |
|
||||||
# print(cmd) |
|
||||||
# cmd = shlex.split(cmd) |
|
||||||
# res = subprocess.run(cmd) |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#print(i) |
|
||||||
@ -1,90 +0,0 @@ |
|||||||
146134 |
|
||||||
146134 |
|
||||||
146598 |
|
||||||
183716 |
|
||||||
185381 |
|
||||||
183400 |
|
||||||
187752 |
|
||||||
184871 |
|
||||||
184956 |
|
||||||
185326 |
|
||||||
187651 |
|
||||||
187802 |
|
||||||
187811 |
|
||||||
187800 |
|
||||||
183096 |
|
||||||
187804 |
|
||||||
184956 |
|
||||||
187808 |
|
||||||
187898 |
|
||||||
187865 |
|
||||||
187870 |
|
||||||
187886 |
|
||||||
187912 |
|
||||||
185102 |
|
||||||
185102 |
|
||||||
185102 |
|
||||||
187450 |
|
||||||
185102 |
|
||||||
187989 |
|
||||||
188170 |
|
||||||
188117 |
|
||||||
188128 |
|
||||||
188200 |
|
||||||
188297 |
|
||||||
188310 |
|
||||||
188344 |
|
||||||
185101 |
|
||||||
188425 |
|
||||||
188428 |
|
||||||
188431 |
|
||||||
188560 |
|
||||||
188494 |
|
||||||
188564 |
|
||||||
188757 |
|
||||||
188779 |
|
||||||
188807 |
|
||||||
188865 |
|
||||||
188827 |
|
||||||
188910 |
|
||||||
188907 |
|
||||||
188913 |
|
||||||
188942 |
|
||||||
188964 |
|
||||||
189045 |
|
||||||
189051 |
|
||||||
189203 |
|
||||||
189214 |
|
||||||
189218 |
|
||||||
189228 |
|
||||||
189246 |
|
||||||
189242 |
|
||||||
189373 |
|
||||||
189312 |
|
||||||
189388 |
|
||||||
189913 |
|
||||||
189911 |
|
||||||
189932 |
|
||||||
189946 |
|
||||||
189106 |
|
||||||
186427 |
|
||||||
186544 |
|
||||||
190092 |
|
||||||
190112 |
|
||||||
188681 |
|
||||||
190138 |
|
||||||
188688 |
|
||||||
189172 |
|
||||||
183292 |
|
||||||
189178 |
|
||||||
184093 |
|
||||||
183080 |
|
||||||
189608 |
|
||||||
189625 |
|
||||||
189618 |
|
||||||
189170 |
|
||||||
189623 |
|
||||||
190151 |
|
||||||
183309 |
|
||||||
190214 |
|
||||||
188521 |
|
||||||
@ -1,340 +0,0 @@ |
|||||||
import os, sys, time, shlex, subprocess, shutil, requests, cv2, numpy as np |
|
||||||
from PIL import Image |
|
||||||
import platform,socket,atexit |
|
||||||
if platform.system() == 'Windows': |
|
||||||
sys.path.append('e:\\libs\\') |
|
||||||
#sys.path.append('libs') |
|
||||||
else: |
|
||||||
sys.path.append('/data/deploy/make3d/make2/libs/') |
|
||||||
import config, libs, libs_db,main_service_db,common |
|
||||||
|
|
||||||
def filter_dark_texture_image(pid): |
|
||||||
start_time = time.time() |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 开始检测射灯异常图片...') |
|
||||||
def get_image_v(image): |
|
||||||
# 图片左上角和右上角各取200*200区域，计算V通道均值 |
|
||||||
left_rect = image[0:200, 0:200] |
|
||||||
right_rect = image[0:200, -200:] |
|
||||||
|
|
||||||
left_hsv = cv2.cvtColor(left_rect, cv2.COLOR_BGR2HSV) |
|
||||||
left_v = left_hsv[:, :, 2] |
|
||||||
left_avg_v = np.mean(left_v) |
|
||||||
# print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 图片左上角V通道均值：{left_avg_v}') |
|
||||||
|
|
||||||
right_hsv = cv2.cvtColor(right_rect, cv2.COLOR_BGR2HSV) |
|
||||||
right_v = right_hsv[:, :, 2] |
|
||||||
right_avg_v = np.mean(right_v) |
|
||||||
# print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 图片右上角V通道均值：{right_avg_v}') |
|
||||||
|
|
||||||
return (left_avg_v + right_avg_v) / 2 |
|
||||||
|
|
||||||
v_list = [] |
|
||||||
for filename in os.listdir(os.path.join(config.workdir, pid, 'photo2')): |
|
||||||
if filename.endswith(".jpg"): |
|
||||||
image = cv2.imread(os.path.join(config.workdir, pid, 'photo2', filename)) |
|
||||||
v = get_image_v(image) |
|
||||||
item = {'filename': filename, 'v': v} |
|
||||||
v_list.append(item) |
|
||||||
v_list.sort(key=lambda x: x['v']) |
|
||||||
v_list = v_list[5: -5] |
|
||||||
avg_v = np.mean([item['v'] for item in v_list]) |
|
||||||
for item in v_list: |
|
||||||
if abs(item['v'] - avg_v) > 50: |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 图片{item["filename"]} V通道值{item["v"]}，低于平均值{avg_v}，将不参与贴图') |
|
||||||
libs.set_photo_join_type(config.workdir, pid, 'photo2', item['filename'].split('_')[0], mesh='1', texture='0') |
|
||||||
|
|
||||||
# 复制xmp文件到photo3目录，如果photo3目录存在的话 |
|
||||||
if os.path.exists(os.path.join(config.workdir, pid, 'photo3')): |
|
||||||
shutil.copyfile(os.path.join(config.workdir, pid, 'photo2', item['filename'].replace('jpg', 'xmp')), os.path.join(config.workdir, pid, 'photo3', item['filename'].replace('jpg', 'xmp'))) |
|
||||||
|
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 射灯异常图片检测完成，共费时{libs.diff_time(start_time)}') |
|
||||||
|
|
||||||
def detect_markers(psid, pid): |
|
||||||
def fix_region(): |
|
||||||
region_filename = os.path.join(config.workdir, pid, f'{pid}.rcbox') |
|
||||||
with open(region_filename, 'r') as f: |
|
||||||
lines = f.readlines() |
|
||||||
lines = [line.replace('"NONE" globalCoordinateSystemWkt="NONE" globalCoordinateSystemName="NONE"', '"+proj=geocent +ellps=WGS84 +no_defs" globalCoordinateSystemName="local:1 - Euclidean"') for line in lines] |
|
||||||
|
|
||||||
start_time = time.time() |
|
||||||
add_photo3 = ' ' |
|
||||||
textpicCmd = ' ' |
|
||||||
if os.path.exists(os.path.join(config.workdir, pid, 'photo3')): |
|
||||||
add_photo3 = ' -addFolder "' + os.path.join(config.workdir, pid, 'photo3') + '" ' |
|
||||||
|
|
||||||
textpicCmd ='-selectImage "'+os.path.join(config.workdir,pid,'photo3')+'\*" -enableTexturingAndColoring true' |
|
||||||
textpicCmd ='-selectImage "'+os.path.join(config.workdir,pid,'photo2')+'\*" -enableTexturingAndColoring false' |
|
||||||
else: |
|
||||||
textpicCmd ='-selectImage "'+os.path.join(config.workdir,pid,'photo2')+'\*" -enableTexturingAndColoring true' |
|
||||||
|
|
||||||
cmd = f'{config.rcbin} {config.r2["init"]} -setInstanceName {pid} \ |
|
||||||
-save "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" \ |
|
||||||
-addFolder "{os.path.join(config.workdir, pid, "photo1")}" {config.r["setTextureFalse"]} -align -addFolder "{os.path.join(config.workdir, pid, "photo2")}" \ |
|
||||||
{add_photo3} -align -selectAllImages \ |
|
||||||
-detectMarkers "D:\\make2\\config\\detectMarkers.config.xml" \ |
|
||||||
{libs.get_defineDistances(psid)} -align -align -update {config.r2["setRegion"]} \ |
|
||||||
-exportXMP "D:\\make2\\config\\exportXMP.config.xml" \ |
|
||||||
-exportControlPointsMeasurements "{os.path.join(config.workdir, pid, f"{pid}.controlPoints.csv")}" "D:\\make2\\config\\exportControlPoints.config.xml" \ |
|
||||||
-exportReconstructionRegion "{os.path.join(config.workdir, pid, f"{pid}.rcbox")}" \ |
|
||||||
{textpicCmd} -save "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" -quit' |
|
||||||
print(cmd) |
|
||||||
cmd = shlex.split(cmd) |
|
||||||
res = subprocess.run(cmd) |
|
||||||
time.sleep(3) |
|
||||||
fix_region() |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 定位点检测完成, 共费时{libs.diff_time(start_time)}') |
|
||||||
|
|
||||||
if os.path.exists(os.path.join(config.workdir, pid, 'photo3')): |
|
||||||
for filename in os.listdir(os.path.join(config.workdir, pid, 'photo2')): |
|
||||||
if filename.endswith('_8.xmp'): |
|
||||||
# photo3 exist, 设置photo2的xmp文件，不参与贴图 |
|
||||||
libs.set_photo_join_type(config.workdir, pid, 'photo2', filename.split('_')[0], mesh='1', texture='0') |
|
||||||
for filename in os.listdir(os.path.join(config.workdir, pid, 'photo3')): |
|
||||||
if filename.endswith('_8.xmp'): |
|
||||||
# photo3 exist, 设置photo3的xmp文件，不参与建模 |
|
||||||
libs.set_photo_join_type(config.workdir, pid, 'photo3', filename.split('_')[0], mesh='0', texture='1') |
|
||||||
|
|
||||||
def step1(pid, experience=False, makeloop=True,task_distributed_id="",isNoColorTexture=""): |
|
||||||
libs_db.start_task({"task_type": "make", "task_key": pid}) |
|
||||||
|
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 开始处理{pid}建模任务') |
|
||||||
psid = libs.getPSid(pid) |
|
||||||
|
|
||||||
# 更新云端任务状态 |
|
||||||
res = requests.post(config.urls['update_status_modeling_url'], data={'id': pid}) |
|
||||||
print('更新建模中状态：', res.text) |
|
||||||
|
|
||||||
# 下载图片 |
|
||||||
start_time = time.time() |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} 开始下载图片...') |
|
||||||
libs.down_from_oss(config.oss_bucket, config.workdir, pid) |
|
||||||
try: |
|
||||||
if str(psid) == "86": |
|
||||||
#移除指定文件 |
|
||||||
os.remove(f"D:\{pid}\photo1\\21_1.jpg") |
|
||||||
os.remove(f"D:\{pid}\photo2\\21_8.jpg") |
|
||||||
if str(psid) == "29": |
|
||||||
#移除指定文件 |
|
||||||
os.remove(f"D:\{pid}\photo1\\15_1.jpg") |
|
||||||
os.remove(f"D:\{pid}\photo2\\15_8.jpg") |
|
||||||
except Exception as e: |
|
||||||
print("移除异常") |
|
||||||
|
|
||||||
#旋转图片 |
|
||||||
os.system(f'python D:\\make2\\photo2rotate.py -w d:\ -p {pid} -i photo1 -o photo1') |
|
||||||
os.system(f'python D:\\make2\\photo2rotate.py -w d:\ -p {pid} -i photo2 -o photo2') |
|
||||||
|
|
||||||
os.system(f'python D:\\make2\\export_build_info\\gen_xmps_for_build_info.py {pid}') |
|
||||||
#print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} 图片下载完成，共费时{libs.diff_time(start_time)}') |
|
||||||
os.system(f'python main_step2.py {pid}') |
|
||||||
# start_time = time.time() |
|
||||||
# # TODO: 上报图片采集数量，更新影棚相机状态 |
|
||||||
|
|
||||||
# # 处理图片，如果experience=True，将图片缩减一半，已节省建模时间和算力成本 |
|
||||||
# # if experience: |
|
||||||
# # libs.resize_photos(os.path.join(config.workdir, pid, 'photo1')) |
|
||||||
# # libs.resize_photos(os.path.join(config.workdir, pid, 'photo2')) |
|
||||||
|
|
||||||
# # 根据配置调整photo2曝光，均衡贴图亮度，可能产生photo3 |
|
||||||
# libs.adjust_photos(config.workdir, pid) |
|
||||||
|
|
||||||
# # TODO: 检测模糊异常图片，上报微信通知 |
|
||||||
# # TODO: 处理图片，去反光算法 |
|
||||||
|
|
||||||
# # 检测图片定位点，定位异常及时上报微信通知给客服人员，人工判断是否需要重新拍摄或更换地贴。定义定位点距离，导出相机位姿信息 |
|
||||||
# detect_markers(psid, pid) |
|
||||||
|
|
||||||
# # 处理图片，检测photo2中的异常图片不参与贴图，以免破坏贴图效果，默认不检测射灯异常图片，以节省算力成本 |
|
||||||
# if not makeloop: |
|
||||||
# filter_dark_texture_image(pid) |
|
||||||
|
|
||||||
# # 处理图片，暗部提亮，提高贴图效果 |
|
||||||
|
|
||||||
# # TODO: 处理图片，去遮挡处理，提高建模与贴图质量 |
|
||||||
|
|
||||||
# # 加入photo2，计算重建区域，导出重建区域与相机位姿配置信息 |
|
||||||
# # cal_reconstruction_region(psid, pid) |
|
||||||
|
|
||||||
# print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} step1图片预处理完成，共费时{libs.diff_time(start_time)}') |
|
||||||
|
|
||||||
# if os.path.exists(os.path.join(config.sharedir, pid)): |
|
||||||
# shutil.rmtree(os.path.join(config.sharedir, pid), ignore_errors=True) |
|
||||||
# shutil.copytree(os.path.join(config.workdir, pid), os.path.join(config.sharedir, pid)) |
|
||||||
# print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} step1任务完成，移动到共享目录') |
|
||||||
|
|
||||||
|
|
||||||
# #指定photo2某些图片不参与贴图 |
|
||||||
# #是否不参与贴图，true 不参与贴图 |
|
||||||
# if isNoColorTexture == "NoColorTexture": |
|
||||||
# arrNoTextureColorPics = config.noTextureColorPics |
|
||||||
# if arrNoTextureColorPics is not None and len(arrNoTextureColorPics) > 0: |
|
||||||
# for camcerIndex in arrNoTextureColorPics: |
|
||||||
# #拼装坐标文件名称 |
|
||||||
# filename = os.path.join(config.sharedir, pid,"photo2",str(camcerIndex)+"_8.xmp") |
|
||||||
# #判断文件是否存在 |
|
||||||
# if os.path.exists(filename): |
|
||||||
# #存在就设置不参与贴图 |
|
||||||
# libs.set_photo_join_type(config.workdir, pid, 'photo2', camcerIndex, mesh='0', texture='0') |
|
||||||
|
|
||||||
|
|
||||||
# #移除当前文件夹 |
|
||||||
# shutil.rmtree(os.path.join(config.workdir, pid)) |
|
||||||
# # TODO: 更新本地step1任务状态，加入step2任务队列 |
|
||||||
# if task_distributed_id == "":#不是分布式任务的时候就自动往下个步骤走，是分布式任务的时候就就执行当前任务 |
|
||||||
# if makeloop: |
|
||||||
# os.system(f'python main_step2.py {pid}') |
|
||||||
# else: |
|
||||||
# os.system(f'python main_step2.py {pid}') |
|
||||||
# # if os.path.exists(os.path.join(config.sharedir, pid)): |
|
||||||
# # shutil.rmtree(os.path.join(config.sharedir, pid), ignore_errors=True) |
|
||||||
# # shutil.move(os.path.join(config.workdir, pid), config.sharedir) |
|
||||||
|
|
||||||
# # print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} step1任务完成，移动到共享目录') |
|
||||||
# else: |
|
||||||
# #分布式服务执行完后,需要更新任务状态,更新字表的finished_at字段 |
|
||||||
# main_service_db.update_task_distributed_detail({"task_distributed_id":task_distributed_id,"finished_at":time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}) |
|
||||||
# print("step1完成,休息6s") |
|
||||||
# time.sleep(4) |
|
||||||
# return |
|
||||||
|
|
||||||
|
|
||||||
def main(pid, experience=False, makeloop=True,task_distributed_id="",isNoColorTexture=""): |
|
||||||
if pid == '0': |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 开始进入本地任务值守模式...') |
|
||||||
while True: |
|
||||||
# 取云端redis多个key任务，TODO:后续要改为api调用 |
|
||||||
experience = False |
|
||||||
pid = libs_db.get_task('make_experience') |
|
||||||
if pid == '': |
|
||||||
time.sleep(3) |
|
||||||
pid = libs_db.get_task('make') |
|
||||||
if pid == '': |
|
||||||
time.sleep(3) |
|
||||||
continue |
|
||||||
else: |
|
||||||
experience = False |
|
||||||
step1(pid, experience, makeloop,task_distributed_id,isNoColorTexture) |
|
||||||
else: |
|
||||||
step1(pid, experience, makeloop,task_distributed_id,isNoColorTexture) |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def is_already(pid): |
|
||||||
|
|
||||||
|
|
||||||
#读取已经完成的pid |
|
||||||
with open("D://make2/export_build_info/finished.txt", 'r') as f: |
|
||||||
lines = f.readlines() |
|
||||||
for line in lines: |
|
||||||
print("line",str(line.strip())) |
|
||||||
if str(pid) == str(line.strip()): |
|
||||||
print("匹配成功") |
|
||||||
return True |
|
||||||
return False |
|
||||||
|
|
||||||
if __name__ == '__main__': |
|
||||||
isNoColorTexture = "" |
|
||||||
if len(sys.argv) == 2: |
|
||||||
v = sys.argv[1] |
|
||||||
|
|
||||||
#判断是否是数字 |
|
||||||
if v.isdigit(): |
|
||||||
#print(f"正在处理输入的pid-{v}") |
|
||||||
main(str(v), experience=False, makeloop=False,task_distributed_id="",isNoColorTexture=isNoColorTexture) |
|
||||||
else: |
|
||||||
filePath = f"D://make2/export_build_info/pids/{v}.txt" |
|
||||||
if not os.path.exists(filePath): |
|
||||||
print(f"{filePath} 文件不存在") |
|
||||||
|
|
||||||
strPids = "" |
|
||||||
with open("D://make2/pid.text","r") as f: |
|
||||||
lines = f.readlines() |
|
||||||
for line in lines: |
|
||||||
strPids = str(line.strip()) |
|
||||||
|
|
||||||
errPids = "" |
|
||||||
with open("D://make2/errPid.text","r") as f: |
|
||||||
lines = f.readlines() |
|
||||||
for line in lines: |
|
||||||
errPids = str(line.strip()) |
|
||||||
|
|
||||||
arrErrPids = errPids.split(",") |
|
||||||
|
|
||||||
arrPids = strPids.split(",") |
|
||||||
ii = 0 |
|
||||||
with open(filePath, 'r') as f: |
|
||||||
lines = f.readlines() |
|
||||||
for line in lines: |
|
||||||
pid = str(line.strip()) |
|
||||||
if pid == "": |
|
||||||
break |
|
||||||
|
|
||||||
#判断是否存在了目录或者已经是处理过了 |
|
||||||
# tempFile = f"D://{pid}//finish.txt" |
|
||||||
# if os.path.exists(tempFile): |
|
||||||
# continue |
|
||||||
|
|
||||||
if str(pid) in arrErrPids: |
|
||||||
continue |
|
||||||
|
|
||||||
if str(pid) in arrPids: |
|
||||||
# print(f"已处理过{pid}") |
|
||||||
ii += 1 |
|
||||||
continue |
|
||||||
|
|
||||||
print(f"正在处理{pid}- 完成{ii}") |
|
||||||
main(str(pid), experience=False, makeloop=False,task_distributed_id="",isNoColorTexture=isNoColorTexture) |
|
||||||
ii += 1 |
|
||||||
print(f"已经完成-{ii}") |
|
||||||
|
|
||||||
|
|
||||||
else: |
|
||||||
print("cmd like python main_step1.py pid/单人/多人/双人/动物/人物") |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# # 取云端redis任务，完成第一步的数据预处理后，将数据放入共享存储目录，将第二步任务塞入本地mysql队列 |
|
||||||
# # 默认循环值守，可传参数运行单一任务，以方便调试 |
|
||||||
# #atexit.register(common.notify,socket.gethostname()+"建模任务已经停止") |
|
||||||
# # pid = '0' |
|
||||||
# isNoColorTexture = "" |
|
||||||
# # if len(sys.argv) == 2: |
|
||||||
# # pids = sys.argv[1].split(',') |
|
||||||
# # for pid in pids: |
|
||||||
# already = [184219,184215,184209,184069,184059,183868,183817,183791,183613,183054,182855,182854,184206,184201,184199,184164,184087,184076,184029,184013,183993,183952,183888,183882,183824,183815,183812] |
|
||||||
|
|
||||||
# arrPids = [184219,184215,184209,184206,184201,184199,184164,184087,184076,184069,184059,184029,184013,183993,183952,183888,183882,183868,183824,183817,183815,183812,183791,183777,183715,183705,183670,183613,183573,182974,182983,182985,182989,182997,183007,183051,183060,183054,183053,183097,183119,183108,183139,183141,183155,183163,183169,183180,183191,183196,183214,183244,183266,183278,183275,183309,183402,183422,183440,182974,182964,182958] |
|
||||||
# #,182956,182916,182855,182854,182820,182787,182800,182809,182780,182772,182770,182766,182754,182742,182740,182723,182707,182647,182643,182614,182595,182563,182557,182547,182546,182538,182530,182517,182498,182478,182452,182386,182352,182285,182266,182260,182230,182224 |
|
||||||
# i = 0 |
|
||||||
# for pid in arrPids: |
|
||||||
# if pid in already: |
|
||||||
# continue |
|
||||||
# print("正在处理-",pid) |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# if is_already(str(pid)) == True: |
|
||||||
# continue |
|
||||||
# # i += 1 |
|
||||||
# # continue |
|
||||||
# #写入到 |
|
||||||
# with open("D://make2/export_build_info/finished.txt", 'a+') as f: |
|
||||||
# f.write(f"{pid}\n") |
|
||||||
|
|
||||||
# main(str(pid), experience=False, makeloop=False,task_distributed_id="",isNoColorTexture=isNoColorTexture) |
|
||||||
|
|
||||||
# print("已处理-",i) |
|
||||||
# #exit() |
|
||||||
# # if len(sys.argv) == 3: |
|
||||||
# # experience = False |
|
||||||
# # if sys.argv[2] == '1': |
|
||||||
# # print('演示测试...') |
|
||||||
# # experience = False |
|
||||||
# # elif sys.argv[2] == 'NoColorTexture': |
|
||||||
# # isNoColorTexture = "NoColorTexture" |
|
||||||
# # pids = sys.argv[1].split(',') |
|
||||||
# # for pid in pids: |
|
||||||
# # main(pid, experience=experience, makeloop=False,task_distributed_id="",isNoColorTexture=isNoColorTexture) |
|
||||||
# # exit() |
|
||||||
# # main(pid, experience=False, makeloop=True,task_distributed_id="",isNoColorTexture=isNoColorTexture) |
|
||||||
@ -1,267 +0,0 @@ |
|||||||
import os, sys, time, shutil, subprocess, shlex |
|
||||||
import platform |
|
||||||
import pyautogui as ag |
|
||||||
if platform.system() == 'Windows': |
|
||||||
sys.path.append('e:\\libs\\') |
|
||||||
#sys.path.append('libs') |
|
||||||
else: |
|
||||||
sys.path.append('/data/deploy/make3d/make2/libs/') |
|
||||||
import config, libs, libs_db,common,main_service_db,computerRecboxCenterByPoint |
|
||||||
import export_already_build |
|
||||||
redisLocal = config.redis_local |
|
||||||
def load_model(pid): |
|
||||||
cmd = f'{config.rcbin} {config.r1["init"]} -load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}"' |
|
||||||
print(cmd) |
|
||||||
cmd = shlex.split(cmd) |
|
||||||
res = subprocess.run(cmd) |
|
||||||
|
|
||||||
def get_rcver(): |
|
||||||
rcbin = '"C:\\Program Files\\Capturing Reality\\RealityCapture\\RealityCapture.exe"' |
|
||||||
if os.path.getsize(rcbin[1:-1]) == 20783616: |
|
||||||
return 1 |
|
||||||
else: |
|
||||||
return 2 |
|
||||||
|
|
||||||
def make3d(pid): |
|
||||||
if get_rcver() == 1: # old version |
|
||||||
#修改重建区域的大小 |
|
||||||
common.change_rcbox_s(pid,"1") |
|
||||||
#获取影棚id |
|
||||||
# psid = libs.getPSid(pid) |
|
||||||
# if int(psid) == 80: |
|
||||||
# change_rcbox_deepth(str(pid),0.03) |
|
||||||
psid = libs.getPSid(pid) |
|
||||||
simplify_value = 1000000 * libs.getHeadCount(pid) |
|
||||||
add_photo3 = ' ' |
|
||||||
#判断是否存在photo3 |
|
||||||
if os.path.exists(os.path.join(config.workdir, pid, 'photo3')): |
|
||||||
add_photo3 = ' -addFolder "' + os.path.join(config.workdir, pid, 'photo3') + '" -align -align ' |
|
||||||
#存在photo3 的情况下，photo2 不参与贴图 |
|
||||||
#遍历获取photo2 目录下的所有文件 |
|
||||||
directory = os.path.join(config.workdir, pid, 'photo2') |
|
||||||
for root, dirs, files in os.walk(directory): |
|
||||||
for file in files: |
|
||||||
#判断file 是否以 _8.xmp 结尾 |
|
||||||
if file.endswith("_8.xmp") == False: |
|
||||||
continue |
|
||||||
#修改文件内容不参与贴图 |
|
||||||
libs.set_photo_join_type(config.workdir, pid, 'photo2', file.split("_")[0], mesh='0', texture='0') |
|
||||||
else: |
|
||||||
pass |
|
||||||
#设置photo2参与贴图，因为有的时候贴图会黑 |
|
||||||
directory = os.path.join(config.workdir, pid, 'photo2') |
|
||||||
for root, dirs, files in os.walk(directory): |
|
||||||
for file in files: |
|
||||||
#判断file 是否以 _8.xmp 结尾 |
|
||||||
if file.endswith("_8.xmp") == False: |
|
||||||
continue |
|
||||||
#修改文件内容不参与贴图 |
|
||||||
libs.set_photo_join_type(config.workdir, pid, 'photo2', file.split("_")[0], mesh='0', texture='1') |
|
||||||
|
|
||||||
if get_rcver() == 1: # old version |
|
||||||
|
|
||||||
#判断oss 上是否存在 controlpoints 文件 |
|
||||||
isExistPoint = common.isExistControlPointsOss(pid) |
|
||||||
cmd = f'{config.rcbin} {config.r1["init"]} \ |
|
||||||
-addFolder "{os.path.join(config.workdir, pid, "photo1")}" -addFolder "{os.path.join(config.workdir, pid, "photo2")}" {add_photo3} \ |
|
||||||
-importControlPointsMeasurements "{os.path.join(config.workdir, pid, f"{pid}.controlPoints.csv")}" \ |
|
||||||
-align -exportRegistration "{os.path.join(config.workdir,"make2", "config","exportRegistration.xml")}" "{os.path.join(config.workdir, pid,str(pid)+"_align.csv")}" -save' |
|
||||||
|
|
||||||
cmd = cmd + f' "{os.path.join(config.workdir, pid, f"{pid}_wait.rcproj")}"' |
|
||||||
print(cmd) |
|
||||||
cmd = shlex.split(cmd) |
|
||||||
res = subprocess.run(cmd) |
|
||||||
time.sleep(2) |
|
||||||
|
|
||||||
|
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 定位点导入完成') |
|
||||||
time.sleep(3) |
|
||||||
# defind_distance |
|
||||||
#死循环阻塞获取 |
|
||||||
while True: |
|
||||||
print("循环阻塞开始") |
|
||||||
time.sleep(3) |
|
||||||
#判断是否有 pid_1 的的值 |
|
||||||
print(pid+"_1") |
|
||||||
if redisLocal.lpos('model:auto_distance',pid+"_1") == None: |
|
||||||
continue |
|
||||||
shutil.move(os.path.join(config.workdir, pid, f'{pid}_wait.rcproj'), os.path.join(config.workdir, pid, f'{pid}.rcproj')) |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 定义定位点距离完成') |
|
||||||
#将 controlpoints_0.dat 文件拷贝到 oss 上作为公共的使用 |
|
||||||
common.uploadControlPointsOss(pid) |
|
||||||
|
|
||||||
#最后处理掉redis中的值 |
|
||||||
redisLocal.lrem('model:auto_distance', 0, pid+"_1") |
|
||||||
break |
|
||||||
print(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),"循环阻塞结束,判断是否对齐成功") |
|
||||||
|
|
||||||
|
|
||||||
#查看对齐的数量 |
|
||||||
alignRes = common.isAlignNums(str(pid)) |
|
||||||
if alignRes == "rebuild": |
|
||||||
os.system(f'python d:\\make2\\tools\push_cmd.py rebuild {pid}') |
|
||||||
os.system(f'python d:\\make2\\main_service.py') |
|
||||||
return |
|
||||||
#区域的设置 建模 |
|
||||||
#update |
|
||||||
cmdSmall = "-align" |
|
||||||
#使用公共point文件的时候，就不能使用update ， 而是要用 align |
|
||||||
if isExistPoint == True: |
|
||||||
cmdSmall = "-align" |
|
||||||
#{config.r1["init"]} |
|
||||||
|
|
||||||
if common.task_need_high_model(pid): |
|
||||||
if str(psid) == "41" or str(psid) == "85": |
|
||||||
calulate_type = "-mvs" |
|
||||||
else: |
|
||||||
calulate_type = '-mvsHigh' |
|
||||||
else: |
|
||||||
calulate_type = '-mvs' |
|
||||||
|
|
||||||
#翻转相机的线向下 |
|
||||||
cmd = f'{config.rcbin} -load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" {cmdSmall} \ |
|
||||||
-set "sfmEnableCameraPrior=True" -align -set "sfmEnableCameraPrior=False" -align -save "{os.path.join(config.workdir, pid, f"{pid}_point.rcproj")}" ' |
|
||||||
cmd = shlex.split(cmd) |
|
||||||
res = subprocess.run(cmd) |
|
||||||
#执行完成后命令会触发导出点云的定时程序 |
|
||||||
while True: |
|
||||||
print("判断是否导出点云文件循环阻塞开始") |
|
||||||
time.sleep(3) |
|
||||||
#判断是否有 pid_1 的的值 |
|
||||||
print(pid+"_3") |
|
||||||
if redisLocal.lpos('model:auto_distance',pid+"_3") == None: |
|
||||||
continue |
|
||||||
shutil.move(os.path.join(config.workdir, pid, f'{pid}_point.rcproj'), os.path.join(config.workdir, pid, f'{pid}.rcproj')) |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 导出点云文件完成') |
|
||||||
#最后处理掉redis中的值 |
|
||||||
redisLocal.lrem('model:auto_distance', 0, pid+"_3") |
|
||||||
break |
|
||||||
|
|
||||||
#处理点云文件 |
|
||||||
centerRcboxValue = computerRecboxCenterByPoint.boxCenter(pid) |
|
||||||
if centerRcboxValue > 0: |
|
||||||
print("修改中心点的坐标",centerRcboxValue) |
|
||||||
#修改中心点的值 |
|
||||||
common.change_rcbox_center(pid,centerRcboxValue) |
|
||||||
|
|
||||||
cmd = f'{config.rcbin} -load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" -align \ |
|
||||||
-set "sfmEnableCameraPrior=True" -align -set "sfmEnableCameraPrior=False" -align -setReconstructionRegion "{os.path.join(config.workdir, pid, f"{pid}.rcbox")}" \ |
|
||||||
{calulate_type} -modelSelectMaximalConnectedComponent -modelInvertSelection -modelRemoveSelectedTriangles -closeHoles -clean -simplify {simplify_value} -smooth -unwrap -calculateTexture -renameModel {pid} -save "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" -quit' |
|
||||||
print(cmd) |
|
||||||
cmd = shlex.split(cmd) |
|
||||||
res = subprocess.run(cmd) |
|
||||||
#保存在导出 |
|
||||||
# common.saveScreenImg(pid) |
|
||||||
# time.sleep(1) |
|
||||||
# ag.hotkey('alt', 'f4') |
|
||||||
time.sleep(3) |
|
||||||
#修改rcproj文件 |
|
||||||
flag = common.changeRcprojFile(pid) |
|
||||||
if flag == False: |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} rcproj文件不存在') |
|
||||||
return |
|
||||||
|
|
||||||
#复制文件 |
|
||||||
shutil.copy(os.path.join(config.workdir, pid, f'{pid}.rcproj'), os.path.join(config.workdir, pid, f'{pid}_screen.rcproj')) |
|
||||||
time.sleep(2) |
|
||||||
#打开工程文件 |
|
||||||
cmd = f'{config.rcbin} -load "{os.path.join(config.workdir, pid, f"{pid}_screen.rcproj")}"' |
|
||||||
cmd = shlex.split(cmd) |
|
||||||
res = subprocess.run(cmd) |
|
||||||
|
|
||||||
#创建指定文件夹 |
|
||||||
if not os.path.exists(os.path.join(config.workdir, pid, "output")): |
|
||||||
os.makedirs(os.path.join(config.workdir, pid, "output")) |
|
||||||
|
|
||||||
#判断output目录下是否有文件，有的话删除 |
|
||||||
files = os.listdir(os.path.join(config.workdir, pid, "output")) |
|
||||||
|
|
||||||
#执行导出 |
|
||||||
cmd = f'{config.rcbin} -load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}"\ |
|
||||||
-exportModel "{pid}" "{os.path.join(config.workdir, pid, "output", f"{pid}.obj")}" "d:\\make2\\config\\ModelExportParams102.xml" -quit' |
|
||||||
print(cmd) |
|
||||||
cmd = shlex.split(cmd) |
|
||||||
res = subprocess.run(cmd) |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
else: # new version |
|
||||||
#判断是否要进行高精模 |
|
||||||
if common.task_need_high_model(pid): |
|
||||||
if str(psid) == "41" or str(psid) == "85": |
|
||||||
calulate_type = "calculateNormalModel" |
|
||||||
else: |
|
||||||
calulate_type = 'calculateHighModel' |
|
||||||
else: |
|
||||||
calulate_type = 'calculateNormalModel' |
|
||||||
|
|
||||||
#创建指定文件夹 |
|
||||||
if not os.path.exists(os.path.join(config.workdir, pid, "output")): |
|
||||||
os.makedirs(os.path.join(config.workdir, pid, "output")) |
|
||||||
|
|
||||||
cmd = f'{config.rcbin} {config.r2["init"]} -setInstanceName {pid} \ |
|
||||||
-load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" \ |
|
||||||
-{calulate_type} \ |
|
||||||
-selectLargestModelComponent -invertTrianglesSelection -removeSelectedTriangles -simplify {simplify_value} -smooth -closeHoles -cleanModel -calculateTexture \ |
|
||||||
-save "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" \ |
|
||||||
-exportSelectedModel "{os.path.join(config.workdir, pid, "output", f"{pid}.obj")}" "d:\\make2\\config\\ModelExportParams.xml" -quit'# -quit |
|
||||||
print(cmd) |
|
||||||
cmd = shlex.split(cmd) |
|
||||||
res = subprocess.run(cmd) |
|
||||||
|
|
||||||
export_already_build.exportInfo(pid) |
|
||||||
# #阻塞判断是否导出完成 |
|
||||||
# while True: |
|
||||||
# #判断 output 目录下是否存在 三个文件 |
|
||||||
# files = os.listdir(os.path.join(config.workdir, pid, "output")) |
|
||||||
# if len(files) >= 3: |
|
||||||
# break |
|
||||||
|
|
||||||
def step2(pid,task_distributed_id=""): |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 开始建模任务step2') |
|
||||||
|
|
||||||
#判断是否要从共享目录拷贝数据 |
|
||||||
if os.path.exists(os.path.join(config.sharedir, pid)) and not os.path.exists(os.path.join(config.workdir, pid)): |
|
||||||
shutil.move(os.path.join(config.sharedir, pid), config.workdir) |
|
||||||
else: |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 目录{os.path.join(config.sharedir, pid)}不存在，或{os.path.join(config.workdir, pid)}已存在') |
|
||||||
# return |
|
||||||
#最后还是要判断有没有存在目录 |
|
||||||
if not os.path.exists(os.path.join(config.workdir, pid)): |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 目录{os.path.join(config.workdir, pid)}不存在') |
|
||||||
return |
|
||||||
|
|
||||||
start_time = time.time() |
|
||||||
make3d(pid) |
|
||||||
|
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 建模任务step2完成，共费时{libs.diff_time(start_time)}，任务已提交到step3') |
|
||||||
# # 更新本地任务状态，加入step3任务队列 |
|
||||||
# if task_distributed_id == "": |
|
||||||
# os.system(f'python d:\\make2\\main_step3.py {pid}') |
|
||||||
# else: |
|
||||||
# #暂时 step2 step3 一起连续执行 |
|
||||||
# print('step2 执行完,开始执行step3') |
|
||||||
# os.system(f'python d:\\make2\\main_step3.py {pid}') |
|
||||||
# main_service_db.update_task_distributed({"id":task_distributed_id,"status":2,"finished_at":time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}) |
|
||||||
# main_service_db.update_task_distributed_detail({"task_distributed_id":task_distributed_id,"finished_at":time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}) |
|
||||||
# #return |
|
||||||
|
|
||||||
def main(pid): |
|
||||||
if pid == '0': |
|
||||||
while True: |
|
||||||
# 取本地mysql队列任务，完成第二步的建模任务 |
|
||||||
|
|
||||||
step2(pid) |
|
||||||
else: |
|
||||||
step2(pid) |
|
||||||
|
|
||||||
if __name__ == '__main__': |
|
||||||
# 取本地mysql队列任务，完成第二步的建模任务 |
|
||||||
# 默认循环值守，可传参数运行单一任务，以方便调试 |
|
||||||
pid = '0' |
|
||||||
if len(sys.argv) > 1: |
|
||||||
pids = sys.argv[1].split(',') |
|
||||||
for pid in pids: |
|
||||||
main(pid) |
|
||||||
exit() |
|
||||||
main(pid) |
|
||||||
@ -1,185 +0,0 @@ |
|||||||
import os, sys, time, shlex, subprocess, shutil, requests, cv2, numpy as np |
|
||||||
from PIL import Image |
|
||||||
import platform |
|
||||||
if platform.system() == 'Windows': |
|
||||||
sys.path.append('e:\\libs\\') |
|
||||||
#sys.path.append('libs') |
|
||||||
else: |
|
||||||
sys.path.append('/data/deploy/make3d/make2/libs/') |
|
||||||
import config, libs, libs_db,main_service_db |
|
||||||
# 2. 手动操作建模做成建模 |
|
||||||
# 2.0 检测是否存在项目，不存在就下载，存在就清除项目其它无用的文件 |
|
||||||
# 2.1 初始化工程， 根据参数 加入 photo1 或者 photo 2 的 照片， 对齐 ， 导出坐标， 将坐标复制到 另外一个文件夹, 测距， 重建区域， 然后调用step2 |
|
||||||
|
|
||||||
#根据pid 检测是否存在项目，不存在就下载，存在就清除项目其它无用的文件 |
|
||||||
def check_pid_file(pid): |
|
||||||
#检测是否存在目录 |
|
||||||
path = os.path.join(config.workdir, pid) |
|
||||||
if not os.path.exists(path): |
|
||||||
#不存在就在就下载目录 |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 开始计算相机位姿...') |
|
||||||
start_time = time.time() |
|
||||||
libs.down_from_oss(config.oss_bucket, config.workdir, pid) |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} 图片下载完成，共费时{libs.diff_time(start_time)}') |
|
||||||
else: |
|
||||||
#存在的话就检测是否有photo1 和 photo2 之外的目录或者文件，有的话就删除 |
|
||||||
for file in os.listdir(path): |
|
||||||
if file != 'photo1' and file != 'photo2': |
|
||||||
if os.path.isfile(os.path.join(path, file)): |
|
||||||
os.remove(os.path.join(path, file)) |
|
||||||
else: |
|
||||||
shutil.rmtree(os.path.join(path, file)) |
|
||||||
|
|
||||||
#判断photo1 和 photo2 目录里的文件是否存在xmp 文件，存在的话就删除 |
|
||||||
for file in os.listdir(os.path.join(path, 'photo1')): |
|
||||||
if file.endswith('.xmp'): |
|
||||||
os.remove(os.path.join(path, 'photo1', file)) |
|
||||||
|
|
||||||
for file in os.listdir(os.path.join(path, 'photo2')): |
|
||||||
if file.endswith('.xmp'): |
|
||||||
os.remove(os.path.join(path, 'photo2', file)) |
|
||||||
|
|
||||||
|
|
||||||
#根据参数初始化操作 |
|
||||||
def cmd_run(pid,usePhoto = "1",lock=False): |
|
||||||
pid = str(pid) |
|
||||||
#检测文件并且下载处理文件 |
|
||||||
check_pid_file(pid) |
|
||||||
|
|
||||||
start_time = time.time() |
|
||||||
#文件路径 |
|
||||||
photo1_path = os.path.join(config.workdir, pid, 'photo1') |
|
||||||
photo2_path = os.path.join(config.workdir, pid, 'photo2') |
|
||||||
#计算文件里的数量 |
|
||||||
photos1_count = len(os.listdir(photo1_path)) |
|
||||||
photos2_count = len(os.listdir(photo2_path)) |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} photo1数量{photos1_count} photo2数量{photos2_count}') |
|
||||||
|
|
||||||
#xmp 坐标是要用 lock的 还是 unlock 的 |
|
||||||
if lock: |
|
||||||
exportxmp = ' -exportXMP "D:\\make2\\config\\exportXMP.config.lock.xml" ' |
|
||||||
else: |
|
||||||
exportxmp = ' -exportXMP "D:\\make2\\config\\exportXMP.config.xml" ' |
|
||||||
|
|
||||||
usePhoto = "photo"+str(usePhoto) |
|
||||||
|
|
||||||
#执行命令 |
|
||||||
cmd = f'{config.rcbin} {config.r2["init"]} -setInstanceName "{pid}" \ |
|
||||||
-save "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" \ |
|
||||||
-addFolder "{os.path.join(config.workdir, pid, usePhoto)}" -selectAllImages \ |
|
||||||
-detectMarkers "D:\\make2\\config\\detectMarkers.config.xml" \ |
|
||||||
-align -align -align -align \ |
|
||||||
{exportxmp} \ |
|
||||||
-save "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" -quit' |
|
||||||
print(cmd) |
|
||||||
cmd = shlex.split(cmd) |
|
||||||
res = subprocess.run(cmd) |
|
||||||
|
|
||||||
#根据参数转变路劲，复制photo1 里的xmp 文件到 photo2 里，或者 photo2 里的xmp 文件到 photo1 里 |
|
||||||
sourceFile = photo1_path |
|
||||||
targetFile = photo2_path |
|
||||||
if usePhoto == "photo2": |
|
||||||
sourceFile = photo2_path |
|
||||||
targetFile = photo1_path |
|
||||||
#复制xmp文件 |
|
||||||
for xmp in os.listdir(sourceFile): |
|
||||||
if xmp.endswith('.xmp'): |
|
||||||
if usePhoto == "photo1": |
|
||||||
shutil.copy(os.path.join(sourceFile, xmp), os.path.join(targetFile,xmp.replace('_1.xmp', '_8.xmp'))) |
|
||||||
|
|
||||||
if usePhoto == "photo2": |
|
||||||
shutil.copy(os.path.join(sourceFile, xmp), os.path.join(targetFile,xmp.replace('_8.xmp', '_1.xmp'))) |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#如果是photo2的话，就要将photo2 的 xmp 重命名成 _8.xmp |
|
||||||
# if usePhoto == "photo2": |
|
||||||
# for xmp in os.listdir(sourceFile): |
|
||||||
# if xmp.endswith('.xmp'): |
|
||||||
# #重名名.xmp 结尾的文件 |
|
||||||
# os.rename(os.path.join(sourceFile, xmp), os.path.join(sourceFile,xmp.replace('_1.xmp', '_8.xmp'))) |
|
||||||
# print("坐标复制完成") |
|
||||||
# exit() |
|
||||||
|
|
||||||
#将两组图片进行重新对齐 然后重建区域 |
|
||||||
psid = libs.getPSid(pid) |
|
||||||
cmd = f'{config.rcbin} -setInstanceName {pid} \ |
|
||||||
-load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" {config.r["setTextureFalse"]} \ |
|
||||||
-addFolder "{targetFile}" -selectAllImages \ |
|
||||||
-detectMarkers "D:\\make2\\config\\detectMarkers.config.xml" \ |
|
||||||
{libs.get_defineDistances(psid)} -update -align -align -align -align {config.r2["setRegion"]} \ |
|
||||||
{exportxmp} \ |
|
||||||
-exportReconstructionRegion "{os.path.join(config.workdir, pid, f"{pid}.rcbox")}" \ |
|
||||||
-selectImage "{os.path.join(config.workdir,pid,"photo2")}/*" -enableTexturingAndColoring true \ |
|
||||||
-save "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" -quit' |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
print(cmd) |
|
||||||
cmd = shlex.split(cmd) |
|
||||||
res = subprocess.run(cmd) |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} step1完成，共费时{libs.diff_time(start_time)}') |
|
||||||
|
|
||||||
#调用step2 |
|
||||||
time.sleep(2) |
|
||||||
# os.system(f'python main_step2.py {pid}') |
|
||||||
os.system(f'python D://make2/export_build_info/main_step2.py {pid}') |
|
||||||
|
|
||||||
|
|
||||||
def is_already(pid): |
|
||||||
|
|
||||||
|
|
||||||
#读取已经完成的pid |
|
||||||
with open("D://make2/export_build_info/finished.txt", 'r') as f: |
|
||||||
lines = f.readlines() |
|
||||||
for line in lines: |
|
||||||
print("line",str(line.strip())) |
|
||||||
if str(pid) == str(line.strip()): |
|
||||||
print("匹配成功") |
|
||||||
return True |
|
||||||
return False |
|
||||||
|
|
||||||
if __name__ == '__main__': |
|
||||||
already = [184219,184215,184209,184069,184059,183868,183817,183791,183613,183054,182855,182854,184206,184201,184199,184164,184087,184076,184029,184013,183993,183952,183888,183882,183824,183815,183812] |
|
||||||
|
|
||||||
arrPids = [184219,184215,184209,184206,184201,184199,184164,184087,184076,184069,184059,184029,184013,183993,183952,183888,183882,183868,183824,183817,183815,183812,183791,183777,183715,183705,183670,183613,183573,182974,182983,182985,182989,182997,183007,183051,183060,183054,183053,183097,183119,183108,183139,183141,183155,183163,183169,183180,183191,183196,183214,183244,183266,183278,183275,183309,183402,183422,183440,182974,182964,182958,182956,182916,182855,182854,182820,182787,182800,182809,182780,182772,182770,182766,182754,182742,182740,182723,182707,182647,182643,182614,182595,182563,182557,182547,182546,182538,182530,182517,182498,182478,182452,182386,182352,182285,182266,182260,182230,182224] |
|
||||||
for pid in arrPids: |
|
||||||
if pid in already: |
|
||||||
continue |
|
||||||
print("正在处理-",pid) |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if is_already(str(pid)) == True: |
|
||||||
continue |
|
||||||
# i += 1 |
|
||||||
# continue |
|
||||||
#写入到 |
|
||||||
with open("D://make2/export_build_info/finished.txt", 'a+') as f: |
|
||||||
f.write(f"{pid}\n") |
|
||||||
|
|
||||||
|
|
||||||
cmd_run(str(pid),usePhoto = "1",lock=False) |
|
||||||
|
|
||||||
# if len(sys.argv) == 2: |
|
||||||
# pids = sys.argv[1].split(',') |
|
||||||
# for pid in pids: |
|
||||||
# cmd_run(pid,usePhoto = "1",lock=False) |
|
||||||
# elif len(sys.argv) == 3: |
|
||||||
# pids = sys.argv[1].split(',') |
|
||||||
# for pid in pids: |
|
||||||
# if sys.argv[2] == '2': |
|
||||||
# cmd_run(pid,usePhoto = "2",lock=False) |
|
||||||
# else: |
|
||||||
# cmd_run(pid,usePhoto = "1",lock=False) |
|
||||||
# elif len(sys.argv) == 4: |
|
||||||
# pids = sys.argv[1].split(',') |
|
||||||
# usePhoto = sys.argv[2] |
|
||||||
# lock = sys.argv[3] |
|
||||||
# for pid in pids: |
|
||||||
# cmd_run(pid,usePhoto = usePhoto,lock=lock) |
|
||||||
|
|
||||||
# else: |
|
||||||
# print(f'useage: python {sys.argv[0]} pid1,pid2,pid3 photo = 1/2 lock = True/False') |
|
||||||
# sys.exit(1) |
|
||||||
@ -1,27 +0,0 @@ |
|||||||
181561 |
|
||||||
181559 |
|
||||||
180990 |
|
||||||
180772 |
|
||||||
180538 |
|
||||||
180256 |
|
||||||
179927 |
|
||||||
179657 |
|
||||||
179627 |
|
||||||
178838 |
|
||||||
178714 |
|
||||||
178616 |
|
||||||
178604 |
|
||||||
178332 |
|
||||||
177920 |
|
||||||
176655 |
|
||||||
184102 |
|
||||||
183728 |
|
||||||
183538 |
|
||||||
183524 |
|
||||||
183274 |
|
||||||
183316 |
|
||||||
183345 |
|
||||||
182867 |
|
||||||
182658 |
|
||||||
182638 |
|
||||||
182560 |
|
||||||
@ -1,28 +0,0 @@ |
|||||||
181307 |
|
||||||
181287 |
|
||||||
181267 |
|
||||||
180402 |
|
||||||
179617 |
|
||||||
179613 |
|
||||||
178991 |
|
||||||
178892 |
|
||||||
178814 |
|
||||||
178787 |
|
||||||
178471 |
|
||||||
178469 |
|
||||||
178400 |
|
||||||
178156 |
|
||||||
177899 |
|
||||||
177894 |
|
||||||
176236 |
|
||||||
184133 |
|
||||||
184039 |
|
||||||
183848 |
|
||||||
183769 |
|
||||||
183637 |
|
||||||
183127 |
|
||||||
183238 |
|
||||||
182450 |
|
||||||
182031 |
|
||||||
181977 |
|
||||||
181960 |
|
||||||
@ -1,393 +0,0 @@ |
|||||||
181763 |
|
||||||
181757 |
|
||||||
181754 |
|
||||||
181687 |
|
||||||
181652 |
|
||||||
181642 |
|
||||||
181619 |
|
||||||
181597 |
|
||||||
181574 |
|
||||||
181542 |
|
||||||
181526 |
|
||||||
181515 |
|
||||||
181482 |
|
||||||
181435 |
|
||||||
181421 |
|
||||||
181408 |
|
||||||
181391 |
|
||||||
181388 |
|
||||||
181367 |
|
||||||
181333 |
|
||||||
181329 |
|
||||||
181327 |
|
||||||
181304 |
|
||||||
181274 |
|
||||||
181261 |
|
||||||
181257 |
|
||||||
181248 |
|
||||||
181232 |
|
||||||
181231 |
|
||||||
181229 |
|
||||||
181205 |
|
||||||
181204 |
|
||||||
181203 |
|
||||||
181202 |
|
||||||
181191 |
|
||||||
181188 |
|
||||||
181177 |
|
||||||
181176 |
|
||||||
181145 |
|
||||||
181143 |
|
||||||
181139 |
|
||||||
181117 |
|
||||||
181115 |
|
||||||
181108 |
|
||||||
181097 |
|
||||||
181062 |
|
||||||
181050 |
|
||||||
181046 |
|
||||||
181040 |
|
||||||
181039 |
|
||||||
181023 |
|
||||||
181020 |
|
||||||
180961 |
|
||||||
180958 |
|
||||||
180944 |
|
||||||
180943 |
|
||||||
180904 |
|
||||||
180903 |
|
||||||
180869 |
|
||||||
180864 |
|
||||||
180862 |
|
||||||
180846 |
|
||||||
180839 |
|
||||||
180819 |
|
||||||
180781 |
|
||||||
180776 |
|
||||||
180771 |
|
||||||
180770 |
|
||||||
180670 |
|
||||||
180661 |
|
||||||
180648 |
|
||||||
180593 |
|
||||||
180591 |
|
||||||
180588 |
|
||||||
180581 |
|
||||||
180461 |
|
||||||
180455 |
|
||||||
180450 |
|
||||||
180429 |
|
||||||
180424 |
|
||||||
180394 |
|
||||||
180327 |
|
||||||
180314 |
|
||||||
180308 |
|
||||||
180297 |
|
||||||
180292 |
|
||||||
180291 |
|
||||||
180281 |
|
||||||
180273 |
|
||||||
180225 |
|
||||||
180185 |
|
||||||
180167 |
|
||||||
180156 |
|
||||||
180145 |
|
||||||
180125 |
|
||||||
180085 |
|
||||||
180084 |
|
||||||
180082 |
|
||||||
179980 |
|
||||||
179978 |
|
||||||
179975 |
|
||||||
179942 |
|
||||||
179935 |
|
||||||
179895 |
|
||||||
179861 |
|
||||||
179850 |
|
||||||
179849 |
|
||||||
179815 |
|
||||||
179722 |
|
||||||
179704 |
|
||||||
179695 |
|
||||||
179688 |
|
||||||
179676 |
|
||||||
179652 |
|
||||||
179646 |
|
||||||
179623 |
|
||||||
179621 |
|
||||||
179585 |
|
||||||
179555 |
|
||||||
179551 |
|
||||||
179493 |
|
||||||
179489 |
|
||||||
179474 |
|
||||||
179471 |
|
||||||
179470 |
|
||||||
179461 |
|
||||||
179457 |
|
||||||
179451 |
|
||||||
179430 |
|
||||||
179355 |
|
||||||
179344 |
|
||||||
179331 |
|
||||||
179316 |
|
||||||
179282 |
|
||||||
179229 |
|
||||||
179226 |
|
||||||
179219 |
|
||||||
179205 |
|
||||||
179204 |
|
||||||
179166 |
|
||||||
179160 |
|
||||||
179151 |
|
||||||
179149 |
|
||||||
179147 |
|
||||||
179139 |
|
||||||
178752 |
|
||||||
178743 |
|
||||||
178677 |
|
||||||
178675 |
|
||||||
178674 |
|
||||||
178624 |
|
||||||
178606 |
|
||||||
178594 |
|
||||||
178591 |
|
||||||
178528 |
|
||||||
178526 |
|
||||||
178521 |
|
||||||
178520 |
|
||||||
178519 |
|
||||||
178493 |
|
||||||
178457 |
|
||||||
178452 |
|
||||||
178447 |
|
||||||
178390 |
|
||||||
178388 |
|
||||||
178369 |
|
||||||
178364 |
|
||||||
178349 |
|
||||||
178348 |
|
||||||
178347 |
|
||||||
178330 |
|
||||||
178319 |
|
||||||
178303 |
|
||||||
178302 |
|
||||||
178294 |
|
||||||
178287 |
|
||||||
178269 |
|
||||||
178236 |
|
||||||
178220 |
|
||||||
178219 |
|
||||||
178214 |
|
||||||
178203 |
|
||||||
178191 |
|
||||||
178187 |
|
||||||
178164 |
|
||||||
178159 |
|
||||||
178129 |
|
||||||
178100 |
|
||||||
178098 |
|
||||||
178054 |
|
||||||
178051 |
|
||||||
178048 |
|
||||||
178042 |
|
||||||
178034 |
|
||||||
178016 |
|
||||||
177467 |
|
||||||
177465 |
|
||||||
177453 |
|
||||||
177449 |
|
||||||
177447 |
|
||||||
177443 |
|
||||||
177431 |
|
||||||
177430 |
|
||||||
177412 |
|
||||||
177360 |
|
||||||
177353 |
|
||||||
177350 |
|
||||||
177337 |
|
||||||
177327 |
|
||||||
177309 |
|
||||||
178008 |
|
||||||
177999 |
|
||||||
177998 |
|
||||||
177992 |
|
||||||
177991 |
|
||||||
177984 |
|
||||||
177973 |
|
||||||
177972 |
|
||||||
177971 |
|
||||||
177948 |
|
||||||
177852 |
|
||||||
177830 |
|
||||||
177820 |
|
||||||
177812 |
|
||||||
177763 |
|
||||||
177706 |
|
||||||
177681 |
|
||||||
177659 |
|
||||||
177648 |
|
||||||
177647 |
|
||||||
177641 |
|
||||||
177628 |
|
||||||
177613 |
|
||||||
177611 |
|
||||||
177550 |
|
||||||
177293 |
|
||||||
177288 |
|
||||||
177280 |
|
||||||
177277 |
|
||||||
177265 |
|
||||||
177262 |
|
||||||
177257 |
|
||||||
177225 |
|
||||||
177213 |
|
||||||
177198 |
|
||||||
177172 |
|
||||||
177159 |
|
||||||
177146 |
|
||||||
177152 |
|
||||||
177108 |
|
||||||
177095 |
|
||||||
177092 |
|
||||||
177048 |
|
||||||
177046 |
|
||||||
177035 |
|
||||||
176822 |
|
||||||
176752 |
|
||||||
176739 |
|
||||||
176688 |
|
||||||
176678 |
|
||||||
176631 |
|
||||||
176609 |
|
||||||
176602 |
|
||||||
176525 |
|
||||||
176338 |
|
||||||
176333 |
|
||||||
176314 |
|
||||||
176305 |
|
||||||
176249 |
|
||||||
176238 |
|
||||||
184219 |
|
||||||
184215 |
|
||||||
184209 |
|
||||||
184206 |
|
||||||
184201 |
|
||||||
184199 |
|
||||||
184164 |
|
||||||
184087 |
|
||||||
184076 |
|
||||||
184069 |
|
||||||
184059 |
|
||||||
184029 |
|
||||||
184013 |
|
||||||
183993 |
|
||||||
183952 |
|
||||||
183888 |
|
||||||
183882 |
|
||||||
183868 |
|
||||||
183824 |
|
||||||
183817 |
|
||||||
183815 |
|
||||||
183812 |
|
||||||
183791 |
|
||||||
183777 |
|
||||||
183715 |
|
||||||
183705 |
|
||||||
183670 |
|
||||||
183613 |
|
||||||
183573 |
|
||||||
182974 |
|
||||||
182983 |
|
||||||
182985 |
|
||||||
182989 |
|
||||||
182997 |
|
||||||
183007 |
|
||||||
183051 |
|
||||||
183060 |
|
||||||
183054 |
|
||||||
183053 |
|
||||||
183097 |
|
||||||
183119 |
|
||||||
183108 |
|
||||||
183139 |
|
||||||
183141 |
|
||||||
183155 |
|
||||||
183163 |
|
||||||
183169 |
|
||||||
183180 |
|
||||||
183191 |
|
||||||
183196 |
|
||||||
183214 |
|
||||||
183244 |
|
||||||
183266 |
|
||||||
183278 |
|
||||||
183275 |
|
||||||
183309 |
|
||||||
183402 |
|
||||||
183422 |
|
||||||
183440 |
|
||||||
182974 |
|
||||||
182964 |
|
||||||
182958 |
|
||||||
182956 |
|
||||||
182916 |
|
||||||
182855 |
|
||||||
182854 |
|
||||||
182820 |
|
||||||
182787 |
|
||||||
182800 |
|
||||||
182809 |
|
||||||
182780 |
|
||||||
182772 |
|
||||||
182770 |
|
||||||
182766 |
|
||||||
182754 |
|
||||||
182742 |
|
||||||
182740 |
|
||||||
182723 |
|
||||||
182707 |
|
||||||
182647 |
|
||||||
182643 |
|
||||||
182614 |
|
||||||
182595 |
|
||||||
182563 |
|
||||||
182557 |
|
||||||
182547 |
|
||||||
182546 |
|
||||||
182538 |
|
||||||
182530 |
|
||||||
182517 |
|
||||||
182498 |
|
||||||
182478 |
|
||||||
182452 |
|
||||||
182386 |
|
||||||
182352 |
|
||||||
182285 |
|
||||||
182266 |
|
||||||
182260 |
|
||||||
182230 |
|
||||||
182224 |
|
||||||
182223 |
|
||||||
182191 |
|
||||||
182185 |
|
||||||
182182 |
|
||||||
182179 |
|
||||||
182106 |
|
||||||
182101 |
|
||||||
182072 |
|
||||||
181966 |
|
||||||
181965 |
|
||||||
181937 |
|
||||||
181928 |
|
||||||
181925 |
|
||||||
181924 |
|
||||||
181923 |
|
||||||
181900 |
|
||||||
181895 |
|
||||||
181860 |
|
||||||
181847 |
|
||||||
181836 |
|
||||||
181833 |
|
||||||
181811 |
|
||||||
181809 |
|
||||||
@ -1,307 +0,0 @@ |
|||||||
181593 |
|
||||||
181384 |
|
||||||
181337 |
|
||||||
181318 |
|
||||||
181263 |
|
||||||
181240 |
|
||||||
181211 |
|
||||||
181197 |
|
||||||
181187 |
|
||||||
181140 |
|
||||||
181103 |
|
||||||
181091 |
|
||||||
181054 |
|
||||||
181027 |
|
||||||
180763 |
|
||||||
180690 |
|
||||||
180684 |
|
||||||
180677 |
|
||||||
180599 |
|
||||||
180576 |
|
||||||
180389 |
|
||||||
180342 |
|
||||||
180337 |
|
||||||
180277 |
|
||||||
180265 |
|
||||||
180253 |
|
||||||
180101 |
|
||||||
180083 |
|
||||||
179905 |
|
||||||
179842 |
|
||||||
179811 |
|
||||||
179808 |
|
||||||
179791 |
|
||||||
179786 |
|
||||||
179700 |
|
||||||
179634 |
|
||||||
179593 |
|
||||||
179576 |
|
||||||
179485 |
|
||||||
179459 |
|
||||||
179400 |
|
||||||
179351 |
|
||||||
178995 |
|
||||||
178898 |
|
||||||
178750 |
|
||||||
178746 |
|
||||||
178723 |
|
||||||
178680 |
|
||||||
178673 |
|
||||||
178654 |
|
||||||
178527 |
|
||||||
178482 |
|
||||||
178465 |
|
||||||
178430 |
|
||||||
178413 |
|
||||||
178409 |
|
||||||
178297 |
|
||||||
178286 |
|
||||||
178281 |
|
||||||
178105 |
|
||||||
178062 |
|
||||||
178026 |
|
||||||
178025 |
|
||||||
178019 |
|
||||||
178017 |
|
||||||
178004 |
|
||||||
177851 |
|
||||||
177678 |
|
||||||
177677 |
|
||||||
177646 |
|
||||||
177545 |
|
||||||
177520 |
|
||||||
177300 |
|
||||||
177080 |
|
||||||
176757 |
|
||||||
176743 |
|
||||||
176648 |
|
||||||
176290 |
|
||||||
176280 |
|
||||||
176273 |
|
||||||
176247 |
|
||||||
184082 |
|
||||||
184074 |
|
||||||
184068 |
|
||||||
184061 |
|
||||||
184041 |
|
||||||
183893 |
|
||||||
183890 |
|
||||||
183810 |
|
||||||
183755 |
|
||||||
183680 |
|
||||||
183656 |
|
||||||
183592 |
|
||||||
183591 |
|
||||||
183589 |
|
||||||
183025 |
|
||||||
183086 |
|
||||||
183104 |
|
||||||
183162 |
|
||||||
183444 |
|
||||||
182907 |
|
||||||
182915 |
|
||||||
182871 |
|
||||||
182804 |
|
||||||
182778 |
|
||||||
182718 |
|
||||||
182675 |
|
||||||
182664 |
|
||||||
182625 |
|
||||||
182623 |
|
||||||
182542 |
|
||||||
182510 |
|
||||||
182473 |
|
||||||
182455 |
|
||||||
182388 |
|
||||||
182365 |
|
||||||
182343 |
|
||||||
182316 |
|
||||||
181957 |
|
||||||
181948 |
|
||||||
181941 |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@ -1,28 +0,0 @@ |
|||||||
180891 |
|
||||||
180673 |
|
||||||
180584 |
|
||||||
180468 |
|
||||||
180164 |
|
||||||
180091 |
|
||||||
179992 |
|
||||||
179962 |
|
||||||
178627 |
|
||||||
178416 |
|
||||||
178255 |
|
||||||
178066 |
|
||||||
177983 |
|
||||||
177722 |
|
||||||
183832 |
|
||||||
183677 |
|
||||||
183183 |
|
||||||
183351 |
|
||||||
183361 |
|
||||||
183360 |
|
||||||
182853 |
|
||||||
182715 |
|
||||||
182630 |
|
||||||
182367 |
|
||||||
182253 |
|
||||||
182057 |
|
||||||
181868 |
|
||||||
181856 |
|
||||||
@ -1,13 +0,0 @@ |
|||||||
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple |
|
||||||
pip config set install.trusted-host pypi.tuna.tsinghua.edu.cn |
|
||||||
|
|
||||||
python -m pip install --upgrade pip |
|
||||||
pip install oss2 redis MySQLdb pillow numpy opencv-python bpy tqdm pyautogui psutil pywin32 pymysql |
|
||||||
|
|
||||||
|
|
||||||
config |
|
||||||
set bin="C:\Program Files\Capturing Reality\RealityCapture\RealityCapture.exe" |
|
||||||
%bin% -disableOnlineCommunication -setInstanceName %pid% |
|
||||||
%bin% -disableOnlineCommunication -delegateTo %pid% |
|
||||||
%bin% -set "appCacheLocation=ProjectFolder" |
|
||||||
|
|
||||||
@ -1,13 +0,0 @@ |
|||||||
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple |
|
||||||
pip config set install.trusted-host pypi.tuna.tsinghua.edu.cn |
|
||||||
|
|
||||||
python -m pip install --upgrade pip |
|
||||||
pip install oss2 redis MySQLdb pillow numpy opencv-python bpy tqdm pyautogui psutil pywin32 pymysql |
|
||||||
|
|
||||||
|
|
||||||
config |
|
||||||
set bin="C:\Program Files\Capturing Reality\RealityCapture\RealityCapture.exe" |
|
||||||
%bin% -disableOnlineCommunication -setInstanceName %pid% |
|
||||||
%bin% -disableOnlineCommunication -delegateTo %pid% |
|
||||||
%bin% -set "appCacheLocation=ProjectFolder" |
|
||||||
|
|
||||||
@ -1,101 +0,0 @@ |
|||||||
#读取需要重新贴图的数据 |
|
||||||
import os,sys,time,shlex,subprocess,shutil |
|
||||||
import platform |
|
||||||
if platform.system() == 'Windows': |
|
||||||
sys.path.append('e:\\libs\\') |
|
||||||
#sys.path.append('libs') |
|
||||||
else: |
|
||||||
sys.path.append('/data/deploy/make3d/make2/libs/') |
|
||||||
import libs_db_repeat_texture |
|
||||||
import libs,config |
|
||||||
|
|
||||||
|
|
||||||
#读取 数据库的信息 |
|
||||||
def get_pid(): |
|
||||||
data = libs_db_repeat_texture.get_pid_task() |
|
||||||
if data != None and data != "error": |
|
||||||
return data[0],data[1] |
|
||||||
else: |
|
||||||
return None,None |
|
||||||
|
|
||||||
#开始处理任务 |
|
||||||
def start_task(pid): |
|
||||||
#下载Obj数据 |
|
||||||
print(f"处理{pid}任务") |
|
||||||
print("开始下载{pid}数据") |
|
||||||
# pid = str("175635") |
|
||||||
complateCloudPath = libs.down_obj_from_oss_for_repeat_texture(config.workdir, pid,"auto") |
|
||||||
if complateCloudPath == "error": |
|
||||||
return -1,"error-从oss上下载出问题" |
|
||||||
#下载下来之后重新加载obj 和texture |
|
||||||
print("complateCloudPath",complateCloudPath) |
|
||||||
|
|
||||||
|
|
||||||
outPath = os.path.join(config.workdir, pid, "output") |
|
||||||
if os.path.exists(outPath): |
|
||||||
print("移除文件夹") |
|
||||||
#移除文件夹 |
|
||||||
shutil.rmtree(outPath) |
|
||||||
|
|
||||||
cmd = f'{config.rcbin} {config.r2["init"]} -setInstanceName {pid} \ |
|
||||||
-load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" \ |
|
||||||
-importModel "{os.path.join(complateCloudPath, f"{pid}.obj")}" \ |
|
||||||
-calculateTexture -exportSelectedModel "{os.path.join(config.workdir, pid, "output", f"{pid}.obj")}" "d:\\make2\\config\\ModelExportParams0722.xml" -quit' |
|
||||||
print(cmd) |
|
||||||
cmd = shlex.split(cmd) |
|
||||||
res = subprocess.run(cmd) |
|
||||||
|
|
||||||
#检查out目录文件夹是否有多个 贴图文件。如果是的话只保留最新的一个，并且重命名为初始图像的名称 |
|
||||||
outFilePath = os.path.join(config.workdir, pid, "output") |
|
||||||
arrTempImage = [] |
|
||||||
for file in os.listdir(outFilePath): |
|
||||||
if file.endswith(".jpg") or file.endswith(".png"): |
|
||||||
#字符串替换 |
|
||||||
fileTempName = file.replace(f"{pid}_u0_v0_diffuse","") |
|
||||||
fileTempName = fileTempName.replace(".jpg","") |
|
||||||
fileTempName = fileTempName.replace(".png","") |
|
||||||
if "_" in fileTempName: |
|
||||||
fileTempName = fileTempName.replace("_","") |
|
||||||
arrTempImage.append(fileTempName) |
|
||||||
print(arrTempImage) |
|
||||||
if len(arrTempImage) > 1: |
|
||||||
#获取 arrTempImage 中最大的值 |
|
||||||
arrTempImage.sort() |
|
||||||
print(arrTempImage) |
|
||||||
#获取最大值 |
|
||||||
maxValue = arrTempImage[-1] |
|
||||||
|
|
||||||
for i in arrTempImage: |
|
||||||
if i == "": |
|
||||||
filePath1 = os.path.join(config.workdir, pid, "output",pid+"_u0_v0_diffuse.jpg") |
|
||||||
filePath2 = os.path.join(config.workdir, pid, "output",pid+"_u0_v0_diffuse.png") |
|
||||||
else: |
|
||||||
filePath1 = os.path.join(config.workdir, pid, "output",pid+"_u0_v0_diffuse_"+str(i)+".jpg") |
|
||||||
filePath2 = os.path.join(config.workdir, pid, "output",pid+"_u0_v0_diffuse_"+str(i)+".png") |
|
||||||
|
|
||||||
if i != maxValue: |
|
||||||
if os.path.exists(filePath1): |
|
||||||
os.remove(filePath1) |
|
||||||
if os.path.exists(filePath2): |
|
||||||
os.remove(filePath2) |
|
||||||
else: |
|
||||||
if os.path.exists(filePath1): |
|
||||||
os.rename(filePath1,os.path.join(config.workdir, pid, "output",pid+"_u0_v0_diffuse.jpg")) |
|
||||||
if os.path.exists(filePath2): |
|
||||||
os.rename(filePath2,os.path.join(config.workdir, pid, "output",pid+"_u0_v0_diffuse.jpg")) |
|
||||||
#执行step3 |
|
||||||
os.system(f'python d:\\make2\\main_step3.py {pid}') |
|
||||||
return 3,"success" |
|
||||||
|
|
||||||
if __name__ == '__main__': |
|
||||||
while True: |
|
||||||
id,pid = get_pid() |
|
||||||
print(f"读取的数据{pid}") |
|
||||||
if pid != None: |
|
||||||
#更新状态为2 |
|
||||||
libs_db_repeat_texture.update_status(id,2) |
|
||||||
statusres,res = start_task(str(pid)) |
|
||||||
libs_db_repeat_texture.update_status(id,statusres) |
|
||||||
else: |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}-没有任务') |
|
||||||
time.sleep(10) |
|
||||||
@ -1,68 +0,0 @@ |
|||||||
import os, sys, time, shutil, subprocess, shlex, json |
|
||||||
import xml.etree.ElementTree as ET |
|
||||||
import platform |
|
||||||
if platform.system() == 'Windows': |
|
||||||
sys.path.append('e:\\libs\\') |
|
||||||
#sys.path.append('libs') |
|
||||||
else: |
|
||||||
sys.path.append('/data/deploy/make3d/make2/libs/') |
|
||||||
|
|
||||||
import config, libs, libs_db,common |
|
||||||
|
|
||||||
|
|
||||||
def old_step2(pid): |
|
||||||
cmd = f'"C:\\Program Files\\Capturing Reality\\RealityCapture\\RealityCapture.exe" -disableOnlineCommunication -set \"sfmEnableCameraPrior=False\" -set \"sfmMaxFeaturesPerMpx=20000\" -set \"sfmMaxFeaturesPerImage=200000\" -set \"sfmImagesOverlap=High\" -set \"sfmMaxFeatureReprojectionError=1\" -addFolder "D:\\{pid}\\photo1" -addFolder "D:\\{pid}\\photo2" -importControlPointsMeasurements "D:\\{pid}\\{pid}.controlPoints.csv" -align -save "D:\\{pid}\\{pid}.rcproj" -quit' |
|
||||||
|
|
||||||
cmd = shlex.split(cmd) |
|
||||||
res = subprocess.run(cmd) |
|
||||||
time.sleep(2) |
|
||||||
|
|
||||||
#修改rcproj文件 |
|
||||||
flag = common.changeRcprojControlpointsFile(pid) |
|
||||||
if flag == False: |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} point文件不存在') |
|
||||||
exit() |
|
||||||
|
|
||||||
simplify_value = 1000000 * libs.getHeadCount(pid) |
|
||||||
|
|
||||||
cmd = f'"C:\\Program Files\\Capturing Reality\\RealityCapture\\RealityCapture.exe" -load "D:\\{pid}\\{pid}.rcproj" -set "sfmEnableCameraPrior=True" -align -set "sfmEnableCameraPrior=False" -align -setReconstructionRegion "D:\\{pid}\\{pid}.rcbox" -mvs -modelSelectMaximalConnectedComponent -modelInvertSelection -modelRemoveSelectedTriangles -closeHoles -clean -simplify {simplify_value} -smooth -unwrap -calculateTexture -renameModel {pid} -save "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" -quit' |
|
||||||
|
|
||||||
cmd = shlex.split(cmd) |
|
||||||
res = subprocess.run(cmd) |
|
||||||
|
|
||||||
time.sleep(3) |
|
||||||
#修改rcproj文件 |
|
||||||
flag = common.changeRcprojFile(pid) |
|
||||||
if flag == False: |
|
||||||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} rcproj文件不存在') |
|
||||||
exit() |
|
||||||
|
|
||||||
#创建指定文件夹 |
|
||||||
if not os.path.exists(os.path.join(config.workdir, pid, "output")): |
|
||||||
os.makedirs(os.path.join(config.workdir, pid, "output")) |
|
||||||
#执行导出 |
|
||||||
cmd = f'{config.rcbin} -load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}"\ |
|
||||||
-exportModel "{pid}" "{os.path.join(config.workdir, pid, "output", f"{pid}.obj")}" "d:\\make2\\config\\ModelExportParams102.xml" -quit' |
|
||||||
print(cmd) |
|
||||||
cmd = shlex.split(cmd) |
|
||||||
res = subprocess.run(cmd) |
|
||||||
|
|
||||||
#阻塞判断是否导出完成 |
|
||||||
while True: |
|
||||||
#判断 output 目录下是否存在 三个文件 |
|
||||||
files = os.listdir(os.path.join(config.workdir, pid, "output")) |
|
||||||
if len(files) >= 3: |
|
||||||
break |
|
||||||
#暂时 step2 step3 一起连续执行 |
|
||||||
print('step2 执行完,开始执行step3') |
|
||||||
os.system(f'python d:\\make2\\main_step3.py {pid}') |
|
||||||
|
|
||||||
if __name__ == '__main__': |
|
||||||
pid = '0' |
|
||||||
if len(sys.argv) == 1: |
|
||||||
exit() |
|
||||||
else: |
|
||||||
pids = sys.argv[1].split(',') |
|
||||||
for pid in pids: |
|
||||||
old_step2(str(pid)) |
|
||||||
exit() |
|
||||||
@ -1,499 +0,0 @@ |
|||||||
189002 |
|
||||||
189819 |
|
||||||
190152 |
|
||||||
190141 |
|
||||||
190204 |
|
||||||
190210 |
|
||||||
190221 |
|
||||||
190234 |
|
||||||
190257 |
|
||||||
190264 |
|
||||||
190297 |
|
||||||
190377 |
|
||||||
190238 |
|
||||||
190409 |
|
||||||
190417 |
|
||||||
190144 |
|
||||||
190233 |
|
||||||
189879 |
|
||||||
190227 |
|
||||||
190463 |
|
||||||
190510 |
|
||||||
190531 |
|
||||||
164671 |
|
||||||
190241 |
|
||||||
190547 |
|
||||||
188521 |
|
||||||
190609 |
|
||||||
184467 |
|
||||||
188521 |
|
||||||
189772 |
|
||||||
188521 |
|
||||||
190724 |
|
||||||
190506 |
|
||||||
190509 |
|
||||||
190511 |
|
||||||
190578 |
|
||||||
183895 |
|
||||||
190584 |
|
||||||
190605 |
|
||||||
190611 |
|
||||||
190620 |
|
||||||
190635 |
|
||||||
190683 |
|
||||||
190684 |
|
||||||
190694 |
|
||||||
190697 |
|
||||||
190708 |
|
||||||
190726 |
|
||||||
183494 |
|
||||||
185507 |
|
||||||
188521 |
|
||||||
183494 |
|
||||||
190751 |
|
||||||
183895 |
|
||||||
190754 |
|
||||||
183895 |
|
||||||
189002 |
|
||||||
190784 |
|
||||||
190783 |
|
||||||
190795 |
|
||||||
190812 |
|
||||||
183895 |
|
||||||
190823 |
|
||||||
190837 |
|
||||||
190910 |
|
||||||
190838 |
|
||||||
190866 |
|
||||||
190905 |
|
||||||
190915 |
|
||||||
190927 |
|
||||||
190944 |
|
||||||
190939 |
|
||||||
190932 |
|
||||||
190379 |
|
||||||
190948 |
|
||||||
190959 |
|
||||||
190980 |
|
||||||
190975 |
|
||||||
190927 |
|
||||||
190996 |
|
||||||
190981 |
|
||||||
191001 |
|
||||||
190927 |
|
||||||
190992 |
|
||||||
182361 |
|
||||||
185080 |
|
||||||
191045 |
|
||||||
185416 |
|
||||||
191069 |
|
||||||
191082 |
|
||||||
191081 |
|
||||||
191103 |
|
||||||
191118 |
|
||||||
191116 |
|
||||||
191152 |
|
||||||
191180 |
|
||||||
191125 |
|
||||||
191227 |
|
||||||
191238 |
|
||||||
191246 |
|
||||||
191234 |
|
||||||
191275 |
|
||||||
191320 |
|
||||||
191283 |
|
||||||
191354 |
|
||||||
191388 |
|
||||||
191435 |
|
||||||
191441 |
|
||||||
191431 |
|
||||||
191380 |
|
||||||
191340 |
|
||||||
191350 |
|
||||||
191297 |
|
||||||
191472 |
|
||||||
191255 |
|
||||||
191268 |
|
||||||
191501 |
|
||||||
191192 |
|
||||||
191156 |
|
||||||
191517 |
|
||||||
191515 |
|
||||||
191525 |
|
||||||
191532 |
|
||||||
191261 |
|
||||||
191422 |
|
||||||
191024 |
|
||||||
191542 |
|
||||||
191587 |
|
||||||
191597 |
|
||||||
191676 |
|
||||||
191678 |
|
||||||
191711 |
|
||||||
191736 |
|
||||||
186628 |
|
||||||
191874 |
|
||||||
191937 |
|
||||||
191924 |
|
||||||
191945 |
|
||||||
191887 |
|
||||||
191868 |
|
||||||
191981 |
|
||||||
191885 |
|
||||||
191838 |
|
||||||
191882 |
|
||||||
191965 |
|
||||||
191822 |
|
||||||
191870 |
|
||||||
191920 |
|
||||||
191894 |
|
||||||
191811 |
|
||||||
191739 |
|
||||||
191681 |
|
||||||
191790 |
|
||||||
191738 |
|
||||||
191699 |
|
||||||
191781 |
|
||||||
191733 |
|
||||||
191746 |
|
||||||
191793 |
|
||||||
191829 |
|
||||||
191834 |
|
||||||
191907 |
|
||||||
191927 |
|
||||||
191969 |
|
||||||
191990 |
|
||||||
191998 |
|
||||||
192005 |
|
||||||
192021 |
|
||||||
192023 |
|
||||||
192026 |
|
||||||
191771 |
|
||||||
190829 |
|
||||||
192038 |
|
||||||
191906 |
|
||||||
192095 |
|
||||||
192107 |
|
||||||
192098 |
|
||||||
192123 |
|
||||||
191066 |
|
||||||
192150 |
|
||||||
192161 |
|
||||||
192178 |
|
||||||
191210 |
|
||||||
191849 |
|
||||||
191832 |
|
||||||
192093 |
|
||||||
192199 |
|
||||||
192208 |
|
||||||
192202 |
|
||||||
192213 |
|
||||||
192211 |
|
||||||
192204 |
|
||||||
183783 |
|
||||||
192220 |
|
||||||
192267 |
|
||||||
192223 |
|
||||||
190880 |
|
||||||
192288 |
|
||||||
188877 |
|
||||||
192283 |
|
||||||
192291 |
|
||||||
192079 |
|
||||||
192077 |
|
||||||
192072 |
|
||||||
192063 |
|
||||||
188201 |
|
||||||
192343 |
|
||||||
192353 |
|
||||||
192367 |
|
||||||
192373 |
|
||||||
192222 |
|
||||||
192399 |
|
||||||
192392 |
|
||||||
189979 |
|
||||||
189988 |
|
||||||
189996 |
|
||||||
190692 |
|
||||||
190709 |
|
||||||
191955 |
|
||||||
191912 |
|
||||||
192444 |
|
||||||
192438 |
|
||||||
187727 |
|
||||||
192479 |
|
||||||
192473 |
|
||||||
192469 |
|
||||||
192476 |
|
||||||
192140 |
|
||||||
192143 |
|
||||||
192158 |
|
||||||
192498 |
|
||||||
192169 |
|
||||||
192175 |
|
||||||
192186 |
|
||||||
192187 |
|
||||||
192197 |
|
||||||
192214 |
|
||||||
190362 |
|
||||||
192236 |
|
||||||
192239 |
|
||||||
192241 |
|
||||||
192242 |
|
||||||
192244 |
|
||||||
192248 |
|
||||||
192304 |
|
||||||
192261 |
|
||||||
192421 |
|
||||||
192425 |
|
||||||
192514 |
|
||||||
192162 |
|
||||||
192548 |
|
||||||
192550 |
|
||||||
184779 |
|
||||||
192495 |
|
||||||
192490 |
|
||||||
192583 |
|
||||||
192586 |
|
||||||
192591 |
|
||||||
192593 |
|
||||||
192595 |
|
||||||
184756 |
|
||||||
192235 |
|
||||||
187195 |
|
||||||
192378 |
|
||||||
192622 |
|
||||||
192129 |
|
||||||
192570 |
|
||||||
192566 |
|
||||||
192480 |
|
||||||
192481 |
|
||||||
192484 |
|
||||||
192482 |
|
||||||
192532 |
|
||||||
192624 |
|
||||||
192630 |
|
||||||
192682 |
|
||||||
192674 |
|
||||||
192690 |
|
||||||
192701 |
|
||||||
188185 |
|
||||||
192725 |
|
||||||
192729 |
|
||||||
192608 |
|
||||||
192732 |
|
||||||
192717 |
|
||||||
192609 |
|
||||||
192611 |
|
||||||
192612 |
|
||||||
192615 |
|
||||||
192744 |
|
||||||
192643 |
|
||||||
192650 |
|
||||||
192653 |
|
||||||
192659 |
|
||||||
192776 |
|
||||||
192664 |
|
||||||
192676 |
|
||||||
192680 |
|
||||||
192758 |
|
||||||
192759 |
|
||||||
192767 |
|
||||||
192775 |
|
||||||
192782 |
|
||||||
192783 |
|
||||||
185658 |
|
||||||
192838 |
|
||||||
192873 |
|
||||||
192875 |
|
||||||
192905 |
|
||||||
192895 |
|
||||||
192869 |
|
||||||
187718 |
|
||||||
188167 |
|
||||||
192929 |
|
||||||
192940 |
|
||||||
192955 |
|
||||||
192947 |
|
||||||
192907 |
|
||||||
192908 |
|
||||||
192887 |
|
||||||
192979 |
|
||||||
193038 |
|
||||||
193003 |
|
||||||
193005 |
|
||||||
192875 |
|
||||||
193112 |
|
||||||
193101 |
|
||||||
193085 |
|
||||||
193082 |
|
||||||
192986 |
|
||||||
192944 |
|
||||||
193132 |
|
||||||
193141 |
|
||||||
193058 |
|
||||||
193160 |
|
||||||
193074 |
|
||||||
193217 |
|
||||||
193239 |
|
||||||
192208 |
|
||||||
193285 |
|
||||||
193260 |
|
||||||
193287 |
|
||||||
193288 |
|
||||||
193298 |
|
||||||
193281 |
|
||||||
193335 |
|
||||||
193332 |
|
||||||
193349 |
|
||||||
193165 |
|
||||||
193185 |
|
||||||
193010 |
|
||||||
193231 |
|
||||||
193364 |
|
||||||
193295 |
|
||||||
193313 |
|
||||||
193339 |
|
||||||
193354 |
|
||||||
193357 |
|
||||||
193360 |
|
||||||
193375 |
|
||||||
193376 |
|
||||||
193377 |
|
||||||
193378 |
|
||||||
193353 |
|
||||||
193193 |
|
||||||
193454 |
|
||||||
192317 |
|
||||||
193405 |
|
||||||
193468 |
|
||||||
193475 |
|
||||||
193483 |
|
||||||
193479 |
|
||||||
193486 |
|
||||||
193508 |
|
||||||
193504 |
|
||||||
193511 |
|
||||||
193503 |
|
||||||
193520 |
|
||||||
193562 |
|
||||||
193571 |
|
||||||
193560 |
|
||||||
193237 |
|
||||||
193569 |
|
||||||
193591 |
|
||||||
193590 |
|
||||||
193593 |
|
||||||
193617 |
|
||||||
193615 |
|
||||||
193637 |
|
||||||
193625 |
|
||||||
193642 |
|
||||||
193646 |
|
||||||
193683 |
|
||||||
193671 |
|
||||||
193677 |
|
||||||
193616 |
|
||||||
193702 |
|
||||||
193700 |
|
||||||
193604 |
|
||||||
193721 |
|
||||||
193745 |
|
||||||
193770 |
|
||||||
193768 |
|
||||||
193758 |
|
||||||
193762 |
|
||||||
193636 |
|
||||||
193652 |
|
||||||
190141 |
|
||||||
193771 |
|
||||||
193781 |
|
||||||
193813 |
|
||||||
193806 |
|
||||||
193812 |
|
||||||
193818 |
|
||||||
193835 |
|
||||||
193863 |
|
||||||
193907 |
|
||||||
193832 |
|
||||||
193896 |
|
||||||
193903 |
|
||||||
193906 |
|
||||||
193380 |
|
||||||
193383 |
|
||||||
193384 |
|
||||||
193925 |
|
||||||
193386 |
|
||||||
193480 |
|
||||||
193387 |
|
||||||
193666 |
|
||||||
193711 |
|
||||||
193757 |
|
||||||
193934 |
|
||||||
193778 |
|
||||||
193795 |
|
||||||
193809 |
|
||||||
193822 |
|
||||||
193817 |
|
||||||
193843 |
|
||||||
193918 |
|
||||||
193876 |
|
||||||
193922 |
|
||||||
193567 |
|
||||||
193564 |
|
||||||
193551 |
|
||||||
192320 |
|
||||||
193117 |
|
||||||
193961 |
|
||||||
193969 |
|
||||||
193524 |
|
||||||
193547 |
|
||||||
193990 |
|
||||||
193995 |
|
||||||
194012 |
|
||||||
194022 |
|
||||||
194027 |
|
||||||
194021 |
|
||||||
194069 |
|
||||||
194048 |
|
||||||
194035 |
|
||||||
194070 |
|
||||||
194015 |
|
||||||
194018 |
|
||||||
194093 |
|
||||||
194098 |
|
||||||
194144 |
|
||||||
194104 |
|
||||||
194112 |
|
||||||
194119 |
|
||||||
194108 |
|
||||||
194126 |
|
||||||
194153 |
|
||||||
194163 |
|
||||||
194044 |
|
||||||
194047 |
|
||||||
194206 |
|
||||||
194004 |
|
||||||
194229 |
|
||||||
194218 |
|
||||||
194240 |
|
||||||
194242 |
|
||||||
194278 |
|
||||||
194271 |
|
||||||
194293 |
|
||||||
194081 |
|
||||||
194173 |
|
||||||
194176 |
|
||||||
194211 |
|
||||||
194245 |
|
||||||
194270 |
|
||||||
194285 |
|
||||||
194292 |
|
||||||
194294 |
|
||||||
194296 |
|
||||||
194300 |
|
||||||
194347 |
|
||||||
190480 |
|
||||||
190483 |
|
||||||
194258 |
|
||||||
194266,194276,194444,194419,194454,194471,194470,194473,194475,194480,194493,194497,194499,194480,194460,192716,194465,194514,194515,194527,194501,194500,194526,194531,194536,194540,194551,194564,194577,194582,194580,187324,194599,194601,194628,194625,194623,189048,186982,194627,194508,194510,189048,194640,194642,194654,194685,194686,194707,194720,194719,194744,194734,194634,194738,194760,194765,194764,194768,194774,190697,194566,194356,194365,194369,194806,194376,194064,194772,194097,194773,194100,194103,194812,194810,194223,194181,194178,194589,194588,186685,194831,194830,194832,194834,194854,194817,194878,191283,191321,194900,194904,194858,194901,194885,194916,194924,194925,194802,194927,194932,194906,194908,194911,194912,194914,194915,194917,194920,194953,194926,194928,194930,194935,194957,194817,190531,189819,194989,194987,194995,194996,192305,194998,195015,195013,195023,195018,195021,195032,195005,195050,189864,195067,195072,195091,195026,195031,195104,195029,195102,195051,195063,195093,195125,195146,193800,195160,195172,195154,195184,194210,186610,191005,187854,193794, |
|
||||||
@ -1,5 +0,0 @@ |
|||||||
old version |
|
||||||
|
|
||||||
init: |
|
||||||
|
|
||||||
'"C:\\Program Files\\Capturing Reality\\RealityCapture\\RealityCapture.exe" -disableOnlineCommunication -set \"sfmEnableCameraPrior=False\" -set \"sfmMaxFeaturesPerMpx=20000\" -set \"sfmMaxFeaturesPerImage=200000\" -set \"sfmImagesOverlap=High\" -set \"sfmMaxFeatureReprojectionError=1\" -addFolder "D:\\{pid}\\photo1" -addFolder "D:\\{pid}\\photo2" -importControlPointsMeasurements "D:\\{pid}\\{pid}..controlPoints.csv" -align -save' |
|
||||||
File diff suppressed because it is too large
Load Diff
Binary file not shown.
@ -1,18 +0,0 @@ |
|||||||
import re |
|
||||||
|
|
||||||
def extract_o_option(file_path): |
|
||||||
pattern = re.compile(r'o\s+(\S+)') |
|
||||||
with open(file_path, 'r') as file: |
|
||||||
for line in file: |
|
||||||
match = pattern.search(line) |
|
||||||
if match: |
|
||||||
return match.group(1) |
|
||||||
return None |
|
||||||
|
|
||||||
# 使用示例 |
|
||||||
file_path = 'C:/Users/Administrator/Desktop/29282/29282_9cm_x1.obj' |
|
||||||
result = extract_o_option(file_path) |
|
||||||
if result: |
|
||||||
print(f'Found -o option with value: {result}') |
|
||||||
else: |
|
||||||
print('No -o option found') |
|
||||||
@ -1,21 +0,0 @@ |
|||||||
# -*- coding: utf-8 -*- |
|
||||||
import os |
|
||||||
import oss2 |
|
||||||
|
|
||||||
# 填写Bucket所在地域对应的Endpoint。以华东1（杭州）为例，Endpoint填写为https://oss-cn-hangzhou.aliyuncs.com。 |
|
||||||
endpoint = 'https://oss-cn-shanghai.aliyuncs.com' |
|
||||||
# 阿里云账号AccessKey拥有所有API的访问权限，风险很高。强烈建议您创建并使用RAM用户进行API访问或日常运维，请登录RAM控制台创建RAM用户。 |
|
||||||
access_key_id = 'LTAI5tSj7s2mKZNqamT1x64v' |
|
||||||
access_key_secret = 'iIqeBLB8R2U72pAwJbZZBB6tuB3ZF9' |
|
||||||
# 指定文件所在的Bucket名称，例如examplebucket。 |
|
||||||
bucket_name = 'suwa3d-securedata' |
|
||||||
# Object完整路径，完整路径中不能包含Bucket名称。 |
|
||||||
key = 'photos/141640/photo2/72_8.jpg' |
|
||||||
|
|
||||||
# 指定Bucket实例，所有文件相关的方法都需要通过Bucket实例来调用。 |
|
||||||
bucket = oss2.Bucket(oss2.Auth(access_key_id, access_key_secret), endpoint, bucket_name) |
|
||||||
|
|
||||||
style = 'image/bodies' |
|
||||||
# 生成带签名的URL，并指定过期时间为10分钟。过期时间单位为秒。 |
|
||||||
url = bucket.sign_url('GET', key, 10 * 60, params={'x-oss-process': style}) |
|
||||||
print(url) |
|
||||||
@ -0,0 +1,202 @@ |
|||||||
|
import os, sys, time, bpy, math, requests, bmesh, json, shutil |
||||||
|
from PIL import Image |
||||||
|
import platform,redis |
||||||
|
if platform.system() == 'Windows': |
||||||
|
sys.path.append('e:\\libs\\') |
||||||
|
#sys.path.append('libs') |
||||||
|
else: |
||||||
|
sys.path.append('/data/deploy/make3d/make2/libs/') |
||||||
|
import config, libs, libs_db,main_service_db,common,foot_mark_seam,libs_db_gpu |
||||||
|
|
||||||
|
def bmesh_copy_from_object(obj, transform=True, triangulate=True, apply_modifiers=False): |
||||||
|
"""Returns a transformed, triangulated copy of the mesh""" |
||||||
|
assert obj.type == 'MESH' |
||||||
|
if apply_modifiers and obj.modifiers: |
||||||
|
import bpy |
||||||
|
depsgraph = bpy.context.evaluated_depsgraph_get() |
||||||
|
obj_eval = obj.evaluated_get(depsgraph) |
||||||
|
me = obj_eval.to_mesh() |
||||||
|
bm = bmesh.new() |
||||||
|
bm.from_mesh(me) |
||||||
|
obj_eval.to_mesh_clear() |
||||||
|
else: |
||||||
|
me = obj.data |
||||||
|
if obj.mode == 'EDIT': |
||||||
|
bm_orig = bmesh.from_edit_mesh(me) |
||||||
|
bm = bm_orig.copy() |
||||||
|
else: |
||||||
|
bm = bmesh.new() |
||||||
|
bm.from_mesh(me) |
||||||
|
if transform: |
||||||
|
matrix = obj.matrix_world.copy() |
||||||
|
if not matrix.is_identity: |
||||||
|
bm.transform(matrix) |
||||||
|
matrix.translation.zero() |
||||||
|
if not matrix.is_identity: |
||||||
|
bm.normal_update() |
||||||
|
if triangulate: |
||||||
|
bmesh.ops.triangulate(bm, faces=bm.faces) |
||||||
|
return bm |
||||||
|
|
||||||
|
def find_pid_objname(pid): |
||||||
|
for obj in bpy.data.objects: |
||||||
|
if obj.name.startswith(str(pid)): |
||||||
|
return obj.name |
||||||
|
|
||||||
|
#加载obj文件 |
||||||
|
def reload_obj(pid): |
||||||
|
|
||||||
|
#下载obj文件 |
||||||
|
libs.down_obj_from_oss(config.workdir, pid, "print") |
||||||
|
# obj = reload_obj(pid, action) |
||||||
|
|
||||||
|
|
||||||
|
obj_filename = os.path.join(config.workdir,'print',pid,f'{pid}.obj') |
||||||
|
bpy.ops.wm.read_homefile() |
||||||
|
bpy.ops.object.delete(use_global=False, confirm=False) |
||||||
|
bpy.ops.import_scene.obj(filepath=obj_filename) |
||||||
|
bpy.context.scene.unit_settings.scale_length = 1 |
||||||
|
bpy.context.scene.unit_settings.length_unit = 'CENTIMETERS' |
||||||
|
bpy.context.scene.unit_settings.mass_unit = 'GRAMS' |
||||||
|
|
||||||
|
obj = bpy.context.selected_objects[0] |
||||||
|
bpy.context.view_layer.objects.active = obj |
||||||
|
obj.select_set(True) |
||||||
|
|
||||||
|
def base_fix(pid): |
||||||
|
# 统一blender环境 |
||||||
|
reload_obj(pid) |
||||||
|
# 统一模型方向、位置、大小... |
||||||
|
pid_objname = find_pid_objname(pid) |
||||||
|
obj = bpy.data.objects[pid_objname] |
||||||
|
bpy.data.objects[pid_objname].rotation_euler = (0, 0, 0) |
||||||
|
# 检查当前模型是否正确使用 z 轴作为高度轴 |
||||||
|
if obj.dimensions.z < obj.dimensions.y: |
||||||
|
# 如果 z 轴比 y 轴小，说明高度错误，应该旋转调整模型 |
||||||
|
print(f"警告：模型的高度轴（z）似乎不正确，使用 y 轴作为高度。正在修正模型方向...") |
||||||
|
bpy.ops.object.transform_apply(location=True, rotation=True, scale=True) # 应用当前变换 |
||||||
|
obj.rotation_euler[0] = math.radians(90) # 旋转模型 90 度，交换 z 和 y 轴 |
||||||
|
bpy.ops.object.transform_apply(rotation=True) # 应用旋转 |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
bpy.ops.object.origin_set(type='ORIGIN_CENTER_OF_VOLUME', center='MEDIAN') |
||||||
|
bpy.context.object.location[0] = 0 |
||||||
|
bpy.context.object.location[1] = 0 |
||||||
|
bpy.ops.object.transform_apply(location=True, rotation=True, scale=True) |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 模型基础校正完成') |
||||||
|
|
||||||
|
|
||||||
|
def export_and_update_glbs(pid): |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 开始导出并上传审核模型和3D相册模型glb文件...') |
||||||
|
start_time = time.time() |
||||||
|
#headcount = libs.getHeadCount(pid) |
||||||
|
headcount = 1 |
||||||
|
pid_objname = find_pid_objname(pid) |
||||||
|
obj = bpy.data.objects[pid_objname] |
||||||
|
obj.select_set(True) |
||||||
|
|
||||||
|
model_info = {} |
||||||
|
model_info['headcount'] = headcount |
||||||
|
model_info['faces'] = round(len(obj.data.polygons) / 10000) |
||||||
|
model_info['height'] = round(obj.dimensions.y * 100) |
||||||
|
|
||||||
|
# bpy.ops.wm.save_as_mainfile(filepath=os.path.join(config.workdir, pid, f'{pid}.blend')) |
||||||
|
|
||||||
|
# 统一缩放到9cm标准尺寸 |
||||||
|
scale = 90 / bpy.data.objects[pid_objname].dimensions.y |
||||||
|
bpy.data.objects[pid_objname].scale = (scale, scale, scale) |
||||||
|
bpy.ops.object.transform_apply(scale=True) |
||||||
|
|
||||||
|
# bpy.ops.wm.save_as_mainfile(filepath=os.path.join(config.workdir, pid, f'{pid}-9cm.blend')) |
||||||
|
|
||||||
|
bm = bmesh_copy_from_object(obj) |
||||||
|
model_info['volume'] = round(bm.calc_volume(), 2) |
||||||
|
model_info['weight'] = round(model_info['volume'] * 1.226, 2) |
||||||
|
print(f'{pid}的模型数据：{model_info}') |
||||||
|
|
||||||
|
res = requests.get(f'{config.urls["upload_model_info_url"]}?pid={pid}&headcount={headcount}&faces={model_info["faces"]}&volume={model_info["volume"]}&weight={model_info["weight"]}&height={model_info["height"]}') |
||||||
|
print('上传模型数据：', res.text) |
||||||
|
# with open(os.path.join(config.sharedir, 'model_info', f'{pid}.json'), 'w') as f: |
||||||
|
# json.dump(model_info, f) |
||||||
|
# f.close() |
||||||
|
|
||||||
|
# 先生成审核模型 |
||||||
|
faces_dest = 500000 * headcount |
||||||
|
# 减面 |
||||||
|
faces_current = len(bpy.data.objects[pid_objname].data.polygons) |
||||||
|
print(f'当前面数：{faces_current}，目标面数：{faces_dest}') |
||||||
|
|
||||||
|
bpy.ops.object.modifier_add(type='DECIMATE') |
||||||
|
bpy.context.object.modifiers["Decimate"].ratio = faces_dest / faces_current |
||||||
|
bpy.ops.object.modifier_apply(modifier="Decimate") |
||||||
|
|
||||||
|
# glb_filename = os.path.join(config.workdir, pid, 'output', f'{pid}.glb') |
||||||
|
# bpy.ops.export_scene.gltf(filepath=glb_filename, export_format='GLB', export_apply=True, export_jpeg_quality=75, export_draco_mesh_compression_enable=False) |
||||||
|
|
||||||
|
# 再生成数字模型 |
||||||
|
faces_dest = 120000 * headcount |
||||||
|
|
||||||
|
# 减面 |
||||||
|
faces_current = len(bpy.data.objects[pid_objname].data.polygons) |
||||||
|
print(f'当前面数：{faces_current}，目标面数：{faces_dest}') |
||||||
|
|
||||||
|
bpy.ops.object.modifier_add(type='DECIMATE') |
||||||
|
bpy.context.object.modifiers["Decimate"].ratio = faces_dest / faces_current |
||||||
|
bpy.ops.object.modifier_apply(modifier="Decimate") |
||||||
|
|
||||||
|
glb_filename = os.path.join(config.workdir,'print',pid,f'{pid}-3d.glb') |
||||||
|
bpy.ops.export_scene.gltf(filepath=glb_filename, export_format='GLB', export_apply=True, export_jpeg_quality=75, export_draco_mesh_compression_enable=False) |
||||||
|
|
||||||
|
# os.system(f'gltfpack -c -i {os.path.join(config.workdir, pid, "output", f"{pid}.glb")} -o {os.path.join(config.workdir, pid, "output", f"{pid}-pack.glb")}') |
||||||
|
# config.oss_bucket.put_object_from_file(f'glbs/auto/{pid}.glb', os.path.join(config.workdir, pid, 'output', f'{pid}-pack.glb')) |
||||||
|
config.oss_bucket.put_object_from_file(f'glbs/3d/{pid}.glb', os.path.join(config.workdir,'print',pid, f'{pid}-3d.glb')) |
||||||
|
|
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} glb文件导出并上传完成，共费时{libs.diff_time(start_time)}') |
||||||
|
|
||||||
|
#移除文件夹 |
||||||
|
shutil.rmtree(os.path.join(config.workdir,'print',pid)) |
||||||
|
|
||||||
|
#更新该笔订单的状态为 3000 |
||||||
|
requests.post("https://shop.api.suwa3d.com/api/printOrder/updateExternalOrderStatus", data={'pid': pid}) |
||||||
|
|
||||||
|
def createGlb(pid): |
||||||
|
|
||||||
|
#移除文件夹 |
||||||
|
try: |
||||||
|
shutil.rmtree(os.path.join(config.workdir,'print',pid)) |
||||||
|
except: |
||||||
|
pass |
||||||
|
base_fix(pid) |
||||||
|
export_and_update_glbs(pid) |
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
arrArgs = sys.argv |
||||||
|
if len(arrArgs) == 2: |
||||||
|
pid = arrArgs[1] |
||||||
|
#判断是否是数字 |
||||||
|
if pid.isdigit(): |
||||||
|
print('请输入正确的pid') |
||||||
|
createGlb(str(pid)) |
||||||
|
|
||||||
|
else: |
||||||
|
r = redis.Redis(host="106.14.158.208",password="kcV2000",port=6379,db=6) |
||||||
|
while True: |
||||||
|
if r.llen('model:3dglb') == 0: |
||||||
|
print('队列为空，等待10秒') |
||||||
|
time.sleep(10) |
||||||
|
continue |
||||||
|
|
||||||
|
info = r.lpop('model:3dglb') |
||||||
|
if info is None: |
||||||
|
print('队列为空，等待10秒') |
||||||
|
time.sleep(10) |
||||||
|
pid = info.decode('utf-8') |
||||||
|
|
||||||
|
if not pid.isdigit(): |
||||||
|
continue |
||||||
|
|
||||||
|
createGlb(pid) |
||||||
|
|
||||||
|
|
||||||
@ -1,8 +1,26 @@ |
|||||||
import time,os |
import os |
||||||
|
import time as t |
||||||
|
from datetime import datetime, time |
||||||
|
def is_time_in_range(start_time_str, end_time_str): |
||||||
|
# 获取当前时间 |
||||||
|
now = datetime.now().time() |
||||||
|
|
||||||
|
# 转换字符串时间为时间对象 |
||||||
|
start_time = datetime.strptime(start_time_str, '%H:%M').time() |
||||||
|
end_time = datetime.strptime(end_time_str, '%H:%M').time() |
||||||
|
|
||||||
|
# 检查当前时间是否在指定范围内 |
||||||
|
return start_time <= now <= end_time |
||||||
|
|
||||||
if __name__ == '__main__': |
if __name__ == '__main__': |
||||||
|
start_time_str = '07:00' |
||||||
|
end_time_str = '23:59' |
||||||
#开启死循环 |
#开启死循环 |
||||||
while True: |
while True: |
||||||
|
|
||||||
|
#判断当前时间是否在 8:00-20:00 之间 |
||||||
|
if is_time_in_range(start_time_str, end_time_str): |
||||||
#检测当前打印 和 脚底板 和 gs 队列长度 |
#检测当前打印 和 脚底板 和 gs 队列长度 |
||||||
os.system("python E:\\make2\\tools\\push_cmd.py auto_view") |
os.system("python E:\\make2\\tools\\push_cmd.py auto_view") |
||||||
#一个小时检测一次 |
#一个小时检测一次 |
||||||
time.sleep(60*60) |
t.sleep(60*60) |
||||||
@ -0,0 +1,40 @@ |
|||||||
|
<Configuration> |
||||||
|
<entry key="ModelExportFormatVersion" value="0"/> |
||||||
|
<entry key="MvsMeshExportCamerasAsModelPart" value="false"/> |
||||||
|
<entry key="MvsMeshExportTexturingAllowed" value="-1"/> |
||||||
|
<entry key="MvsExportIsModelCoordinates" value="0"/> |
||||||
|
<entry key="MvsExportIsGeoreferenced" value="0x1"/> |
||||||
|
<entry key="MvsExportScaleZ" value="1.0"/> |
||||||
|
<entry key="MvsMeshExportTileType" value="0"/> |
||||||
|
<entry key="MvsMeshExportNormals" value="false"/> |
||||||
|
<entry key="MvsExportScaleY" value="1.0"/> |
||||||
|
<entry key="MvsMeshExportTexAlpha" value="false"/> |
||||||
|
<entry key="MvsExportScaleX" value="1.0"/> |
||||||
|
<entry key="MvsMeshExportTexImgFormat_Color8_0" value="jpg"/> |
||||||
|
<entry key="MvsExportcoordinatesystemtype" value="0"/> |
||||||
|
<entry key="MvsMeshExportTexPixFormat_Color8_0" value="32bppBGRA"/> |
||||||
|
<entry key="MvsMeshExportNormalsAllowed" value="-1"/> |
||||||
|
<entry key="MvsMeshExportNumberFormatAllowed" value="-1"/> |
||||||
|
<entry key="MvsExportMoveZ" value="0.0"/> |
||||||
|
<entry key="MvsExportMoveX" value="0.0"/> |
||||||
|
<entry key="MvsExportMoveY" value="0.0"/> |
||||||
|
<entry key="MvsExportNormalRange" value="ZeroToOne"/> |
||||||
|
<entry key="MvsMeshExportInfoFile" value="true"/> |
||||||
|
<entry key="MvsMeshExportByParts" value="0"/> |
||||||
|
<entry key="MvsMeshExportClassificationAllowed" value="0"/> |
||||||
|
<entry key="MvsMeshExportNumberFormat" value="5"/> |
||||||
|
<entry key="MvsExportRotationY" value="0.0"/> |
||||||
|
<entry key="MvsExportNormalFlipZ" value="false"/> |
||||||
|
<entry key="MvsExportRotationX" value="0.0"/> |
||||||
|
<entry key="MvsExportNormalFlipY" value="false"/> |
||||||
|
<entry key="MvsMeshExportCamerasAllowed" value="0"/> |
||||||
|
<entry key="MvsMeshExportColors" value="true"/> |
||||||
|
<entry key="MvsExportNormalSpace" value="Mikktspace"/> |
||||||
|
<entry key="MvsExportNormalFlipX" value="false"/> |
||||||
|
<entry key="MvsExportTransformationPreset" value="[[Default]]"/> |
||||||
|
<entry key="MvsExportRotationZ" value="0.0"/> |
||||||
|
<entry key="MvsMeshExportFileTypeSelectionDisplay" value="0"/> |
||||||
|
<entry key="MvsMeshExportTexOneFile" value="1"/> |
||||||
|
<entry key="MvsMeshExportTexturing" value="-1"/> |
||||||
|
<entry key="MvsMeshExportEmbeddTxrsAllowed" value="0"/> |
||||||
|
</Configuration> |
||||||
@ -0,0 +1,17 @@ |
|||||||
|
{ |
||||||
|
"workdir": "D://", |
||||||
|
"local_need_build": "/\\172.31.1.253/modeling/local_need_build", |
||||||
|
"local_complate_build":"/\\172.31.1.253/modeling/local_complate_build", |
||||||
|
"rcbin": "'C:\\Program Files\\Capturing Reality\\RealityCapture\\RealityCapture.exe'", |
||||||
|
"urls": { |
||||||
|
"update_status_modelsuccess_url": "http://172.16.20.13:8383/update_take_photo" |
||||||
|
}, |
||||||
|
"r": { |
||||||
|
"setTextureTrue": "-selectAllImages -enableTexturingAndColoring true", |
||||||
|
"setTextureFalse": "-selectAllImages -enableTexturingAndColoring false" |
||||||
|
}, |
||||||
|
"r2": { |
||||||
|
"init": "-disableOnlineCommunication -setProjectCoordinateSystem Local:1 -setOutputCoordinateSystem epsg:4326 -set \"sfmEnableCameraPrior=False\" -set \"sfmMaxFeaturesPerMpx=20000\" -set \"sfmMaxFeaturesPerImage=200000\" -set \"sfmImagesOverlap=High\" -set \"sfmMaxFeatureReprojectionError=1\"", |
||||||
|
"setRegion": "-setReconstructionRegionOnCPs 36h11:001 36h11:002 36h11:003 2.1 -moveReconstructionRegion 0 0 -2.1 -rotateReconstructionRegion 180 0 180 -setGroundPlaneFromReconstructionRegion -scaleReconstructionRegion 1.8 1.6 2.1 absolute center -moveReconstructionRegion 0 0 0.0025" |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,26 @@ |
|||||||
|
|
||||||
|
workdir = 'D://' #建模过程数据存放的目录 |
||||||
|
local_need_build = '/\\172.31.1.253/modeling/local_need_build' #需要建模的数据存放目录 |
||||||
|
local_complate_build = '/\\172.31.1.253/modeling/local_complate_build' #建模完成的存放目录 |
||||||
|
rcbin = '"C:\\Program Files\\Capturing Reality\\RealityCapture\\RealityCapture.exe"' |
||||||
|
urls = { |
||||||
|
'update_status_modelsuccess_url': 'http://172.16.20.13:8383/update_take_photo', |
||||||
|
} |
||||||
|
|
||||||
|
r = { |
||||||
|
"setTextureTrue" : "-selectAllImages -enableTexturingAndColoring true", |
||||||
|
"setTextureFalse" : "-selectAllImages -enableTexturingAndColoring false", |
||||||
|
} |
||||||
|
|
||||||
|
r1 = { |
||||||
|
"init" : "-disableOnlineCommunication -set \"sfmEnableCameraPrior=False\" -set \"sfmMaxFeaturesPerMpx=20000\" -set \"sfmMaxFeaturesPerImage=200000\" -set \"sfmImagesOverlap=High\" -set \"sfmMaxFeatureReprojectionError=1\"", |
||||||
|
} |
||||||
|
|
||||||
|
r2 = { |
||||||
|
"init" : "-disableOnlineCommunication -setProjectCoordinateSystem Local:1 -setOutputCoordinateSystem epsg:4326 -set \"sfmEnableCameraPrior=False\" -set \"sfmMaxFeaturesPerMpx=20000\" -set \"sfmMaxFeaturesPerImage=200000\" -set \"sfmImagesOverlap=High\" -set \"sfmMaxFeatureReprojectionError=1\"", |
||||||
|
"setRegion" : "-setReconstructionRegionOnCPs 36h11:001 36h11:002 36h11:003 2.1 -moveReconstructionRegion 0 0 -2.1 -rotateReconstructionRegion 180 0 180 -setGroundPlaneFromReconstructionRegion -scaleReconstructionRegion 1.8 1.6 2.1 absolute center -moveReconstructionRegion 0 0 0.0025" |
||||||
|
} |
||||||
|
|
||||||
|
rH = { |
||||||
|
"init" : "-disableOnlineCommunication -set \"sfmEnableCameraPrior=False\" -set \"sfmMaxFeaturesPerMpx=10000\" -set \"sfmMaxFeaturesPerImage=80000\" -set \"sfmImagesOverlap=High\" -set \"sfmMaxFeatureReprojectionError=2\"", |
||||||
|
} |
||||||
@ -0,0 +1,7 @@ |
|||||||
|
<Configuration id="{8D21413B-0848-49A9-BF6E-8EBCCA356BC7}"> |
||||||
|
<entry key="minMarkerMeasurements" value="0x1"/> |
||||||
|
<entry key="generateMarkersPaperSize" value="0"/> |
||||||
|
<entry key="generateMarkersMarkersPerPage" value="0x4"/> |
||||||
|
<entry key="generateMarkersCount" value="0x4"/> |
||||||
|
<entry key="markerType" value="AprilTag36h11"/> |
||||||
|
</Configuration> |
||||||
@ -0,0 +1,6 @@ |
|||||||
|
<Configuration id="{E80BAEB3-7A87-4889-B150-0CF6486D7089}"> |
||||||
|
<entry key="icpsAcc" value="4.0"/> |
||||||
|
<entry key="csvCPMSep" value="2"/> |
||||||
|
<entry key="icpsTmode" value="1"/> |
||||||
|
<entry key="csvCPMIgn" value="false"/> |
||||||
|
</Configuration> |
||||||
@ -0,0 +1,8 @@ |
|||||||
|
<Configuration id="{EC40D990-B2AF-42A4-9637-1208A0FD1322}"> |
||||||
|
<entry key="xmpMerge" value="true"/> |
||||||
|
<entry key="xmpExGps" value="true"/> |
||||||
|
<entry key="xmpFlags" value="true"/> |
||||||
|
<entry key="xmpCalibGroups" value="true"/> |
||||||
|
<entry key="xmpCamera" value="3"/> |
||||||
|
<entry key="xmpRig" value="true"/> |
||||||
|
</Configuration> |
||||||
@ -0,0 +1,139 @@ |
|||||||
|
import os, time, json, requests, shutil, psutil |
||||||
|
from tqdm import tqdm |
||||||
|
from PIL import Image, ImageEnhance |
||||||
|
import config |
||||||
|
from concurrent.futures import ThreadPoolExecutor |
||||||
|
|
||||||
|
def find_blender_bin_path(): |
||||||
|
base_path = 'C:\\Program Files\\Blender Foundation\\' |
||||||
|
if os.path.exists(base_path): |
||||||
|
for dir in os.listdir(base_path): |
||||||
|
if dir.startswith('Blender'): |
||||||
|
blender_bin_path = base_path + dir + '\\blender.exe' |
||||||
|
return f'"{blender_bin_path}"' |
||||||
|
else: |
||||||
|
print('未找到blender安装目录') |
||||||
|
exit(1) |
||||||
|
|
||||||
|
# def resize_photos(photo_path, ratio=0.5): |
||||||
|
# for filename in os.listdir(photo_path): |
||||||
|
# if filename.endswith('.jpg'): |
||||||
|
# img = Image.open(os.path.join(photo_path, filename)) |
||||||
|
# img = rotate_image(img) |
||||||
|
# w, h = img.size |
||||||
|
# img = img.resize((int(w * ratio), int(h * ratio))) |
||||||
|
# img.save(os.path.join(photo_path, filename)) |
||||||
|
|
||||||
|
# def rotate_image(image): |
||||||
|
# # 检查图像的EXIF数据是否包含方向信息 |
||||||
|
# try: |
||||||
|
# exif = image._getexif() |
||||||
|
# orientation = exif.get(0x0112) |
||||||
|
# except: |
||||||
|
# orientation = None |
||||||
|
|
||||||
|
# # 根据方向信息旋转图像 |
||||||
|
# if orientation == 3: |
||||||
|
# image = image.rotate(180, expand=True) |
||||||
|
# elif orientation == 6: |
||||||
|
# image = image.rotate(270, expand=True) |
||||||
|
# elif orientation == 8: |
||||||
|
# image = image.rotate(90, expand=True) |
||||||
|
# return image |
||||||
|
|
||||||
|
# def get_ps_adjust_photo_para(psid): |
||||||
|
# res = requests.get(config.urls['get_ps_adjust_photo_para_url'], params={'id': psid}) |
||||||
|
# print(res.json()) |
||||||
|
# paras = res.json()['data'] |
||||||
|
# brightness_factor, saturation_factor, temperature_factor = float(paras['brightness']), float(paras['saturation']), float(paras['colorTemperature']) |
||||||
|
# return brightness_factor, saturation_factor, temperature_factor |
||||||
|
|
||||||
|
|
||||||
|
# def getPSid(pid): |
||||||
|
# res = requests.get(config.urls['get_psid_url'], params={'pid': pid}) |
||||||
|
# print('get_psid_url:', res.url) |
||||||
|
# print('res:', res.text) |
||||||
|
# res = json.loads(res.text) |
||||||
|
# return str(res['data']) |
||||||
|
|
||||||
|
# def getHeadCount(pid): |
||||||
|
# return 1 |
||||||
|
# res = requests.get(config.urls['get_printinfo_url'], params={'id': pid}) |
||||||
|
# print('get_printinfo_url:', res.url) |
||||||
|
# print('res:', res.text) |
||||||
|
# if res.status_code != 200: |
||||||
|
# print('获取人数失败，程序退出') |
||||||
|
# exit(1) |
||||||
|
# res = json.loads(res.text) |
||||||
|
# return res['data']['headcount'] |
||||||
|
|
||||||
|
# def get_ps_type(pid): |
||||||
|
# # return 1:圆形影棚 2:方形影棚 |
||||||
|
# #res = requests.get(config.urls['get_ps_type_url'], params={'pid': pid}) |
||||||
|
# return 2 |
||||||
|
|
||||||
|
# def find_valid_camera_on_oss(pid): |
||||||
|
# if get_ps_type(pid) == 1: |
||||||
|
# print('当前拍照影棚为：圆形影棚') |
||||||
|
# cameras = (103, 93, 113) |
||||||
|
# else: |
||||||
|
# print('当前拍照影棚为：方形影棚') |
||||||
|
# cameras = (74, 64, 84) |
||||||
|
# find_camera = 0 |
||||||
|
# for camera in cameras: |
||||||
|
# objectkey = f'photos/{pid}/photo2/{camera}_8.jpg' |
||||||
|
# find = config.oss_bucket.object_exists(objectkey) |
||||||
|
# if find: |
||||||
|
# find_camera = camera |
||||||
|
# break |
||||||
|
|
||||||
|
# print('找到有效正脸相机:', find_camera) |
||||||
|
# if find_camera == 0: |
||||||
|
# print('{cameras}没有找到照片，程序退出') |
||||||
|
# exit(1) |
||||||
|
# return find_camera |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# def set_photos_join_type(workdir, pid, photoN, mesh = '0', texture='0'): |
||||||
|
# photoN_path = os.path.join(workdir, pid, photoN) |
||||||
|
# for xmp in os.listdir(photoN_path): |
||||||
|
# if xmp.endswith('.xmp'): |
||||||
|
# xmp_path = os.path.join(photoN_path, xmp) |
||||||
|
# with open(xmp_path, 'r') as f: |
||||||
|
# lines = f.readlines() |
||||||
|
# lines = [line.replace('xcr:InMeshing="0"', f'xcr:InMeshing="{mesh}"') for line in lines] |
||||||
|
# lines = [line.replace('xcr:InMeshing="1"', f'xcr:InMeshing="{mesh}"') for line in lines] |
||||||
|
# lines = [line.replace('xcr:InTexturing="0"', f'xcr:InTexturing="{texture}"') for line in lines] |
||||||
|
# lines = [line.replace('xcr:InTexturing="1"', f'xcr:InTexturing="{texture}"') for line in lines] |
||||||
|
# with open(xmp_path, 'w') as f: |
||||||
|
# f.writelines(lines) |
||||||
|
|
||||||
|
def set_photo_join_type(workdir, pid, photoN, camera_id, mesh = '0', texture='0'): |
||||||
|
if photoN == 'photo1': |
||||||
|
filename = os.path.join(workdir, pid, photoN, f'{camera_id}_1.xmp') |
||||||
|
else: |
||||||
|
filename = os.path.join(workdir, pid, photoN, f'{camera_id}_8.xmp') |
||||||
|
with open(filename, 'r') as f: |
||||||
|
lines = f.readlines() |
||||||
|
lines = [line.replace('xcr:InMeshing="0"', f'xcr:InMeshing="{mesh}"') for line in lines] |
||||||
|
lines = [line.replace('xcr:InMeshing="1"', f'xcr:InMeshing="{mesh}"') for line in lines] |
||||||
|
lines = [line.replace('xcr:InTexturing="0"', f'xcr:InTexturing="{texture}"') for line in lines] |
||||||
|
lines = [line.replace('xcr:InTexturing="1"', f'xcr:InTexturing="{texture}"') for line in lines] |
||||||
|
with open(filename, 'w') as f: |
||||||
|
f.writelines(lines) |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def diff_time(start_time): |
||||||
|
# 按照分:秒的方式返回时间差 |
||||||
|
end_time = time.time() |
||||||
|
diff = end_time - start_time |
||||||
|
m, s = divmod(diff, 60) |
||||||
|
return f'{int(m)}分{int(s)}秒' |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@ -0,0 +1,233 @@ |
|||||||
|
import os, sys, time, shlex, subprocess, shutil, requests, cv2, numpy as np |
||||||
|
from PIL import Image |
||||||
|
import json |
||||||
|
|
||||||
|
|
||||||
|
with open('config.json', 'r') as f: |
||||||
|
config = json.load(f) |
||||||
|
|
||||||
|
|
||||||
|
def set_photo_join_type(workdir, pid, photoN, camera_id, mesh = '0', texture='0'): |
||||||
|
if photoN == 'photo1': |
||||||
|
filename = os.path.join(workdir, pid, photoN, f'{camera_id}_1.xmp') |
||||||
|
else: |
||||||
|
filename = os.path.join(workdir, pid, photoN, f'{camera_id}_8.xmp') |
||||||
|
with open(filename, 'r') as f: |
||||||
|
lines = f.readlines() |
||||||
|
lines = [line.replace('xcr:InMeshing="0"', f'xcr:InMeshing="{mesh}"') for line in lines] |
||||||
|
lines = [line.replace('xcr:InMeshing="1"', f'xcr:InMeshing="{mesh}"') for line in lines] |
||||||
|
lines = [line.replace('xcr:InTexturing="0"', f'xcr:InTexturing="{texture}"') for line in lines] |
||||||
|
lines = [line.replace('xcr:InTexturing="1"', f'xcr:InTexturing="{texture}"') for line in lines] |
||||||
|
with open(filename, 'w') as f: |
||||||
|
f.writelines(lines) |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def diff_time(start_time): |
||||||
|
# 按照分:秒的方式返回时间差 |
||||||
|
end_time = time.time() |
||||||
|
diff = end_time - start_time |
||||||
|
m, s = divmod(diff, 60) |
||||||
|
return f'{int(m)}分{int(s)}秒' |
||||||
|
|
||||||
|
|
||||||
|
def filter_dark_texture_image(pid): |
||||||
|
start_time = time.time() |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 开始检测射灯异常图片...') |
||||||
|
def get_image_v(image): |
||||||
|
# 图片左上角和右上角各取200*200区域，计算V通道均值 |
||||||
|
left_rect = image[0:200, 0:200] |
||||||
|
right_rect = image[0:200, -200:] |
||||||
|
|
||||||
|
left_hsv = cv2.cvtColor(left_rect, cv2.COLOR_BGR2HSV) |
||||||
|
left_v = left_hsv[:, :, 2] |
||||||
|
left_avg_v = np.mean(left_v) |
||||||
|
# print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 图片左上角V通道均值：{left_avg_v}') |
||||||
|
|
||||||
|
right_hsv = cv2.cvtColor(right_rect, cv2.COLOR_BGR2HSV) |
||||||
|
right_v = right_hsv[:, :, 2] |
||||||
|
right_avg_v = np.mean(right_v) |
||||||
|
# print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 图片右上角V通道均值：{right_avg_v}') |
||||||
|
return (left_avg_v + right_avg_v) / 2 |
||||||
|
|
||||||
|
v_list = [] |
||||||
|
for filename in os.listdir(os.path.join(config['workdir'], pid, 'photo2')): |
||||||
|
if filename.endswith(".jpg"): |
||||||
|
image = cv2.imread(os.path.join(config['workdir'], pid, 'photo2', filename)) |
||||||
|
v = get_image_v(image) |
||||||
|
item = {'filename': filename, 'v': v} |
||||||
|
v_list.append(item) |
||||||
|
v_list.sort(key=lambda x: x['v']) |
||||||
|
v_list = v_list[5: -5] |
||||||
|
avg_v = np.mean([item['v'] for item in v_list]) |
||||||
|
for item in v_list: |
||||||
|
if abs(item['v'] - avg_v) > 50: |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 图片{item["filename"]} V通道值{item["v"]}，低于平均值{avg_v}，将不参与贴图') |
||||||
|
set_photo_join_type(config['workdir'], pid, 'photo2', item['filename'].split('_')[0], mesh='1', texture='0') |
||||||
|
|
||||||
|
# 复制xmp文件到photo3目录，如果photo3目录存在的话 |
||||||
|
if os.path.exists(os.path.join(config['workdir'], pid, 'photo3')): |
||||||
|
shutil.copyfile(os.path.join(config['workdir'], pid, 'photo2', item['filename'].replace('jpg', 'xmp')), os.path.join(config['workdir'], pid, 'photo3', item['filename'].replace('jpg', 'xmp'))) |
||||||
|
|
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 射灯异常图片检测完成，共费时{diff_time(start_time)}') |
||||||
|
|
||||||
|
def detect_markers(psid, pid): |
||||||
|
def fix_region(): |
||||||
|
region_filename = os.path.join(config['workdir'], pid, f'{pid}.rcbox') |
||||||
|
with open(region_filename, 'r') as f: |
||||||
|
lines = f.readlines() |
||||||
|
lines = [line.replace('"NONE" globalCoordinateSystemWkt="NONE" globalCoordinateSystemName="NONE"', '"+proj=geocent +ellps=WGS84 +no_defs" globalCoordinateSystemName="local:1 - Euclidean"') for line in lines] |
||||||
|
|
||||||
|
start_time = time.time() |
||||||
|
textpicCmd ='-selectImage "'+os.path.join(config['workdir'],pid,'photo2')+'\*" -enableTexturingAndColoring true' |
||||||
|
|
||||||
|
#获取当前的工作目录 |
||||||
|
dirCurr = os.getcwd() |
||||||
|
|
||||||
|
cmd = f'{config["rcbin"]} {config["r2"]["init"]} -setInstanceName {pid} \ |
||||||
|
-save "{os.path.join(config["workdir"], pid, f"{pid}.rcproj")}" \ |
||||||
|
-addFolder "{os.path.join(config["workdir"], pid, "photo1")}" {config["r"]["setTextureFalse"]} -align -addFolder "{os.path.join(config["workdir"], pid, "photo2")}" \ |
||||||
|
-align -selectAllImages \ |
||||||
|
-detectMarkers "{dirCurr}\detectMarkers.config.xml" \ |
||||||
|
-defineDistance 36h11:001 36h11:002 1 -defineDistance 36h11:002 36h11:004 1 -defineDistance 36h11:004 36h11:003 1 -defineDistance 36h11:003 36h11:001 1 -align -align -update {config["r2"]["setRegion"]} \ |
||||||
|
-exportXMP "{dirCurr}\exportXMP.config.xml" \ |
||||||
|
-exportControlPointsMeasurements "{os.path.join(config["workdir"], pid, f"{pid}.controlPoints.csv")}" "{dirCurr}\exportControlPoints.config.xml" \ |
||||||
|
-exportReconstructionRegion "{os.path.join(config["workdir"], pid, f"{pid}.rcbox")}" \ |
||||||
|
{textpicCmd} -save "{os.path.join(config["workdir"], pid, f"{pid}.rcproj")}" -quit' |
||||||
|
print(cmd) |
||||||
|
cmd = shlex.split(cmd) |
||||||
|
res = subprocess.run(cmd) |
||||||
|
time.sleep(3) |
||||||
|
fix_region() |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 定位点检测完成, 共费时{diff_time(start_time)}') |
||||||
|
|
||||||
|
def step1(pid,headsCount): |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 开始处理{pid}建模任务') |
||||||
|
start_time = time.time() |
||||||
|
# 检测图片定位点，定位异常及时上报微信通知给客服人员，人工判断是否需要重新拍摄或更换地贴。定义定位点距离，导出相机位姿信息 |
||||||
|
detect_markers(0, pid) |
||||||
|
# 处理图片，检测photo2中的异常图片不参与贴图，以免破坏贴图效果，默认不检测射灯异常图片，以节省算力成本 |
||||||
|
filter_dark_texture_image(pid) |
||||||
|
|
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} step1图片预处理完成，共费时{diff_time(start_time)}') |
||||||
|
# os.system(f'python {os.getcwd()}/main_step2.py {pid}') |
||||||
|
step2(pid,headsCount) |
||||||
|
|
||||||
|
def step2(pid,headsCount): |
||||||
|
dirCurr = os.getcwd() |
||||||
|
simplify_value = 1000000 * int(headsCount) |
||||||
|
calulate_type = 'calculateNormalModel' |
||||||
|
cmd = f'{config["rcbin"]} {config["r2"]["init"]} -setInstanceName {pid} \ |
||||||
|
-load "{os.path.join(config["workdir"], pid, f"{pid}.rcproj")}" \ |
||||||
|
-{calulate_type} \ |
||||||
|
-selectLargestModelComponent -invertTrianglesSelection -removeSelectedTriangles -simplify {simplify_value} -smooth -closeHoles -cleanModel -calculateTexture \ |
||||||
|
-save "{os.path.join(config["workdir"], pid, f"{pid}.rcproj")}" \ |
||||||
|
-exportSelectedModel "{os.path.join(config["workdir"], pid, "output", f"{pid}.obj")}" "{dirCurr}\\ModelExportParams.xml" -quit' |
||||||
|
print(cmd) |
||||||
|
cmd = shlex.split(cmd) |
||||||
|
res = subprocess.run(cmd) |
||||||
|
#阻塞判断是否导出完成 |
||||||
|
while True: |
||||||
|
#判断 output 目录下是否存在 三个文件 |
||||||
|
files = os.listdir(os.path.join(config['workdir'], pid, "output")) |
||||||
|
if len(files) >= 3: |
||||||
|
break |
||||||
|
|
||||||
|
#将导出的文件移动到指定目录 |
||||||
|
sourceDir = config["local_complate_build"] |
||||||
|
#判断目录是否存在，不存在就创建 |
||||||
|
if not os.path.exists(sourceDir): |
||||||
|
os.makedirs(sourceDir) |
||||||
|
|
||||||
|
#将 output 目录里的文件移动到指定目录 |
||||||
|
shutil.move(os.path.join(config['workdir'], pid, "output"),os.path.join(sourceDir,pid)) |
||||||
|
#发起请求告知后端建模完成 |
||||||
|
try: |
||||||
|
res = requests.post(url, data={'id': pid,'status':5000}) |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid:{pid} 任务完成- {res.text}') |
||||||
|
except Exception as e: |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid:{pid} - 建模完成状态变更失败 - {e}') |
||||||
|
|
||||||
|
|
||||||
|
def main(pid,headsCount): |
||||||
|
step1(pid, headsCount) |
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
try: |
||||||
|
url = config['urls']['update_status_modelsuccess_url'] |
||||||
|
if url == "": |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 请配置建模完成回调地址') |
||||||
|
exit() |
||||||
|
|
||||||
|
# 取云端redis任务，完成第一步的数据预处理后，将数据放入共享存储目录，将第二步任务塞入本地mysql队列 |
||||||
|
# 默认循环值守，可传参数运行单一任务，以方便调试 |
||||||
|
pid = '0' |
||||||
|
if len(sys.argv) == 3: |
||||||
|
pids = sys.argv[1].split(',') |
||||||
|
headsCount = sys.argv[2] |
||||||
|
for pid in pids: |
||||||
|
main(pid,headsCount) |
||||||
|
exit() |
||||||
|
|
||||||
|
#获取文件夹下的所有文件 |
||||||
|
needBuildDir = config['local_need_build'] |
||||||
|
workdir = config['workdir'] |
||||||
|
if workdir == '': |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 请配置建模工作目录') |
||||||
|
exit() |
||||||
|
|
||||||
|
#不存在就创建工作目录 |
||||||
|
if not os.path.exists(workdir): |
||||||
|
os.makedirs(workdir) |
||||||
|
|
||||||
|
if not os.path.exists(needBuildDir): |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 需要建模目录不存在- {needBuildDir}') |
||||||
|
exit() |
||||||
|
except Exception as e: |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 初始化异常 - {e}') |
||||||
|
|
||||||
|
|
||||||
|
#遍历文件夹下的所有文件进行处理 |
||||||
|
try: |
||||||
|
while True: |
||||||
|
time.sleep(5) |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 建模任务循环开始') |
||||||
|
for pid in os.listdir(needBuildDir): |
||||||
|
if not os.path.isdir(os.path.join(needBuildDir,pid)): |
||||||
|
continue |
||||||
|
|
||||||
|
#判断是否存在 canDo.txt 文件，如果不存在就跳过 |
||||||
|
if not os.path.exists(os.path.join(needBuildDir,pid,"canDo.txt")): |
||||||
|
continue |
||||||
|
|
||||||
|
#读取 canDo.txt 文件内容 |
||||||
|
headsCounts = 1 |
||||||
|
with open(os.path.join(needBuildDir,pid,"canDo.txt"),"r") as f: |
||||||
|
headsCounts = f.read() |
||||||
|
|
||||||
|
|
||||||
|
#判断文件夹里是否存在 do.txt , 如果存在就跳过，不存在就创建 |
||||||
|
if os.path.exists(os.path.join(needBuildDir,pid,"do.txt")): |
||||||
|
continue |
||||||
|
else: |
||||||
|
with open(os.path.join(needBuildDir,pid,"do.txt"),"w") as f: |
||||||
|
f.write("1") |
||||||
|
|
||||||
|
#判断是否已经存在了，存在就删除 |
||||||
|
if os.path.exists(os.path.join(workdir,pid)): |
||||||
|
shutil.rmtree(os.path.join(workdir,pid)) |
||||||
|
|
||||||
|
#将文件夹移动到工作目录 |
||||||
|
shutil.move(os.path.join(needBuildDir,pid),os.path.join(workdir,pid)) |
||||||
|
#执行建模流程,建模中的流程 |
||||||
|
try: |
||||||
|
requests.post(url, data={'id': pid,'status':4000}) |
||||||
|
except Exception as e: |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid:{pid} - 状态变更失败 - {e}') |
||||||
|
#continue |
||||||
|
|
||||||
|
main(pid,headsCounts) |
||||||
|
except Exception as e: |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 建模任务循环异常 - {e}') |
||||||
|
exit() |
||||||
|
|
||||||
Loading…
Reference in new issue